diff --git a/mindspore/_akg/gpu/hswish.py b/mindspore/_akg/gpu/hswish.py index 44fcf109183899e4c8bb38f6adaf2199000c9d11..3def3c4b353eb0d19d0ccd430a4afd11bef99baf 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):