| 103 | } |
| 104 | |
| 105 | void ConcatAxisTest() |
| 106 | { |
| 107 | std::vector<int32_t> inputShape { 1, 2, 2 }; |
| 108 | std::vector<int32_t> expectedOutputShape { 1, 2, 4 }; |
| 109 | |
| 110 | std::vector<std::vector<uint8_t>> inputValues; |
| 111 | std::vector<uint8_t> inputValue1 { 0, 1, 2, 3 }; |
| 112 | std::vector<uint8_t> inputValue3 { 252, 253, 254, 255 }; |
| 113 | inputValues.push_back(inputValue1); |
| 114 | inputValues.push_back(inputValue3); |
| 115 | |
| 116 | std::vector<uint8_t> expectedOutputValues { 0, 1, 252, 253, 2, 3, 254, 255 }; |
| 117 | |
| 118 | ConcatenationTest<uint8_t>(tflite::BuiltinOperator_CONCATENATION, |
| 119 | ::tflite::TensorType_UINT8, |
| 120 | inputShape, |
| 121 | expectedOutputShape, |
| 122 | inputValues, |
| 123 | expectedOutputValues, |
| 124 | 2); |
| 125 | } |
| 126 | |
| 127 | // MEAN Operator |
| 128 | void MeanUint8KeepDimsTest() |
no test coverage detected