Searched refs:blockSizeData (Results 1 – 2 of 2) sorted by relevance
600 const int32_t* blockSizeData, in batchToSpacePrepare() argument617 NN_OPS_CHECK(batches % (blockSizeData[0] * blockSizeData[1]) == 0); in batchToSpacePrepare()619 output->dimensions = {batches / (blockSizeData[0] * blockSizeData[1]), in batchToSpacePrepare()620 height * blockSizeData[0], in batchToSpacePrepare()621 width * blockSizeData[1], in batchToSpacePrepare()630 const int32_t* blockSizeData, in spaceToBatchPrepare() argument658 NN_OPS_CHECK(paddedHeight % blockSizeData[0] == 0); in spaceToBatchPrepare()659 NN_OPS_CHECK(paddedWidth % blockSizeData[1] == 0); in spaceToBatchPrepare()662 output->dimensions = {batches * (blockSizeData[0] * blockSizeData[1]), in spaceToBatchPrepare()663 paddedHeight / blockSizeData[0], in spaceToBatchPrepare()[all …]
249 const int32_t* blockSizeData,254 const int32_t* blockSizeData,