21.08
|
Convolution Layer Weights Information class. More...
#include <Types.h>
Public Member Functions | |
WeightsInfo () | |
Default constructor. More... | |
WeightsInfo (bool are_reshaped, unsigned int kernel_width, unsigned int kernel_height, unsigned int num_kernels, bool retain_internal_weights=false) | |
Constructor. More... | |
bool | are_reshaped () const |
Flag which specifies if the weights tensor has been reshaped. More... | |
unsigned int | num_kernels () const |
Return the number of convolution kernels. More... | |
std::pair< unsigned int, unsigned int > | kernel_size () const |
Return the width and height of the kernel. More... | |
bool | retain_internal_weights () const |
Convolution Layer Weights Information class.
This class stores the necessary information to compute convolution layer when the weights are already reshaped
|
inline |
|
inline |
Constructor.
[in] | are_reshaped | True if the weights have been reshaped |
[in] | kernel_width | Kernel width. |
[in] | kernel_height | Kernel height. |
[in] | num_kernels | Number of convolution kernels. |
[in] | retain_internal_weights | (Optional) True if internal reshaped weights must be retained. Used for reconfiguration purposes. Default is false. |
Definition at line 1710 of file Types.h.
|
inline |
Flag which specifies if the weights tensor has been reshaped.
Definition at line 1718 of file Types.h.
Referenced by arm_compute::operator<<(), ClGemmConvolution::validate(), and CpuGemmConvolution::validate().
|
inline |
Return the width and height of the kernel.
Definition at line 1734 of file Types.h.
Referenced by arm_compute::operator<<().
|
inline |
Return the number of convolution kernels.
Definition at line 1726 of file Types.h.
Referenced by arm_compute::operator<<().
|
inline |
Definition at line 1738 of file Types.h.
Referenced by ClGemmConvolution::configure(), CpuGemmConvolution::configure(), and CLDirectDeconvolutionLayer::configure().