From c1b76faf6be5c33dbf3269faea95e185ac37992f Mon Sep 17 00:00:00 2001 From: Manuel Bottini Date: Mon, 17 Jun 2019 12:04:40 +0100 Subject: COMPMID-2092: Refactoring interface for the deconvolution kernels (NEON/CL) 3RDPARTY_UPDATE Change-Id: Id7ddf97e2c9ceb2cb84084fab2c6f5697890c193 Signed-off-by: giuros01 Reviewed-on: https://review.mlplatform.org/c/1424 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Georgios Pinitas --- arm_compute/graph/frontend/Layers.h | 20 ++++++++------------ 1 file changed, 8 insertions(+), 12 deletions(-) (limited to 'arm_compute/graph/frontend/Layers.h') diff --git a/arm_compute/graph/frontend/Layers.h b/arm_compute/graph/frontend/Layers.h index f73b1f49d9..3fc4af46d5 100644 --- a/arm_compute/graph/frontend/Layers.h +++ b/arm_compute/graph/frontend/Layers.h @@ -370,26 +370,23 @@ class DeconvolutionLayer final : public ILayer public: /** Construct a convolution layer. * - * @param[in] conv_width Convolution width. - * @param[in] conv_height Convolution height. - * @param[in] ofm Output feature map. - * @param[in] weights Accessor to get kernel weights from. - * @param[in] bias Accessor to get kernel bias from. - * @param[in] deconv_info Padding and stride information. - * @param[in] inner_border Inner border padding (right, top) + * @param[in] conv_width Convolution width. + * @param[in] conv_height Convolution height. + * @param[in] ofm Output feature map. + * @param[in] weights Accessor to get kernel weights from. + * @param[in] bias Accessor to get kernel bias from. + * @param[in] deconv_info Padding and stride information. */ DeconvolutionLayer(unsigned int conv_width, unsigned int conv_height, unsigned int ofm, ITensorAccessorUPtr weights, ITensorAccessorUPtr bias, - PadStrideInfo deconv_info, - Size2D inner_border) + PadStrideInfo deconv_info) : _conv_width(conv_width), _conv_height(conv_height), _ofm(ofm), _deconv_info(std::move(deconv_info)), - _inner_border(inner_border), _weights(std::move(weights)), _bias(std::move(bias)) { @@ -400,7 +397,7 @@ public: NodeIdxPair input = { s.tail_node(), 0 }; NodeParams common_params = { name(), s.hints().target_hint }; return GraphBuilder::add_deconvolution_node(s.graph(), common_params, input, - Size2D(_conv_width, _conv_height), _ofm, _deconv_info, _inner_border, + Size2D(_conv_width, _conv_height), _ofm, _deconv_info, std::move(_weights), std::move(_bias)); } @@ -409,7 +406,6 @@ private: unsigned int _conv_height; unsigned int _ofm; const PadStrideInfo _deconv_info; - Size2D _inner_border; ITensorAccessorUPtr _weights; ITensorAccessorUPtr _bias; }; -- cgit v1.2.1