8 #include <doctest/doctest.h> 12 TEST_CASE(
"ArgMinTest")
17 std::vector<float> inputValues({ 1.0f, 5.0f, 3.0f, 4.0f, 2.0f, 6.0f});
18 std::vector<int64_t> outputValues(outputInfo.GetNumElements());
19 std::vector<int64_t> expectedValues({ 0, 1, 0 });
21 ArgMinMax(*armnn::MakeDecoder<float>(inputInfo, inputValues.data()),
28 CHECK(std::equal(outputValues.begin(), outputValues.end(), expectedValues.begin(), expectedValues.end()));
32 TEST_CASE(
"ArgMaxTest")
37 std::vector<float> inputValues({ 1.0f, 5.0f, 3.0f, 4.0f, 2.0f, 6.0f });
38 std::vector<int64_t> outputValues(outputInfo.GetNumElements());
39 std::vector<int64_t> expectedValues({ 1, 0, 1 });
41 ArgMinMax(*armnn::MakeDecoder<float>(inputInfo, inputValues.data()),
48 CHECK(std::equal(outputValues.begin(), outputValues.end(), expectedValues.begin(), expectedValues.end()));
void ArgMinMax(Decoder< float > &in, OUT *out, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, ArgMinMaxFunction function, int axis)
TEST_SUITE("RefArgMinMax")