Home
last modified time | relevance | path

Searched refs:blockSizeShape (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp437 const Shape& blockSizeShape, Shape* output) { in batchToSpacePrepare() argument
442 NN_OPS_CHECK(blockSizeShape.type == OperandType::TENSOR_INT32); in batchToSpacePrepare()
443 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in batchToSpacePrepare()
445 NN_OPS_CHECK(getSizeOfDimension(blockSizeShape, 0) == 2); in batchToSpacePrepare()
463 const Shape& blockSizeShape, const int32_t* paddingsData, in spaceToBatchPrepare() argument
469 NN_OPS_CHECK(blockSizeShape.type == OperandType::TENSOR_INT32); in spaceToBatchPrepare()
470 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in spaceToBatchPrepare()
472 NN_OPS_CHECK(getSizeOfDimension(blockSizeShape, 0) == 2); in spaceToBatchPrepare()
/packages/modules/NeuralNetworks/common/include/
DOperationsExecutionUtils.h248 const Shape& blockSizeShape, Shape* output);
251 const Shape& blockSizeShape, const int32_t* paddingsData,