diff options
Diffstat (limited to 'arm_compute/core/Types.h')
-rw-r--r-- | arm_compute/core/Types.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 1363324e3b..343952f0b2 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -682,6 +682,15 @@ private: DimensionRoundingType _round_type; }; +/** Fully connected layer info */ +struct FullyConnectedLayerInfo +{ + 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 }; /**< Reshape the weights tensor if false. */ + bool retain_internal_weights{ false }; /**< Retain internal reshaped weights. */ +}; + /** Pooling Layer Information class */ class PoolingLayerInfo { |