diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-04-04 11:03:51 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:16 +0000 |
commit | a8ca2b0cfe052c9a28b691317a674f28f495c139 (patch) | |
tree | 32be05d07424ed4a7142904101e86090f33ace33 /arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h | |
parent | c0d1c86b1bb1b4e129c292549845e00dfd8abfee (diff) | |
download | ComputeLibrary-a8ca2b0cfe052c9a28b691317a674f28f495c139.tar.gz |
COMPMID-959 Change DeConvolution weights in doxygen comments
Update weights shape in doxygen from [W, H, OFM, IFM] to
[W, H, IFM, OFM] in order to match ConvolutionLayer. The
shape calculation is luckily correct.
Change-Id: I589fe379b1cb5331dfc43a60b060adb17b2a5517
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/126570
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h index 1d5dd3fbb4..66c6d427ba 100644 --- a/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h @@ -81,7 +81,7 @@ public: /** Set the input, weights, biases and output tensors. * * @param[in,out] input Input tensor. 3 lower dimensions represent a single input, and an optional 4th dimension for batch of inputs. Data types supported: F32. - * @param[in] weights The 4d weights with dimensions [width, height, OFM, IFM]. Data type supported: Same as @p input. + * @param[in] weights The 4d weights with dimensions [width, height, IFM, OFM]. Data type supported: Same as @p input. * @param[in] bias Optional, ignored if NULL. The biases have one dimension. Data type supported: Same as @p input. * @param[out] output Output tensor. The output has the same number of dimensions as the @p input. * @param[in] info Contains padding and policies to be used in the deconvolution, this is decribed in @ref PadStrideInfo. @@ -94,7 +94,7 @@ public: /** Static function to check if given info will lead to a valid configuration of @ref NEDeconvolutionLayer * * @param[in] input Input tensor info. 3 lower dimensions represent a single input, and an optional 4th dimension for batch of inputs. Data types supported: F32. - * @param[in] weights The 4d weights info with dimensions [width, height, OFM, IFM]. Data type supported: Same as @p input. + * @param[in] weights The 4d weights info with dimensions [width, height, IFM, OFM]. Data type supported: Same as @p input. * @param[in] bias (Optional) The biases have one dimension. Data type supported: Same as @p input. * @param[in] output Output tensor info. The output has the same number of dimensions as the @p input. * @param[in] info Contains padding and policies to be used in the deconvolution, this is decribed in @ref PadStrideInfo. |