| 192 | } |
| 193 | |
| 194 | void MeanFp32Test() |
| 195 | { |
| 196 | std::vector<int32_t> input0Shape { 1, 2, 2, 1 }; |
| 197 | std::vector<int32_t> input1Shape { 1 }; |
| 198 | std::vector<int32_t> expectedOutputShape { 1, 2, 1 }; |
| 199 | |
| 200 | std::vector<float> input0Values { 1.0f, 1.5f, 2.0f, 2.5f }; // Inputs |
| 201 | std::vector<int32_t> input1Values { 2 }; // Axis |
| 202 | |
| 203 | std::vector<float> expectedOutputValues { 1.25f, 2.25f }; |
| 204 | |
| 205 | MeanTest<float>(tflite::BuiltinOperator_MEAN, |
| 206 | ::tflite::TensorType_FLOAT32, |
| 207 | input0Shape, |
| 208 | input1Shape, |
| 209 | expectedOutputShape, |
| 210 | input0Values, |
| 211 | input1Values, |
| 212 | expectedOutputValues, |
| 213 | false); |
| 214 | } |
| 215 | |
| 216 | // CONCATENATION Tests. |
| 217 | TEST_SUITE("Concatenation_Tests") |