| 38 | } |
| 39 | |
| 40 | void ConcatInt16TwoInputsTest() |
| 41 | { |
| 42 | std::vector<int32_t> inputShape { 2, 2 }; |
| 43 | std::vector<int32_t> expectedOutputShape { 4, 2 }; |
| 44 | |
| 45 | std::vector<std::vector<int16_t>> inputValues; |
| 46 | std::vector<int16_t> inputValue1 { -32768, -16384, -1, 0 }; |
| 47 | std::vector<int16_t> inputValue2 { 1, 2, 16384, 32767 }; |
| 48 | inputValues.push_back(inputValue1); |
| 49 | inputValues.push_back(inputValue2); |
| 50 | |
| 51 | std::vector<int16_t> expectedOutputValues { -32768, -16384, -1, 0, 1, 2, 16384, 32767}; |
| 52 | |
| 53 | ConcatenationTest<int16_t>(tflite::BuiltinOperator_CONCATENATION, |
| 54 | ::tflite::TensorType_INT16, |
| 55 | inputShape, |
| 56 | expectedOutputShape, |
| 57 | inputValues, |
| 58 | expectedOutputValues); |
| 59 | } |
| 60 | |
| 61 | void ConcatFloat32TwoInputsTest() |
| 62 | { |
no test coverage detected