| 83 | } |
| 84 | |
| 85 | void ReduceFp32Test(tflite::BuiltinOperator reduceOperatorCode, |
| 86 | std::vector<float>& expectedOutputValues, |
| 87 | const std::vector<armnn::BackendId>& backends = {}) |
| 88 | { |
| 89 | std::vector<int32_t> input0Shape { 1, 1, 2, 3 }; |
| 90 | std::vector<int32_t> input1Shape { 1 }; |
| 91 | std::vector<int32_t> expectedOutputShape { 1, 1, 3 }; |
| 92 | |
| 93 | std::vector<float> input0Values { 1001.0f, 11.0f, 1003.0f, |
| 94 | 10.0f, 1002.0f, 12.0f }; // Inputs |
| 95 | std::vector<int32_t> input1Values { 2 }; // Axis |
| 96 | |
| 97 | ReduceTest<float>(reduceOperatorCode, |
| 98 | ::tflite::TensorType_FLOAT32, |
| 99 | input0Shape, |
| 100 | input1Shape, |
| 101 | expectedOutputShape, |
| 102 | input0Values, |
| 103 | input1Values, |
| 104 | expectedOutputValues, |
| 105 | false, |
| 106 | backends); |
| 107 | } |
| 108 | |
| 109 | // REDUCE_MAX Tests |
| 110 | TEST_SUITE("ReduceMaxTests") |