Searched refs:blockSizeData (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 533 bool batchToSpacePrepare(const Shape& input, const int32_t* blockSizeData, in batchToSpacePrepare() argument
549 NN_OPS_CHECK(batches % (blockSizeData[0] * blockSizeData[1]) == 0); in batchToSpacePrepare()
551 output->dimensions = {batches / (blockSizeData[0] * blockSizeData[1]), in batchToSpacePrepare()
552 height * blockSizeData[0], width * blockSizeData[1], channels}; in batchToSpacePrepare()
559 bool spaceToBatchPrepare(const Shape& input, const int32_t* blockSizeData, in spaceToBatchPrepare() argument
585 NN_OPS_CHECK(paddedHeight % blockSizeData[0] == 0); in spaceToBatchPrepare()
586 NN_OPS_CHECK(paddedWidth % blockSizeData[1] == 0); in spaceToBatchPrepare()
589 output->dimensions = {batches * (blockSizeData[0] * blockSizeData[1]), in spaceToBatchPrepare()
590 paddedHeight / blockSizeData[0], paddedWidth / blockSizeData[1], in spaceToBatchPrepare()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 321 bool batchToSpacePrepare(const Shape& input, const int32_t* blockSizeData,
324 bool spaceToBatchPrepare(const Shape& input, const int32_t* blockSizeData,
|