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
enable GraphKernel for layernorm and layernormGrad (sync from r1.1)
tags/v1.2.0-rc1
hanhuifeng2020
4 years ago
parent
24ef3efcb8
commit
c5f261d894
1 changed files
with
2 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+2
-0
mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc
+ 2
- 0
mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_helper.cc
View File
@@ -718,6 +718,8 @@ std::unordered_set<PrimitivePtr> GetExpandOps() {
prim::kPrimGkDropout,
prim::kPrimDropoutGrad,
prim::kPrimSoftMax,
prim::kPrimLayerNorm,
prim::kPrimLayerNormGrad,
#endif
};
return expand_ops;
Write
Preview
Loading…
Cancel
Save