Searched refs:kPrimAssignSub (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | fused_batch_norm_fusion.h | 39 …assign_sub0_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAssignSub->name())… in PatternProcessPass()
40 …assign_sub1_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAssignSub->name())… in PatternProcessPass()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | branch_culling.cc | 73 {prim::kPrimAssignSub, {1}}, in InConvertWhiteList()
92 {prim::kPrimAssignSub, {1}}, {prim::kPrimApplyRMSProp, {6, 7, 8}}, in InConvertWhiteList()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 203 {prim::kPrimAssignSub, std::make_shared<OpUMonadExpander>(kAssignInputIdx)}, in GetExpander()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 541 inline const PrimitivePtr kPrimAssignSub = std::make_shared<Primitive>("AssignSub"); variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | kernel_graph.cc | 39 prim::kPrimAssignSub->name()};
|