提交 d8dd9634 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!388 fix gpu op register

Merge pull request !388 from Maoweiyong/fix-gpu-op-register
......@@ -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") \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册