diff options
author | Mohamed Nour Abouelseoud <mohamednour.abouelseoud@arm.com> | 2018-10-12 12:26:24 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
commit | 7420e55aefe545452639992ab1972fd355a9ed30 (patch) | |
tree | deae0ab925ca138e2c5f1130477d92f5d45c9de6 /src/backends/cl/workloads | |
parent | b5acbb77918df98debac200ebe082ce9aaab6a8c (diff) | |
download | armnn-7420e55aefe545452639992ab1972fd355a9ed30.tar.gz |
IVGCVSW-1885 add RefPadWorkload implementation and associated unit tests
* Added RefPadWorkload implementation
* Added unit tests and applied them to CL and Ref backends
* Fixed a bug in ClPadWorkload
Change-Id: I8cb76bc9d60ae8a39b08d40f05d628e3b72f6410
Diffstat (limited to 'src/backends/cl/workloads')
-rw-r--r-- | src/backends/cl/workloads/ClPadWorkload.cpp | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/src/backends/cl/workloads/ClPadWorkload.cpp b/src/backends/cl/workloads/ClPadWorkload.cpp index e75af83f44..3e63d5c210 100644 --- a/src/backends/cl/workloads/ClPadWorkload.cpp +++ b/src/backends/cl/workloads/ClPadWorkload.cpp @@ -22,7 +22,14 @@ ClPadWorkload::ClPadWorkload(const PadQueueDescriptor& descriptor, const Workloa arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(this->m_Data.m_Inputs[0])->GetTensor(); arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(this->m_Data.m_Outputs[0])->GetTensor(); - arm_compute::PaddingList padList = static_cast<arm_compute::PaddingList>(descriptor.m_Parameters.m_PadList); + + std::vector<std::pair<unsigned int, unsigned int>> reversed_PadList(descriptor.m_Parameters.m_PadList.size()); + + std::reverse_copy(std::begin(descriptor.m_Parameters.m_PadList), + std::end(descriptor.m_Parameters.m_PadList), + std::begin(reversed_PadList)); + + arm_compute::PaddingList padList = static_cast<arm_compute::PaddingList>(reversed_PadList); m_Layer.configure(&input, &output, padList); } |