aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/NEON/functions/NEFFTConvolutionLayer.h
diff options
context:
space:
mode:
authorSheri Zhang <sheri.zhang@arm.com>2021-04-22 14:41:12 +0100
committerSheri Zhang <sheri.zhang@arm.com>2021-04-28 12:52:32 +0000
commita47dcc229d912d4e4bb5afa37220d20451f243a7 (patch)
treef8b296701fbdebfc7d29abc09144c49619bcca1c /arm_compute/runtime/NEON/functions/NEFFTConvolutionLayer.h
parent2b7fee089c76226bfafcae77ba49f1eddb1e01da (diff)
downloadComputeLibrary-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/NEFFTConvolutionLayer.h')
-rw-r--r--arm_compute/runtime/NEON/functions/NEFFTConvolutionLayer.h8
1 files changed, 8 insertions, 0 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEFFTConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEFFTConvolutionLayer.h
index 213fa6093b..c5f4d45b6b 100644
--- a/arm_compute/runtime/NEON/functions/NEFFTConvolutionLayer.h
+++ b/arm_compute/runtime/NEON/functions/NEFFTConvolutionLayer.h
@@ -73,6 +73,14 @@ public:
~NEFFTConvolutionLayer();
/** Set the input and output tensors.
*
+ * Valid data layouts:
+ * - All
+ *
+ * Valid data type configurations:
+ * |src |dst |
+ * |:------|:------|
+ * |F32 |F32 |
+ *
* @note: This function only works with any square kernel size and unit strides for both NCHW and NHWC data layout
*
* @param[in] input Source tensor. 3 lower dimensions represent a single input [width, height, IFM],