p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!328 Modify akg register 

Merge pull request !328 from zjun/Modify_akg_register
This commit is contained in:
mindspore-ci-bot 2020-04-15 09:43:33 +08:00 committed by Gitee
parent dc20a1dcba 0099f0108e
commit c3373ede46
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ops/_op_impl/akg/gpu/hsigmoid.py
View File

@ -24,7 +24,7 @@ hsigmoid_op_info = AkgRegOp("HSigmoid") \
.get_op_info()
@op_info_register(hsigmoidgrad_op_info)
@op_info_register(hsigmoid_op_info)
def _hsigmoid_akg():
"""HSigmoid AutoDiff register"""
return

2
mindspore/ops/_op_impl/akg/gpu/hswish.py
View File

@ -24,7 +24,7 @@ hswish_op_info = AkgRegOp("HSwish") \
.get_op_info()
@op_info_register(hsigmoidgrad_op_info)
@op_info_register(hswish_op_info)
def _hswish_akg():
"""HSwish AutoDiff register"""
return

4
mindspore/ops/_op_impl/akg/gpu/hswish_grad.py
View File

@ -15,7 +15,7 @@
"""HSwishGrad op"""
from mindspore.ops.op_info_register import op_info_register, AkgRegOp, DataType
hswishgrad_op_info = AkgRegOp("HSwishGrad") \
hswish_grad_op_info = AkgRegOp("HSwishGrad") \
.fusion_type("OPAQUE") \
.input(0, "y_grad") \
.input(1, "x") \
@ -25,7 +25,7 @@ hswishgrad_op_info = AkgRegOp("HSwishGrad") \
.get_op_info()
@op_info_register(hsigmoidgrad_op_info)
@op_info_register(hswish_grad_op_info)
def _hswish_grad_akg():
"""HSwishGrad AutoDiff register"""
return