diff options
| author | FinnWilliamsArm <Finn.Williams@arm.com> | 2019-07-09 14:46:28 +0100 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-07-09 15:41:07 +0000 |
| commit | ab801628d857cfadf258bb27916ebdaf587c02d3 (patch) | |
| tree | 58c7dd463ab958a31111c3eb22374598541c9bc9 | |
| parent | a2ec9092f0bff018bfe7ae0cacb7e30bcc17c1c7 (diff) | |
| download | armnn-ab801628d857cfadf258bb27916ebdaf587c02d3.tar.gz | |
IVGCVSW-3467 Add CL Backend support for Pad_v2 with Unit Test
Signed-off-by: FinnWilliamsArm <Finn.Williams@arm.com>
Change-Id: I4c51e49ef57065e9fb0038175e67290c59d5f396
| -rw-r--r-- | src/backends/cl/test/ClLayerTests.cpp | 11 | ||||
| -rw-r--r-- | src/backends/cl/workloads/ClPadWorkload.cpp | 4 |
2 files changed, 14 insertions, 1 deletions
diff --git a/src/backends/cl/test/ClLayerTests.cpp b/src/backends/cl/test/ClLayerTests.cpp index 5575a05b99..7d72e177eb 100644 --- a/src/backends/cl/test/ClLayerTests.cpp +++ b/src/backends/cl/test/ClLayerTests.cpp @@ -329,9 +329,20 @@ ARMNN_AUTO_TEST_CASE(SimpleReshapeUint8, SimpleReshapeTest<armnn::DataType::Quan // Pad ARMNN_AUTO_TEST_CASE(PadFloat322d, PadFloat322dTest) +ARMNN_AUTO_TEST_CASE(PadFloat322dCustomPadding, PadFloat322dCustomPaddingTest) ARMNN_AUTO_TEST_CASE(PadFloat323d, PadFloat323dTest) ARMNN_AUTO_TEST_CASE(PadFloat324d, PadFloat324dTest) +ARMNN_AUTO_TEST_CASE(PadUint82d, PadUint82dTest) +ARMNN_AUTO_TEST_CASE(PadUint82dCustomPadding, PadUint82dCustomPaddingTest) +ARMNN_AUTO_TEST_CASE(PadUint83d, PadUint83dTest) +ARMNN_AUTO_TEST_CASE(PadUint84d, PadUint84dTest) + +ARMNN_AUTO_TEST_CASE(Pad2dQSymm16, Pad2dTestCommon<armnn::DataType::QuantisedSymm16>, 2.0f, 0, 0.0f) +ARMNN_AUTO_TEST_CASE(Pad2dQSymm16CustomPadding, Pad2dTestCommon<armnn::DataType::QuantisedSymm16>, 2.0f, 0, 1.0f) +ARMNN_AUTO_TEST_CASE(Pad3dQSymm16, Pad3dTestCommon<armnn::DataType::QuantisedSymm16>, 2.0f, 0) +ARMNN_AUTO_TEST_CASE(Pad4dQSymm16, Pad4dTestCommon<armnn::DataType::QuantisedSymm16>, 2.0f, 0) + // PReLU ARMNN_AUTO_TEST_CASE(PreluFloat32, PreluTest<armnn::DataType::Float32>) ARMNN_AUTO_TEST_CASE(PreluUint8, PreluTest<armnn::DataType::QuantisedAsymm8>) diff --git a/src/backends/cl/workloads/ClPadWorkload.cpp b/src/backends/cl/workloads/ClPadWorkload.cpp index 8fd62f2ff1..1c072baf58 100644 --- a/src/backends/cl/workloads/ClPadWorkload.cpp +++ b/src/backends/cl/workloads/ClPadWorkload.cpp @@ -31,7 +31,9 @@ ClPadWorkload::ClPadWorkload(const PadQueueDescriptor& descriptor, const Workloa arm_compute::PaddingList padList = static_cast<arm_compute::PaddingList>(reversed_PadList); - m_Layer.configure(&input, &output, padList); + arm_compute::PixelValue pixelValue = descriptor.m_Parameters.m_PadValue; + + m_Layer.configure(&input, &output, padList, pixelValue); } void ClPadWorkload::Execute() const |