diff options
Diffstat (limited to 'src/backends/reference/test/ArgMinMaxTests.cpp')
| -rw-r--r-- | src/backends/reference/test/ArgMinMaxTests.cpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/src/backends/reference/test/ArgMinMaxTests.cpp b/src/backends/reference/test/ArgMinMaxTests.cpp index 201a2c0c2e..dce15b29ef 100644 --- a/src/backends/reference/test/ArgMinMaxTests.cpp +++ b/src/backends/reference/test/ArgMinMaxTests.cpp @@ -12,11 +12,11 @@ BOOST_AUTO_TEST_SUITE(RefArgMinMax) BOOST_AUTO_TEST_CASE(ArgMinTest) { const armnn::TensorInfo inputInfo({ 1, 2, 3 } , armnn::DataType::Float32); - const armnn::TensorInfo outputInfo({ 1, 3 }, armnn::DataType::Float32); + const armnn::TensorInfo outputInfo({ 1, 3 }, armnn::DataType::Signed64); std::vector<float> inputValues({ 1.0f, 5.0f, 3.0f, 4.0f, 2.0f, 6.0f}); - std::vector<int32_t> outputValues(outputInfo.GetNumElements()); - std::vector<int32_t> expectedValues({ 0, 1, 0 }); + std::vector<int64_t> outputValues(outputInfo.GetNumElements()); + std::vector<int64_t> expectedValues({ 0, 1, 0 }); ArgMinMax(*armnn::MakeDecoder<float>(inputInfo, inputValues.data()), outputValues.data(), @@ -35,11 +35,11 @@ BOOST_AUTO_TEST_CASE(ArgMinTest) BOOST_AUTO_TEST_CASE(ArgMaxTest) { const armnn::TensorInfo inputInfo({ 1, 2, 3 } , armnn::DataType::Float32); - const armnn::TensorInfo outputInfo({ 1, 3 }, armnn::DataType::Float32); + const armnn::TensorInfo outputInfo({ 1, 3 }, armnn::DataType::Signed64); std::vector<float> inputValues({ 1.0f, 5.0f, 3.0f, 4.0f, 2.0f, 6.0f }); - std::vector<int32_t> outputValues(outputInfo.GetNumElements()); - std::vector<int32_t> expectedValues({ 1, 0, 1 }); + std::vector<int64_t> outputValues(outputInfo.GetNumElements()); + std::vector<int64_t> expectedValues({ 1, 0, 1 }); ArgMinMax(*armnn::MakeDecoder<float>(inputInfo, inputValues.data()), outputValues.data(), |