From 86efd3368294f6a85a958de0a47d0d2cb75e1527 Mon Sep 17 00:00:00 2001 From: yanghaoran Date: Fri, 17 Jul 2020 18:18:33 +0800 Subject: [PATCH] fix avg_pool operator, adding filter input --- mindspore/ops/_op_impl/tbe/avg_pool.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/ops/_op_impl/tbe/avg_pool.py b/mindspore/ops/_op_impl/tbe/avg_pool.py index 5db5947b01d..90d174474b9 100644 --- a/mindspore/ops/_op_impl/tbe/avg_pool.py +++ b/mindspore/ops/_op_impl/tbe/avg_pool.py @@ -28,8 +28,9 @@ avg_pool_op_info = TBERegOp("AvgPool") \ .attr("padding", "required", "str", "all") \ .attr("data_format", "optional", "str", "all") \ .input(0, "x", False, "required", "all") \ + .input(1, "filter", False, "optional", "all") \ .output(0, "y", False, "required", "all") \ - .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \ + .dtype_format(DataType.F16_5HD, DataType.F16_FracZ, DataType.F16_5HD) \ .get_op_info()