24.04
|
Fully connected layer info. More...
#include <FullyConnectedLayerInfo.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} |
bool | retain_internal_weights {false} |
Retain internal reshaped weights. More... | |
bool | enable_fast_math {false} |
Enable fast math computation. More... | |
bool | fp_mixed_precision {false} |
Use wider accumulators (32 bit instead of 16 for FP16) to improve accuracy. More... | |
Fully connected layer info.
Definition at line 33 of file FullyConnectedLayerInfo.h.
|
inline |
Sets the transpose weights flag.
[in] | should_transpose_weights | Boolean flag indicating if weights should be transposed |
Definition at line 63 of file FullyConnectedLayerInfo.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 52 of file FullyConnectedLayerInfo.h.
References FullyConnectedLayerInfo::weights_trained_layout.
ActivationLayerInfo activation_info {} |
Fused activation to apply after the matrix multiplication.
Definition at line 36 of file FullyConnectedLayerInfo.h.
Referenced by CpuFullyConnected::configure(), CpuFullyConnected::has_opt_impl(), arm_compute::operator<<(), FullyConnectedLayerNode::set_fused_activation(), CpuFullyConnected::validate(), and ClFullyConnected::validate().
bool are_weights_reshaped {false} |
Definition at line 40 of file FullyConnectedLayerInfo.h.
Referenced by CLFullyConnectedLayer::configure(), CpuFullyConnected::configure(), ClFullyConnected::configure(), NEFullyConnectedLayer::configure(), arm_compute::operator<<(), CpuFullyConnected::validate(), and ClFullyConnected::validate().
bool enable_fast_math {false} |
Enable fast math computation.
Definition at line 42 of file FullyConnectedLayerInfo.h.
Referenced by CpuFullyConnected::configure(), arm_compute::graph::backends::detail::create_fully_connected_layer(), CpuFullyConnected::has_opt_impl(), and CpuFullyConnected::validate().
bool fp_mixed_precision {false} |
Use wider accumulators (32 bit instead of 16 for FP16) to improve accuracy.
Definition at line 44 of file FullyConnectedLayerInfo.h.
Referenced by arm_compute::operator<<().
bool retain_internal_weights {false} |
Retain internal reshaped weights.
Definition at line 41 of file FullyConnectedLayerInfo.h.
Referenced by CLFullyConnectedLayer::configure(), CpuFullyConnected::configure(), ClFullyConnected::configure(), NEFullyConnectedLayer::configure(), arm_compute::operator<<(), and CpuFullyConnected::validate().
bool transpose_weights {true} |
Transpose weights if true.
Definition at line 39 of file FullyConnectedLayerInfo.h.
Referenced by FullyConnectedLayerNode::compute_weights_descriptor(), CLFullyConnectedLayer::configure(), CpuFullyConnected::configure(), ClFullyConnected::configure(), NEFullyConnectedLayer::configure(), arm_compute::operator<<(), FullyConnectedLayerInfo::set_transpose_weights(), CpuFullyConnected::validate(), and ClFullyConnected::validate().
DataLayout weights_trained_layout {DataLayout::NCHW} |
Layout that the weights have been trained with.
Definition at line 38 of file FullyConnectedLayerInfo.h.
Referenced by CpuFullyConnected::configure(), ClFullyConnected::configure(), arm_compute::operator<<(), FullyConnectedLayerInfo::set_weights_trained_layout(), CpuFullyConnected::validate(), and ClFullyConnected::validate().