aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorÉanna Ó Catháin <eanna.ocathain@arm.com>2019-01-31 10:28:20 +0000
committerEanna O Cathain Arm <eanna.ocathain@arm.com>2019-01-31 16:17:01 +0000
commit4a55201f9d24d6bed46781ec56ba8862ddc7c26b (patch)
treee00f3dbc9cf2d4a476eaa80e2873ed4be750753b
parent93f5f97e40070cad5beb7ccf9c959684d0e93684 (diff)
downloadarmnn-4a55201f9d24d6bed46781ec56ba8862ddc7c26b.tar.gz
MLCE-90 Fixing issues with NEON pad tests
Change-Id: I3e971c66798610d2c96445fa3babb7cc90cf8031 Signed-off-by: Éanna Ó Catháin <eanna.ocathain@arm.com>
-rw-r--r--src/backends/neon/workloads/NeonPadWorkload.cpp9
1 files changed, 8 insertions, 1 deletions
diff --git a/src/backends/neon/workloads/NeonPadWorkload.cpp b/src/backends/neon/workloads/NeonPadWorkload.cpp
index 60d6b8a537..79645f148d 100644
--- a/src/backends/neon/workloads/NeonPadWorkload.cpp
+++ b/src/backends/neon/workloads/NeonPadWorkload.cpp
@@ -49,7 +49,14 @@ arm_compute::Status NeonPadWorkloadValidate(const TensorInfo& input,
{
const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input);
const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output);
- arm_compute::PaddingList padList = static_cast<arm_compute::PaddingList>(descriptor.m_PadList);
+
+ std::vector<std::pair<unsigned int, unsigned int>> reversed_PadList(descriptor.m_PadList.size());
+
+ std::reverse_copy(std::begin(descriptor.m_PadList),
+ std::end(descriptor.m_PadList),
+ std::begin(reversed_PadList));
+
+ arm_compute::PaddingList padList = static_cast<arm_compute::PaddingList>(reversed_PadList);
return arm_compute::NEPadLayer::validate(&aclInputInfo, &aclOutputInfo, padList);
}