21.08
|
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 | |
ActivationLayerInfo | activation_info {} |
Fused activation to apply after the matrix multiplication. More... | |
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 | constant_weights { true } |
If false, weights can vary between runs. More... | |
bool | fp_mixed_precision { false } |
Use wider accumulators (32 bit instead of 16 for FP16) to improve accuracy. More... | |
|
inline |
Sets the transpose weights flag.
[in] | should_transpose_weights | Boolean flag indicating if weights should be transposed |
Definition at line 1568 of file Types.h.
|
inline |
Sets the weights trained data layout.
[in] | layout | Data layout that the weights were trained with |
Definition at line 1557 of file Types.h.
ActivationLayerInfo activation_info {} |
Fused activation to apply after the matrix multiplication.
Definition at line 1541 of file Types.h.
Referenced by CpuFullyConnected::configure(), FullyConnectedLayerNode::set_fused_activation(), ClFullyConnected::validate(), and CpuFullyConnected::validate().
bool are_weights_reshaped { false } |
Reshape the weights tensor if false.
Definition at line 1545 of file Types.h.
Referenced by ClFullyConnected::configure(), CpuFullyConnected::configure(), arm_compute::test::validation::DATA_TEST_CASE(), ClFullyConnected::validate(), and CpuFullyConnected::validate().
bool constant_weights { true } |
If false, weights can vary between runs.
Definition at line 1547 of file Types.h.
Referenced by ClFullyConnected::validate(), and CpuFullyConnected::validate().
bool fp_mixed_precision { false } |
bool retain_internal_weights { false } |
Retain internal reshaped weights.
Definition at line 1546 of file Types.h.
Referenced by CLFullyConnectedLayer::configure(), ClFullyConnected::configure(), CpuFullyConnected::configure(), and CpuFullyConnected::validate().
bool transpose_weights { true } |
Transpose weights if true.
Definition at line 1544 of file Types.h.
Referenced by FullyConnectedLayerNode::compute_weights_descriptor(), ClFullyConnected::configure(), CpuFullyConnected::configure(), arm_compute::test::validation::DATA_TEST_CASE(), ClFullyConnected::validate(), and CpuFullyConnected::validate().
DataLayout weights_trained_layout { DataLayout::NCHW } |
Layout that the weights have been trained with.
Definition at line 1543 of file Types.h.
Referenced by ClFullyConnected::configure(), CpuFullyConnected::configure(), ClFullyConnected::validate(), and CpuFullyConnected::validate().