This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
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
549bfb97ad
257464ef7b
commit
d8dd963441
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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("Squeeze
Grad
") \
squeeze_op_info = AkgRegOp("Squeeze") \
.fusion_type("OPAQUE") \
.input(0, "x") \
.output(0, "output") \
Write
Preview
Loading…
Cancel
Save