diff --git a/mindspore/ops/_op_impl/tbe/avg_pool.py b/mindspore/ops/_op_impl/tbe/avg_pool.py
index 5db5947b01daf4e81e098718373662a2592477e6..90d174474b9b090e22a5a88349d9682b9d08b091 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()