diff options
| author | Sheri Zhang <sheri.zhang@arm.com> | 2021-04-22 14:41:12 +0100 |
|---|---|---|
| committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-04-28 12:52:32 +0000 |
| commit | a47dcc229d912d4e4bb5afa37220d20451f243a7 (patch) | |
| tree | f8b296701fbdebfc7d29abc09144c49619bcca1c /arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h | |
| parent | 2b7fee089c76226bfafcae77ba49f1eddb1e01da (diff) | |
| download | ComputeLibrary-a47dcc229d912d4e4bb5afa37220d20451f243a7.tar.gz | |
Update operator list documentation
All the common information for the operators are stored in OperatorList.h.
All data type and data layout information for the operators are store in the function header files.
Partially resolve: COMPMID-4199
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I272948cfb3f84e42232a82dd84c0158d84642099
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5511
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h b/arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h index f9ce66db13..218877d421 100644 --- a/arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h +++ b/arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h @@ -51,6 +51,15 @@ public: ~NEConvertFullyConnectedWeights(); /** Initialize the function. * + * Valid data layouts: + * - NHWC + * - NCHW + * + * Valid data type configurations: + * |src |dst | + * |:--------------|:--------------| + * |All |All | + * * @param[in] input Source weights tensor to convert. Must be 2 dimensional. Data types supported: All. * @param[out] output The converted weights tensor. Shape and Data Type: Same as @p input. * @param[in] original_input_shape Shape of the original input tensor (the one entering fully connected layer). |