WeightSizeVisitor Class Reference

WeightSizeVisitor returns the number of weights of the given module. More...

Inheritance diagram for WeightSizeVisitor:

Public Member Functions

template
<
typename
LayerType
>
size_t operator() (LayerType *layer) const
 Return the number of weights. More...

 
size_t operator() (MoreTypes layer) const
 

Detailed Description

WeightSizeVisitor returns the number of weights of the given module.

Definition at line 27 of file weight_size_visitor.hpp.

Member Function Documentation

◆ operator()() [1/2]

size_t operator() ( LayerType *  layer) const

Return the number of weights.

◆ operator()() [2/2]

size_t operator() ( MoreTypes  layer) const

The documentation for this class was generated from the following file: