diff options
Diffstat (limited to 'src/backends/aclCommon/test/MemCopyTestImpl.hpp')
| -rw-r--r-- | src/backends/aclCommon/test/MemCopyTestImpl.hpp | 29 |
1 files changed, 13 insertions, 16 deletions
diff --git a/src/backends/aclCommon/test/MemCopyTestImpl.hpp b/src/backends/aclCommon/test/MemCopyTestImpl.hpp index 1f542d24b4..91ba4eae17 100644 --- a/src/backends/aclCommon/test/MemCopyTestImpl.hpp +++ b/src/backends/aclCommon/test/MemCopyTestImpl.hpp @@ -15,8 +15,6 @@ #include <test/TensorHelpers.hpp> -#include <boost/multi_array.hpp> - namespace { @@ -28,21 +26,20 @@ LayerTestResult<T, 4> MemCopyTest(armnn::IWorkloadFactory& srcWorkloadFactory, const std::array<unsigned int, 4> shapeData = { { 1u, 1u, 6u, 5u } }; const armnn::TensorShape tensorShape(4, shapeData.data()); const armnn::TensorInfo tensorInfo(tensorShape, dataType); - boost::multi_array<T, 4> inputData = MakeTensor<T, 4>(tensorInfo, std::vector<T>( - { - 1, 2, 3, 4, 5, - 6, 7, 8, 9, 10, - 11, 12, 13, 14, 15, - 16, 17, 18, 19, 20, - 21, 22, 23, 24, 25, - 26, 27, 28, 29, 30, - }) - ); + std::vector<T> inputData = + { + 1, 2, 3, 4, 5, + 6, 7, 8, 9, 10, + 11, 12, 13, 14, 15, + 16, 17, 18, 19, 20, + 21, 22, 23, 24, 25, + 26, 27, 28, 29, 30, + }; LayerTestResult<T, 4> ret(tensorInfo); - ret.outputExpected = inputData; + ret.m_ExpectedData = inputData; - boost::multi_array<T, 4> outputData(shapeData); + std::vector<T> actualOutput(tensorInfo.GetNumElements()); ARMNN_NO_DEPRECATE_WARN_BEGIN auto inputTensorHandle = srcWorkloadFactory.CreateTensorHandle(tensorInfo); @@ -71,8 +68,8 @@ LayerTestResult<T, 4> MemCopyTest(armnn::IWorkloadFactory& srcWorkloadFactory, dstWorkloadFactory.CreateMemCopy(memCopyQueueDesc, workloadInfo)->Execute(); - CopyDataFromITensorHandle(outputData.data(), workloadOutput.get()); - ret.output = outputData; + CopyDataFromITensorHandle(actualOutput.data(), workloadOutput.get()); + ret.m_ActualData = actualOutput; return ret; } |