From 4e53cabfd8621252555156d66f2ede726228c8cf Mon Sep 17 00:00:00 2001 From: yanghaoran Date: Thu, 16 Jul 2020 21:29:16 +0800 Subject: [PATCH] synchronize latest Ascend Software suite 17 Jul 2020 --- graphengine | 2 +- mindspore/ops/_op_impl/tbe/avg_pool.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/graphengine b/graphengine index 31aa96ef41..eee707935c 160000 --- a/graphengine +++ b/graphengine @@ -1 +1 @@ -Subproject commit 31aa96ef41067a0ecdc4113ef245f8ede48f3457 +Subproject commit eee707935c066c16e9b9cd207f8125871b6b97cf diff --git a/mindspore/ops/_op_impl/tbe/avg_pool.py b/mindspore/ops/_op_impl/tbe/avg_pool.py index 90d174474b..09fa83cf1b 100644 --- a/mindspore/ops/_op_impl/tbe/avg_pool.py +++ b/mindspore/ops/_op_impl/tbe/avg_pool.py @@ -29,8 +29,10 @@ avg_pool_op_info = TBERegOp("AvgPool") \ .attr("data_format", "optional", "str", "all") \ .input(0, "x", False, "required", "all") \ .input(1, "filter", False, "optional", "all") \ + .input(2, "bias", False, "optional", "all") \ .output(0, "y", False, "required", "all") \ - .dtype_format(DataType.F16_5HD, DataType.F16_FracZ, DataType.F16_5HD) \ + .dtype_format(DataType.F16_5HD, DataType.F16_FracZ, DataType.F16_Default, DataType.F16_5HD) \ + .dtype_format(DataType.I8_5HD, DataType.I8_C1HWNCoC0, DataType.I32_Default, DataType.I32_5HD) \ .get_op_info()