diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-02-13 17:59:09 +0000 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-02-19 21:01:13 +0000 |
commit | 102b0ce919bf5415993cdd6b0e4385b35734d38d (patch) | |
tree | f711c876711e178a04c3684bb15e202ebf4d14ac /arm_compute/graph/nodes | |
parent | 51545e4acbb6a613da5ee19110ca8c8d7fd39776 (diff) | |
download | ComputeLibrary-102b0ce919bf5415993cdd6b0e4385b35734d38d.tar.gz |
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 <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2727
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/graph/nodes')
-rw-r--r-- | arm_compute/graph/nodes/PadLayerNode.h | 13 |
1 files changed, 10 insertions, 3 deletions
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 |