| 129 | } |
| 130 | |
| 131 | void SpaceToBatchNDUint8Test() |
| 132 | { |
| 133 | std::vector<int32_t> inputShape { 1, 2, 2, 3 }; |
| 134 | std::vector<int32_t> expectedOutputShape { 4, 1, 1, 3 }; |
| 135 | |
| 136 | std::vector<uint8_t> inputValues { 1, 2, 3, 4, 5, 6, 7 }; |
| 137 | std::vector<uint8_t> expectedOutputValues { 1, 2, 3, 4, 5, 6, 7 }; |
| 138 | |
| 139 | std::vector<unsigned int> blockShape({2, 2}); |
| 140 | std::vector<std::pair<unsigned int, unsigned int>> padding = {{0, 0}, {0, 0}}; |
| 141 | |
| 142 | BatchSpaceTest<uint8_t>(tflite::BuiltinOperator_SPACE_TO_BATCH_ND, |
| 143 | ::tflite::TensorType_UINT8, |
| 144 | inputShape, |
| 145 | expectedOutputShape, |
| 146 | inputValues, |
| 147 | blockShape, |
| 148 | padding, |
| 149 | expectedOutputValues); |
| 150 | } |
| 151 | |
| 152 | // BatchToSpaceND Tests |
| 153 | TEST_SUITE("BatchToSpaceNDTests") |