diff options
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/ResizeTestImpl.cpp')
| -rw-r--r-- | src/backends/backendsCommon/test/layerTests/ResizeTestImpl.cpp | 23 |
1 files changed, 11 insertions, 12 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/ResizeTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/ResizeTestImpl.cpp index a2a804d54e..7706bde60d 100644 --- a/src/backends/backendsCommon/test/layerTests/ResizeTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/ResizeTestImpl.cpp @@ -95,10 +95,10 @@ LayerTestResult<T, NumDims> ResizeTestImpl( std::vector<T> inputData = armnnUtils::QuantizedVector<T>(params.m_InputData, params.m_InQuantScale, params.m_InQuantOffset); - std::vector<T> expectedOutputData = - armnnUtils::QuantizedVector<T>(params.m_ExpectedOutputData, - params.m_OutQuantScale, - params.m_OutQuantOffset); + std::vector<T> actualOutput(outputInfo.GetNumElements()); + std::vector<T> expectedOutputData = armnnUtils::QuantizedVector<T>(params.m_ExpectedOutputData, + params.m_OutQuantScale, + params.m_OutQuantOffset); if (params.m_DataLayout == armnn::DataLayout::NHWC) { @@ -106,11 +106,6 @@ LayerTestResult<T, NumDims> ResizeTestImpl( PermuteTensorNchwToNhwc(outputInfo, expectedOutputData); } - auto input = MakeTensor<T, NumDims>(inputInfo, inputData); - - LayerTestResult<T, NumDims> result(outputInfo); - result.outputExpected = MakeTensor<T, NumDims>(outputInfo, expectedOutputData); - std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputInfo); std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputInfo); @@ -132,13 +127,17 @@ LayerTestResult<T, NumDims> ResizeTestImpl( inputHandle->Allocate(); outputHandle->Allocate(); - CopyDataToITensorHandle(inputHandle.get(), input.origin()); + CopyDataToITensorHandle(inputHandle.get(), inputData.data()); workload->PostAllocationConfigure(); workload->Execute(); - CopyDataFromITensorHandle(result.output.origin(), outputHandle.get()); - return result; + CopyDataFromITensorHandle(actualOutput.data(), outputHandle.get()); + + return LayerTestResult<T, NumDims>(actualOutput, + expectedOutputData, + outputHandle->GetShape(), + outputInfo.GetShape()); } } // anonymous namespace |