24.02.1
|
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, arm_compute::WeightFormat weight_format=arm_compute::WeightFormat::UNSPECIFIED) | |
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 |
arm_compute::WeightFormat | weight_format () const |
void | set_weight_format (arm_compute::WeightFormat weight_format) |
unsigned int | kernel_width () const |
unsigned int | kernel_height () 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. |
[in] | weight_format | (Optional) arm_gemm:WeightFormat enumeration requested by the user. Default is arm_compute::WeightFormat::UNSPECIFIED. |
Definition at line 1692 of file Types.h.
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
Definition at line 1738 of file Types.h.
References WeightsInfo::weight_format().
|
inline |
Definition at line 1734 of file Types.h.
Referenced by WeightsInfo::set_weight_format().