| 175 | } |
| 176 | |
| 177 | void ReshapeUint8Test(bool useOption = true, const std::vector<armnn::BackendId>& backends = {}) |
| 178 | { |
| 179 | // Set input data |
| 180 | std::vector<int32_t> inputShape { 1, 3, 4, 1 }; |
| 181 | std::vector<int32_t> outputShape { 6, 2 }; |
| 182 | std::vector<int32_t> targetShape { -1, 2 }; |
| 183 | |
| 184 | std::vector<uint8_t> inputValues = { 5, 8, 10, 7, |
| 185 | 8, 12, 15, 2, |
| 186 | 3, 4, 1, 11 }; |
| 187 | |
| 188 | std::vector<uint8_t> expectedOutputValues = { 5, 8, 10, 7, |
| 189 | 8, 12, 15, 2, |
| 190 | 3, 4, 1, 11 }; |
| 191 | |
| 192 | RedefineTest<uint8_t>(tflite::BuiltinOperator_RESHAPE, |
| 193 | ::tflite::TensorType_UINT8, |
| 194 | inputShape, |
| 195 | outputShape, |
| 196 | inputValues, |
| 197 | expectedOutputValues, |
| 198 | targetShape, |
| 199 | useOption, |
| 200 | backends, |
| 201 | 2.5f, |
| 202 | 1); |
| 203 | } |
| 204 | |
| 205 | void ReshapeInt16Test(bool useOption = true, const std::vector<armnn::BackendId>& backends = {}) |
| 206 | { |