Searched refs:blockSizeData (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsUtils.cpp | 527 bool batchToSpacePrepare(const Shape& input, const int32_t* blockSizeData, in batchToSpacePrepare() argument
543 NN_OPS_CHECK(batches % (blockSizeData[0] * blockSizeData[1]) == 0); in batchToSpacePrepare()
545 output->dimensions = {batches / (blockSizeData[0] * blockSizeData[1]), in batchToSpacePrepare()
546 height * blockSizeData[0], width * blockSizeData[1], channels}; in batchToSpacePrepare()
553 bool spaceToBatchPrepare(const Shape& input, const int32_t* blockSizeData, in spaceToBatchPrepare() argument
579 NN_OPS_CHECK(paddedHeight % blockSizeData[0] == 0); in spaceToBatchPrepare()
580 NN_OPS_CHECK(paddedWidth % blockSizeData[1] == 0); in spaceToBatchPrepare()
583 output->dimensions = {batches * (blockSizeData[0] * blockSizeData[1]), in spaceToBatchPrepare()
584 paddedHeight / blockSizeData[0], paddedWidth / blockSizeData[1], in spaceToBatchPrepare()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsUtils.h | 305 bool batchToSpacePrepare(const Shape& input, const int32_t* blockSizeData,
308 bool spaceToBatchPrepare(const Shape& input, const int32_t* blockSizeData,
|