diff options
author | giuros01 <giuseppe.rossini@arm.com> | 2019-04-01 13:50:22 +0100 |
---|---|---|
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-05-10 08:26:44 +0000 |
commit | 46a49a0a8206f0efa7afd514940e180a88ffd732 (patch) | |
tree | 0ec53af4ef65037e357b1d8f6a1d1f65075659f7 /arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h | |
parent | 879e8dd2fc8523e4059ba9ced9ea0edb57103778 (diff) | |
download | ComputeLibrary-46a49a0a8206f0efa7afd514940e180a88ffd732.tar.gz |
COMPMID-1635: Optimize CLDeconvolutionLayer - Part III
Change-Id: Id2661e093a669ef3eaf2a5116cd278a80c1d5a89
Signed-off-by: giuros01 <giuseppe.rossini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/935
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-by: Isabella Gottardi <isabella.gottardi@arm.com>
Comments-Addressed: Isabella Gottardi <isabella.gottardi@arm.com>
Tested-by: Isabella Gottardi <isabella.gottardi@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h')
-rw-r--r-- | arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h b/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h index b613708c50..e5b406ee5e 100644 --- a/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h +++ b/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h @@ -25,6 +25,7 @@ #define __ARM_COMPUTE_CLDECONVOLUTIONLAYER_H__ #include "arm_compute/runtime/CL/functions/CLDirectDeconvolutionLayer.h" +#include "arm_compute/runtime/CL/functions/CLGEMMDeconvolutionLayer.h" #include "arm_compute/runtime/IFunction.h" #include "arm_compute/runtime/IMemoryManager.h" @@ -34,6 +35,7 @@ namespace arm_compute { /** Basic function to compute the deconvolution layer. This function calls the following OpenCL kernels/functions: * + * -# @ref CLGEMMDeconvolutionLayer * -# @ref CLDirectDeconvolutionLayer */ class CLDeconvolutionLayer : public IFunction @@ -44,7 +46,7 @@ public: /** Set the input, weights, biases and output tensors. * - * @deprecated This method is deprecated and will be removed in release 19.05 + * @note This method will be deprecated in the next release. * * @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: QASYMM8/F16/F32. * @param[in] weights The 4d weights with dimensions [width, height, IFM, OFM]. Data type supported: Same as @p input. @@ -60,7 +62,7 @@ public: unsigned int inner_border_right, unsigned int inner_border_top, const WeightsInfo &weights_info = WeightsInfo()); /** Static function to check if given info will lead to a valid configuration of @ref CLDeconvolutionLayer * - * @deprecated This method is deprecated and will be removed in release 19.05 + * @note This method will be deprecated in the next release. * * @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: QASYMM8/F16/F32. * @param[in] weights The 4d weights info with dimensions [width, height, IFM, OFM]. Data type supported: Same as @p input. @@ -101,6 +103,8 @@ public: static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *bias, ITensorInfo *output, const PadStrideInfo &deconv_info, const WeightsInfo &weights_info = WeightsInfo()); + static DeconvolutionMethod get_deconvolution_method(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *bias, ITensorInfo *output, const PadStrideInfo &deconv_info, + const WeightsInfo &weights_info); // Inherited methods overridden: void run() override; void prepare() override; @@ -109,5 +113,5 @@ private: std::shared_ptr<IMemoryManager> _memory_manager; std::unique_ptr<IFunction> _function; }; -} +} // namespace arm_compute #endif /* __ARM_COMPUTE_CLDECONVOLUTIONLAYER_H__ */ |