!13560 Add need_check_supported for MaxPoolWithArgmax and MaxPoolGradWithArgmax.

From: @liu_xiao_93
Reviewed-by: 
Signed-off-by:
pull/13560/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 686d3ca18d

@ -30,6 +30,7 @@ max_pool_grad_with_argmax_op_info = TBERegOp("MaxPoolGradWithArgmax") \
.input(1, "grad", False, "required", "all") \
.input(2, "argmax", False, "optional", "all") \
.output(0, "y", False, "required", "all") \
.need_check_supported(True) \
.dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.U16_5HD, DataType.F16_5HD) \
.dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.I64_5HD, DataType.F16_5HD) \
.get_op_info()

@ -29,6 +29,7 @@ max_pool_with_argmax_op_info = TBERegOp("MaxPoolWithArgmax") \
.input(0, "x", False, "required", "all") \
.output(0, "y", False, "required", "all") \
.output(1, "argmax", False, "required", "all") \
.need_check_supported(True) \
.dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.U16_5HD) \
.get_op_info()

Loading…
Cancel
Save