Searched refs:blockSizeData (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsExecutionUtils.cpp | 436 bool batchToSpacePrepare(const Shape& input, const int32_t* blockSizeData, in batchToSpacePrepare() argument
452 NN_OPS_CHECK(batches % (blockSizeData[0] * blockSizeData[1]) == 0); in batchToSpacePrepare()
454 output->dimensions = {batches / (blockSizeData[0] * blockSizeData[1]), in batchToSpacePrepare()
455 height * blockSizeData[0], width * blockSizeData[1], channels}; in batchToSpacePrepare()
462 bool spaceToBatchPrepare(const Shape& input, const int32_t* blockSizeData, in spaceToBatchPrepare() argument
488 NN_OPS_CHECK(paddedHeight % blockSizeData[0] == 0); in spaceToBatchPrepare()
489 NN_OPS_CHECK(paddedWidth % blockSizeData[1] == 0); in spaceToBatchPrepare()
492 output->dimensions = {batches * (blockSizeData[0] * blockSizeData[1]), in spaceToBatchPrepare()
493 paddedHeight / blockSizeData[0], paddedWidth / blockSizeData[1], in spaceToBatchPrepare()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsExecutionUtils.h | 247 bool batchToSpacePrepare(const Shape& input, const int32_t* blockSizeData,
250 bool spaceToBatchPrepare(const Shape& input, const int32_t* blockSizeData,
|