| 73 | } |
| 74 | |
| 75 | void ArgMaxUint8Test(int axisValue) |
| 76 | { |
| 77 | // Set input data |
| 78 | std::vector<int32_t> inputShape { 1, 1, 1, 5 }; |
| 79 | std::vector<int32_t> outputShape { 1, 1, 1 }; |
| 80 | std::vector<int32_t> axisShape { 1 }; |
| 81 | |
| 82 | std::vector<uint8_t> inputValues = { 5, 2, 8, 10, 9 }; |
| 83 | |
| 84 | std::vector<int32_t> expectedOutputValues = { 3 }; |
| 85 | |
| 86 | ArgMinMaxTest<uint8_t, int32_t>(tflite::BuiltinOperator_ARG_MAX, |
| 87 | ::tflite::TensorType_UINT8, |
| 88 | inputShape, |
| 89 | axisShape, |
| 90 | outputShape, |
| 91 | inputValues, |
| 92 | expectedOutputValues, |
| 93 | axisValue, |
| 94 | ::tflite::TensorType_INT32); |
| 95 | } |
| 96 | |
| 97 | TEST_SUITE("ArgMinMax_Tests") |
| 98 | { |