| 59 | } |
| 60 | |
| 61 | void ConcatFloat32TwoInputsTest() |
| 62 | { |
| 63 | std::vector<int32_t> inputShape { 2, 2 }; |
| 64 | std::vector<int32_t> expectedOutputShape { 4, 2 }; |
| 65 | |
| 66 | std::vector<std::vector<float>> inputValues; |
| 67 | std::vector<float> inputValue1 { -127.f, -126.f, -1.f, 0.f }; |
| 68 | std::vector<float> inputValue2 { 1.f, 2.f, 126.f, 127.f }; |
| 69 | inputValues.push_back(inputValue1); |
| 70 | inputValues.push_back(inputValue2); |
| 71 | |
| 72 | std::vector<float> expectedOutputValues { -127.f, -126.f, -1.f, 0.f, 1.f, 2.f, 126.f, 127.f }; |
| 73 | |
| 74 | ConcatenationTest<float>(tflite::BuiltinOperator_CONCATENATION, |
| 75 | ::tflite::TensorType_FLOAT32, |
| 76 | inputShape, |
| 77 | expectedOutputShape, |
| 78 | inputValues, |
| 79 | expectedOutputValues); |
| 80 | } |
| 81 | |
| 82 | void ConcatThreeInputsTest() |
| 83 | { |
no test coverage detected