| 37 | } |
| 38 | |
| 39 | void BatchToSpaceNDFp32BatchOneTest() |
| 40 | { |
| 41 | std::vector<int32_t> inputShape { 1, 2, 2, 1 }; |
| 42 | std::vector<int32_t> expectedOutputShape { 1, 2, 2, 1 }; |
| 43 | |
| 44 | std::vector<float> inputValues { 1.0f, 2.0f, 3.0f, 4.0f }; |
| 45 | std::vector<float> expectedOutputValues { 1.0f, 2.0f, 3.0f, 4.0f }; |
| 46 | |
| 47 | std::vector<unsigned int> blockShape({1, 1}); |
| 48 | std::vector<std::pair<unsigned int, unsigned int>> crops = {{0, 0}, {0, 0}}; |
| 49 | |
| 50 | BatchSpaceTest<float>(tflite::BuiltinOperator_BATCH_TO_SPACE_ND, |
| 51 | ::tflite::TensorType_FLOAT32, |
| 52 | inputShape, |
| 53 | expectedOutputShape, |
| 54 | inputValues, |
| 55 | blockShape, |
| 56 | crops, |
| 57 | expectedOutputValues); |
| 58 | } |
| 59 | |
| 60 | void BatchToSpaceNDUint8Test() |
| 61 | { |