Searched refs:blockSizeShape (Results 1 – 2 of 2) sorted by relevance
534 const Shape& blockSizeShape, Shape* output) { in batchToSpacePrepare() argument539 NN_OPS_CHECK(blockSizeShape.type == OperandType::TENSOR_INT32); in batchToSpacePrepare()540 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in batchToSpacePrepare()542 NN_OPS_CHECK(getSizeOfDimension(blockSizeShape, 0) == 2); in batchToSpacePrepare()560 const Shape& blockSizeShape, const int32_t* paddingsData, in spaceToBatchPrepare() argument566 NN_OPS_CHECK(blockSizeShape.type == OperandType::TENSOR_INT32); in spaceToBatchPrepare()567 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in spaceToBatchPrepare()569 NN_OPS_CHECK(getSizeOfDimension(blockSizeShape, 0) == 2); in spaceToBatchPrepare()
322 const Shape& blockSizeShape, Shape* output);325 const Shape& blockSizeShape, const int32_t* paddingsData,