21.05
|
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 1567 of file Types.h.
References FullyConnectedLayerInfo::transpose_weights.
|
inline |
Sets the weights trained data layout.
[in] | layout | Data layout that the weights were trained with |
Definition at line 1556 of file Types.h.
References FullyConnectedLayerInfo::weights_trained_layout.
ActivationLayerInfo activation_info {} |
Fused activation to apply after the matrix multiplication.
Definition at line 1548 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 1545 of file Types.h.
Referenced by NEFullyConnectedLayer::configure(), CLFullyConnectedLayer::configure(), NEFullyConnectedLayer::validate(), and CLFullyConnectedLayer::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 NEFullyConnectedLayer::configure(), CLFullyConnectedLayer::configure(), and NEFullyConnectedLayer::validate().
bool transpose_weights { true } |
Transpose weights if true.
Definition at line 1544 of file Types.h.
Referenced by FullyConnectedLayerNode::compute_weights_descriptor(), NEFullyConnectedLayer::configure(), CLFullyConnectedLayer::configure(), FullyConnectedLayerInfo::set_transpose_weights(), NEFullyConnectedLayer::validate(), and CLFullyConnectedLayer::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 NEFullyConnectedLayer::configure(), CLFullyConnectedLayer::configure(), FullyConnectedLayerInfo::set_weights_trained_layout(), NEFullyConnectedLayer::validate(), and CLFullyConnectedLayer::validate().