From 189f88fbe18382bb040bd402e2a2106227ca4967 Mon Sep 17 00:00:00 2001 From: chenzomi Date: Tue, 12 May 2020 11:29:49 +0800 Subject: [PATCH] change the hswish to HSwish --- mindspore/_akg/gpu/hswish.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mindspore/_akg/gpu/hswish.py b/mindspore/_akg/gpu/hswish.py index 44fcf1091..3def3c4b3 100644 --- a/mindspore/_akg/gpu/hswish.py +++ b/mindspore/_akg/gpu/hswish.py @@ -12,16 +12,16 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""hswish""" +"""HSwish""" import _akg.topi as topi import _akg.tvm as tvm from _akg.topi import tag @tvm.tag_scope(tag=tag.ELEMWISE) -def topi_nn_hswish(x): +def topi_nn_HSwish(x): """ - topi hswish + topi HSwish Args: x: @@ -42,7 +42,7 @@ def HSwish(x): Returns: """ - return topi_nn_hswish(x) + return topi_nn_HSwish(x) def gpu_schedule_HSwish(outs): -- GitLab