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

!388 fix gpu op register 

Merge pull request !388 from Maoweiyong/fix-gpu-op-register
This commit is contained in:
mindspore-ci-bot 2020-04-16 19:50:03 +08:00 committed by Gitee
parent 549bfb97ad 257464ef7b
commit d8dd963441
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,7 +15,7 @@
"""Squeeze op"""
from mindspore.ops.op_info_register import op_info_register, AkgRegOp, DataType
squeeze_op_info = AkgRegOp("SqueezeGrad") \
squeeze_op_info = AkgRegOp("Squeeze") \
.fusion_type("OPAQUE") \
.input(0, "x") \
.output(0, "output") \