Searched refs:blockSizeData (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 529 bool batchToSpacePrepare(const Shape& input, const int32_t* blockSizeData, in batchToSpacePrepare() argument
545 NN_OPS_CHECK(batches % (blockSizeData[0] * blockSizeData[1]) == 0); in batchToSpacePrepare()
547 output->dimensions = {batches / (blockSizeData[0] * blockSizeData[1]), in batchToSpacePrepare()
548 height * blockSizeData[0], width * blockSizeData[1], channels}; in batchToSpacePrepare()
555 bool spaceToBatchPrepare(const Shape& input, const int32_t* blockSizeData, in spaceToBatchPrepare() argument
581 NN_OPS_CHECK(paddedHeight % blockSizeData[0] == 0); in spaceToBatchPrepare()
582 NN_OPS_CHECK(paddedWidth % blockSizeData[1] == 0); in spaceToBatchPrepare()
585 output->dimensions = {batches * (blockSizeData[0] * blockSizeData[1]), in spaceToBatchPrepare()
586 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,
|