From 102b0ce919bf5415993cdd6b0e4385b35734d38d Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Thu, 13 Feb 2020 17:59:09 +0000 Subject: COMPMID-3105: Update pad layer to PadV2 Adds the ability to set the padding value in the pad layer node. Change-Id: I2f323146cbdfdfd6cfd617ee4bcafb5ffafedbbb Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2727 Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- arm_compute/graph/nodes/PadLayerNode.h | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) (limited to 'arm_compute/graph/nodes') diff --git a/arm_compute/graph/nodes/PadLayerNode.h b/arm_compute/graph/nodes/PadLayerNode.h index 0b1bddac9b..852427a593 100644 --- a/arm_compute/graph/nodes/PadLayerNode.h +++ b/arm_compute/graph/nodes/PadLayerNode.h @@ -36,15 +36,21 @@ class PadLayerNode final : public INode public: /** Constructor * - * @param[in] padding The padding for each spatial dimension of the input tensor. The pair padding[i] - * specifies the front and the end padding in the i-th dimension. + * @param[in] padding The padding for each spatial dimension of the input tensor. The pair padding[i] + * specifies the front and the end padding in the i-th dimension. + * @param[in] pad_value Padding value to be used. Defaults to 0 */ - PadLayerNode(const PaddingList &padding); + PadLayerNode(const PaddingList &padding, PixelValue pad_value = PixelValue()); /** Padding list accessor * * @return Padding list */ const PaddingList &padding() const; + /** Padding value accessor + * + * @return Padding value + */ + PixelValue pad_value() const; // Inherited overridden methods: NodeType type() const override; @@ -54,6 +60,7 @@ public: private: PaddingList _padding; + PixelValue _pad_value; }; } // namespace graph } // namespace arm_compute -- cgit v1.2.1