Browse Source

!388 fix gpu op register

Merge pull request !388 from Maoweiyong/fix-gpu-op-register
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
d8dd963441
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/_op_impl/akg/gpu/squeeze.py

+ 1
- 1
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") \


Loading…
Cancel
Save