Searched refs:blockSizeData (Results 1 – 2 of 2) sorted by relevance
573 const int32_t* blockSizeData, in batchToSpacePrepare() argument590 NN_OPS_CHECK(batches % (blockSizeData[0] * blockSizeData[1]) == 0); in batchToSpacePrepare()592 output->dimensions = {batches / (blockSizeData[0] * blockSizeData[1]), in batchToSpacePrepare()593 height * blockSizeData[0], in batchToSpacePrepare()594 width * blockSizeData[1], in batchToSpacePrepare()603 const int32_t* blockSizeData, in spaceToBatchPrepare() argument631 NN_OPS_CHECK(paddedHeight % blockSizeData[0] == 0); in spaceToBatchPrepare()632 NN_OPS_CHECK(paddedWidth % blockSizeData[1] == 0); in spaceToBatchPrepare()635 output->dimensions = {batches * (blockSizeData[0] * blockSizeData[1]), in spaceToBatchPrepare()636 paddedHeight / blockSizeData[0], in spaceToBatchPrepare()[all …]
335 const int32_t* blockSizeData,340 const int32_t* blockSizeData,