diff options
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/TransposeConvolution2dTestImpl.cpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/TransposeConvolution2dTestImpl.cpp | 24 |
1 files changed, 13 insertions, 11 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/TransposeConvolution2dTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/TransposeConvolution2dTestImpl.cpp index 85ce7e5e6f..cd775729cd 100644 --- a/src/backends/backendsCommon/test/layerTests/TransposeConvolution2dTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/TransposeConvolution2dTestImpl.cpp @@ -183,8 +183,8 @@ LayerTestResult<T, 4> TransposeConvolution2dTest( // execute test TransposeConvolution2dTestImpl(workloadFactory, - memoryManager, - tensorHandleFactory, + memoryManager, + tensorHandleFactory, descriptor, input, output, @@ -193,11 +193,10 @@ LayerTestResult<T, 4> TransposeConvolution2dTest( // construct result object LayerTestResult<T, 4> testResult(outputInfo); - testResult.output = MakeTensor<T, 4>(outputInfo, output.second); - testResult.outputExpected = MakeTensor<T, 4>(outputInfo, - armnnUtils::QuantizedVector<T>(expectedOutputData, - outputInfo.GetQuantizationScale(), - outputInfo.GetQuantizationOffset())); + testResult.m_ActualData = output.second; + testResult.m_ExpectedData = armnnUtils::QuantizedVector<T>(expectedOutputData, + outputInfo.GetQuantizationScale(), + outputInfo.GetQuantizationOffset()); return testResult; } @@ -611,6 +610,8 @@ LayerTestResult<uint8_t, 4> TransposeConvolution2dPerAxisQuantTest( std::vector<int32_t> biasData = { -12, -8 }; + std::vector<uint8_t> actualOutput(outputInfo.GetNumElements()); + std::vector<uint8_t> expectedOutputData = { 9, 13, 21, 19, 27, @@ -665,11 +666,12 @@ LayerTestResult<uint8_t, 4> TransposeConvolution2dPerAxisQuantTest( ExecuteWorkload(*workload, memoryManager); - LayerTestResult<uint8_t, 4> ret(outputInfo); - CopyDataFromITensorHandle(ret.output.origin(), outputHandle.get()); - ret.outputExpected = MakeTensor<uint8_t, 4>(outputInfo, expectedOutputData); + CopyDataFromITensorHandle(actualOutput.data(), outputHandle.get()); - return ret; + return LayerTestResult<uint8_t, 4>(actualOutput, + expectedOutputData, + outputHandle->GetShape(), + outputInfo.GetShape()); } // |