Narumol Prangnawarat | 4dc64a6 | 2019-09-16 17:00:22 +0100 | [diff] [blame] | 1 | // |
| 2 | // Copyright © 2019 Arm Ltd. All rights reserved. |
| 3 | // SPDX-License-Identifier: MIT |
| 4 | // |
| 5 | |
| 6 | #include <reference/workloads/ArgMinMax.hpp> |
| 7 | |
| 8 | #include <boost/test/unit_test.hpp> |
| 9 | |
| 10 | BOOST_AUTO_TEST_SUITE(RefArgMinMax) |
| 11 | |
| 12 | BOOST_AUTO_TEST_CASE(ArgMinTest) |
| 13 | { |
| 14 | const armnn::TensorInfo inputInfo({ 1, 2, 3 } , armnn::DataType::Float32); |
| 15 | const armnn::TensorInfo outputInfo({ 1, 3 }, armnn::DataType::Float32); |
| 16 | |
| 17 | std::vector<float> inputValues({ 1.0f, 5.0f, 3.0f, 4.0f, 2.0f, 6.0f}); |
| 18 | std::vector<int32_t> outputValues(outputInfo.GetNumElements()); |
| 19 | std::vector<int32_t> expectedValues({ 0, 1, 0 }); |
| 20 | |
| 21 | ArgMinMax(*armnn::MakeDecoder<float>(inputInfo, inputValues.data()), |
| 22 | outputValues.data(), |
| 23 | inputInfo, |
| 24 | outputInfo, |
| 25 | armnn::ArgMinMaxFunction::Min, |
| 26 | -2); |
| 27 | |
| 28 | BOOST_CHECK_EQUAL_COLLECTIONS(outputValues.begin(), |
| 29 | outputValues.end(), |
| 30 | expectedValues.begin(), |
| 31 | expectedValues.end()); |
| 32 | |
| 33 | } |
| 34 | |
| 35 | BOOST_AUTO_TEST_CASE(ArgMaxTest) |
| 36 | { |
| 37 | const armnn::TensorInfo inputInfo({ 1, 2, 3 } , armnn::DataType::Float32); |
| 38 | const armnn::TensorInfo outputInfo({ 1, 3 }, armnn::DataType::Float32); |
| 39 | |
| 40 | std::vector<float> inputValues({ 1.0f, 5.0f, 3.0f, 4.0f, 2.0f, 6.0f }); |
| 41 | std::vector<int32_t> outputValues(outputInfo.GetNumElements()); |
| 42 | std::vector<int32_t> expectedValues({ 1, 0, 1 }); |
| 43 | |
| 44 | ArgMinMax(*armnn::MakeDecoder<float>(inputInfo, inputValues.data()), |
| 45 | outputValues.data(), |
| 46 | inputInfo, |
| 47 | outputInfo, |
| 48 | armnn::ArgMinMaxFunction::Max, |
| 49 | -2); |
| 50 | |
| 51 | BOOST_CHECK_EQUAL_COLLECTIONS(outputValues.begin(), |
| 52 | outputValues.end(), |
| 53 | expectedValues.begin(), |
| 54 | expectedValues.end()); |
| 55 | |
| 56 | } |
| 57 | |
| 58 | BOOST_AUTO_TEST_SUITE_END() |