aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/layerTests/StridedSliceTestImpl.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/StridedSliceTestImpl.cpp')
-rw-r--r--src/backends/backendsCommon/test/layerTests/StridedSliceTestImpl.cpp16
1 files changed, 8 insertions, 8 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/StridedSliceTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/StridedSliceTestImpl.cpp
index 66a3b14e3f..af4b089cde 100644
--- a/src/backends/backendsCommon/test/layerTests/StridedSliceTestImpl.cpp
+++ b/src/backends/backendsCommon/test/layerTests/StridedSliceTestImpl.cpp
@@ -40,12 +40,9 @@ LayerTestResult<T, OutDim> StridedSliceTestImpl(
outputTensorInfo.SetQuantizationOffset(qOffset);
}
- boost::multi_array<T, InDim> input =
- MakeTensor<T, InDim>(inputTensorInfo, armnnUtils::QuantizedVector<T>(inputData, qScale, qOffset));
-
- LayerTestResult<T, OutDim> ret(outputTensorInfo);
- ret.outputExpected =
- MakeTensor<T, OutDim>(outputTensorInfo, armnnUtils::QuantizedVector<T>(outputExpectedData, qScale, qOffset));
+ std::vector<T> input = armnnUtils::QuantizedVector<T>(inputData, qScale, qOffset);
+ std::vector<T> expectedOutput = armnnUtils::QuantizedVector<T>(outputExpectedData, qScale, qOffset);
+ std::vector<T> actualOutput(outputTensorInfo.GetNumElements());
std::unique_ptr<armnn::ITensorHandle> inputHandle =
tensorHandleFactory.CreateTensorHandle(inputTensorInfo);
@@ -66,9 +63,12 @@ LayerTestResult<T, OutDim> StridedSliceTestImpl(
ExecuteWorkload(*workload, memoryManager);
- CopyDataFromITensorHandle(ret.output.data(), outputHandle.get());
+ CopyDataFromITensorHandle(actualOutput.data(), outputHandle.get());
- return ret;
+ return LayerTestResult<T, OutDim>(actualOutput,
+ expectedOutput,
+ outputHandle->GetShape(),
+ outputTensorInfo.GetShape());
}
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>