| 147 | } |
| 148 | |
| 149 | void ReshapeInt8Test(bool useOption = true, const std::vector<armnn::BackendId>& backends = {}) |
| 150 | { |
| 151 | // Set input data |
| 152 | std::vector<int32_t> inputShape { 1, 3, 4, 1 }; |
| 153 | std::vector<int32_t> outputShape { 6, 2 }; |
| 154 | std::vector<int32_t> targetShape { -1, 2 }; |
| 155 | |
| 156 | std::vector<int8_t> inputValues = { -5, 8, -10, 7, |
| 157 | 8, 12, -15, 2, |
| 158 | 3, -4, -1, -11 }; |
| 159 | |
| 160 | std::vector<int8_t> expectedOutputValues = { -5, 8, -10, 7, |
| 161 | 8, 12, -15, 2, |
| 162 | 3, -4, -1, -11 }; |
| 163 | |
| 164 | RedefineTest<int8_t>(tflite::BuiltinOperator_RESHAPE, |
| 165 | ::tflite::TensorType_INT8, |
| 166 | inputShape, |
| 167 | outputShape, |
| 168 | inputValues, |
| 169 | expectedOutputValues, |
| 170 | targetShape, |
| 171 | useOption, |
| 172 | backends, |
| 173 | 2.5f, |
| 174 | 1); |
| 175 | } |
| 176 | |
| 177 | void ReshapeUint8Test(bool useOption = true, const std::vector<armnn::BackendId>& backends = {}) |
| 178 | { |