333     uint32_t batches = getSizeOfDimension(input, 0);  in depthToSpacePrepare()  local
340     output->dimensions = {batches, height * blockSize, width * blockSize,  in depthToSpacePrepare()
352     uint32_t batches = getSizeOfDimension(input, 0);  in spaceToDepthPrepare()  local
361     output->dimensions = {batches, height / blockSize, width / blockSize,  in spaceToDepthPrepare()
447     uint32_t batches = getSizeOfDimension(input, 0);  in batchToSpacePrepare()  local
452     NN_OPS_CHECK(batches % (blockSizeData[0] * blockSizeData[1]) == 0);  in batchToSpacePrepare()
454     output->dimensions = {batches / (blockSizeData[0] * blockSizeData[1]),  in batchToSpacePrepare()
480     uint32_t batches = getSizeOfDimension(input, 0);  in spaceToBatchPrepare()  local
492     output->dimensions = {batches * (blockSizeData[0] * blockSizeData[1]),  in spaceToBatchPrepare()
646     uint32_t batches = getSizeOfDimension(input, 0);  in groupedConvPrepare()  local
659     output->dimensions = {batches, outHeight, outWidth, channels_out};  in groupedConvPrepare()