Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/
DOperationsUtils.cpp528 const Shape& blockSizeShape, Shape* output) { in batchToSpacePrepare() argument
533 NN_OPS_CHECK(blockSizeShape.type == OperandType::TENSOR_INT32); in batchToSpacePrepare()
534 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in batchToSpacePrepare()
536 NN_OPS_CHECK(getSizeOfDimension(blockSizeShape, 0) == 2); in batchToSpacePrepare()
554 const Shape& blockSizeShape, const int32_t* paddingsData, in spaceToBatchPrepare() argument
560 NN_OPS_CHECK(blockSizeShape.type == OperandType::TENSOR_INT32); in spaceToBatchPrepare()
561 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in spaceToBatchPrepare()
563 NN_OPS_CHECK(getSizeOfDimension(blockSizeShape, 0) == 2); in spaceToBatchPrepare()
/packages/modules/NeuralNetworks/common/include/
DOperationsUtils.h306 const Shape& blockSizeShape, Shape* output);
309 const Shape& blockSizeShape, const int32_t* paddingsData,