21.02
|
Fully connected layer info. More...
#include <Types.h>
Public Member Functions | |
FullyConnectedLayerInfo & | set_weights_trained_layout (DataLayout layout) |
Sets the weights trained data layout. More... | |
FullyConnectedLayerInfo & | set_transpose_weights (bool should_transpose_weights) |
Sets the transpose weights flag. More... | |
Data Fields | |
DataLayout | weights_trained_layout { DataLayout::NCHW } |
Layout that the weights have been trained with. More... | |
bool | transpose_weights { true } |
Transpose weights if true. More... | |
bool | are_weights_reshaped { false } |
Reshape the weights tensor if false. More... | |
bool | retain_internal_weights { false } |
Retain internal reshaped weights. More... | |
bool | fp_mixed_precision { false } |
Use wider accumulators (32 bit instead of 16 for FP16) to improve accuracy. More... | |
ActivationLayerInfo | activation_info {} |
Fused activation to apply after the matrix multiplication. More... | |
|
inline |
Sets the transpose weights flag.
[in] | should_transpose_weights | Boolean flag indicating if weights should be transposed |
Definition at line 1639 of file Types.h.
|
inline |
Sets the weights trained data layout.
[in] | layout | Data layout that the weights were trained with |
Definition at line 1628 of file Types.h.
ActivationLayerInfo activation_info {} |
Fused activation to apply after the matrix multiplication.
Definition at line 1620 of file Types.h.
Referenced by NEFullyConnectedLayer::configure(), FullyConnectedLayerNode::set_fused_activation(), NEFullyConnectedLayer::validate(), and CLFullyConnectedLayer::validate().
bool are_weights_reshaped { false } |
Reshape the weights tensor if false.
Definition at line 1617 of file Types.h.
Referenced by GCFullyConnectedLayer::configure(), NEFullyConnectedLayer::configure(), CLFullyConnectedLayer::configure(), arm_compute::test::validation::DATA_TEST_CASE(), NEFullyConnectedLayer::validate(), and CLFullyConnectedLayer::validate().
bool fp_mixed_precision { false } |
bool retain_internal_weights { false } |
Retain internal reshaped weights.
Definition at line 1618 of file Types.h.
Referenced by GCFullyConnectedLayer::configure(), NEFullyConnectedLayer::configure(), CLFullyConnectedLayer::configure(), and NEFullyConnectedLayer::validate().
bool transpose_weights { true } |
Transpose weights if true.
Definition at line 1616 of file Types.h.
Referenced by FullyConnectedLayerNode::compute_weights_descriptor(), GCFullyConnectedLayer::configure(), NEFullyConnectedLayer::configure(), CLFullyConnectedLayer::configure(), arm_compute::test::validation::DATA_TEST_CASE(), NEFullyConnectedLayer::validate(), and CLFullyConnectedLayer::validate().
DataLayout weights_trained_layout { DataLayout::NCHW } |
Layout that the weights have been trained with.
Definition at line 1615 of file Types.h.
Referenced by NEFullyConnectedLayer::configure(), CLFullyConnectedLayer::configure(), NEFullyConnectedLayer::validate(), and CLFullyConnectedLayer::validate().