Searched refs:OpLevel_0 (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 44 using context::OpLevel_0;
53 {kAllTarget, OpLevel_0, prim::kPrimAddN}, in GetExpandOps()
54 {kAllTarget, OpLevel_0, prim::kPrimAssignAdd}, in GetExpandOps()
55 {kAllTarget, OpLevel_0, prim::kPrimErfc}, in GetExpandOps()
57 {kAllTarget, OpLevel_0, prim::kPrimGeLU}, in GetExpandOps()
58 {kAllTarget, OpLevel_0, prim::kPrimGeLUGrad}, in GetExpandOps()
59 {kAllTarget, OpLevel_0, prim::kPrimSquare}, in GetExpandOps()
60 {kAllTarget, OpLevel_0, prim::kPrimTile}, in GetExpandOps()
61 {kAscendDevice, OpLevel_0, prim::kLambApplyOptimizerAssign}, in GetExpandOps()
62 {kAscendDevice, OpLevel_0, prim::kLambApplyWeightAssign}, in GetExpandOps()
[all …]
|
| D | graph_kernel_cluster.cc | 30 using context::OpLevel_0;
35 {kAllTarget, OpLevel_0, prim::kPrimAbs}, in GetClusterableOpList()
36 {kAllTarget, OpLevel_0, prim::kPrimAdd}, in GetClusterableOpList()
37 {kAllTarget, OpLevel_0, prim::kPrimCast}, in GetClusterableOpList()
38 {kAllTarget, OpLevel_0, prim::kPrimEqual}, in GetClusterableOpList()
39 {kAllTarget, OpLevel_0, prim::kPrimExp}, in GetClusterableOpList()
40 {kAllTarget, OpLevel_0, prim::kPrimInplaceAssign}, in GetClusterableOpList()
41 {kAllTarget, OpLevel_0, prim::kPrimLog}, in GetClusterableOpList()
42 {kAllTarget, OpLevel_0, prim::kPrimMaximum}, in GetClusterableOpList()
43 {kAllTarget, OpLevel_0, prim::kPrimMinimum}, in GetClusterableOpList()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/utils/context/ |
| D | graph_kernel_flags.h | 35 constexpr unsigned int OpLevel_0 = 0; variable
96 unsigned int fusion_ops_level{OpLevel_0};
|
| D | graph_kernel_flags.cc | 195 reg.AddFlag("fusion_ops_level", &fusion_ops_level, is_gpu ? OpLevel_MAX : OpLevel_0); in RegisterFlags()
|