diff options
Diffstat (limited to 'arm_compute/function_info/FullyConnectedLayerInfo.h')
| -rw-r--r-- | arm_compute/function_info/FullyConnectedLayerInfo.h | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/arm_compute/function_info/FullyConnectedLayerInfo.h b/arm_compute/function_info/FullyConnectedLayerInfo.h index 5f5578eadd..e65daeb2d4 100644 --- a/arm_compute/function_info/FullyConnectedLayerInfo.h +++ b/arm_compute/function_info/FullyConnectedLayerInfo.h @@ -35,13 +35,13 @@ struct FullyConnectedLayerInfo /* Fused-activation parameters */ ActivationLayerInfo activation_info{}; /**< Fused activation to apply after the matrix multiplication. */ /* Information about weights */ - DataLayout weights_trained_layout{ DataLayout::NCHW }; /**< Layout that the weights have been trained with. */ - bool transpose_weights{ true }; /**< Transpose weights if true. */ - bool are_weights_reshaped{ false }; /**< @deprecated Reshape the weights tensor if false. */ - bool retain_internal_weights{ false }; /**< Retain internal reshaped weights. */ - bool enable_fast_math{ false }; /**< Enable fast math computation. */ + DataLayout weights_trained_layout{DataLayout::NCHW}; /**< Layout that the weights have been trained with. */ + bool transpose_weights{true}; /**< Transpose weights if true. */ + bool are_weights_reshaped{false}; /**< @deprecated Reshape the weights tensor if false. */ + bool retain_internal_weights{false}; /**< Retain internal reshaped weights. */ + bool enable_fast_math{false}; /**< Enable fast math computation. */ /* Other parameters */ - bool fp_mixed_precision{ false }; /**< Use wider accumulators (32 bit instead of 16 for FP16) to improve accuracy. */ + bool fp_mixed_precision{false}; /**< Use wider accumulators (32 bit instead of 16 for FP16) to improve accuracy. */ /** Sets the weights trained data layout * |