Searched refs:blockSizeShape (Results 1 – 2 of 2) sorted by relevance
528 const Shape& blockSizeShape, Shape* output) { in batchToSpacePrepare() argument533 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() argument560 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()
306 const Shape& blockSizeShape, Shape* output);309 const Shape& blockSizeShape, const int32_t* paddingsData,