| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | adam_apply_one_fusion.cc | 111 VectorRef assign0 = VectorRef({prim::kPrimAssign, input_vars_[kIndex3], sub0}); in DefinePattern()
113 VectorRef assign1 = VectorRef({prim::kPrimAssign, input_vars_[kIndex2], add0}); in DefinePattern()
115 VectorRef assign2 = VectorRef({prim::kPrimAssign, input_vars_[kIndex1], add1}); in DefinePattern()
133 VectorRef assign0 = VectorRef({prim::kPrimAssign, input_vars_[kIndex3], sub0}); in DefinePattern()
135 VectorRef assign1 = VectorRef({prim::kPrimAssign, input_vars_[kIndex2], add0}); in DefinePattern()
137 VectorRef assign2 = VectorRef({prim::kPrimAssign, input_vars_[kIndex1], add1}); in DefinePattern()
155 VectorRef assign0 = VectorRef({prim::kPrimAssign, input_vars_[kIndex3], sub0}); in DefinePattern()
157 VectorRef assign1 = VectorRef({prim::kPrimAssign, input_vars_[kIndex2], add0}); in DefinePattern()
159 VectorRef assign2 = VectorRef({prim::kPrimAssign, input_vars_[kIndex1], add1}); in DefinePattern()
177 VectorRef assign0 = VectorRef({prim::kPrimAssign, input_vars_[kIndex3], sub0}); in DefinePattern()
[all …]
|
| D | adam_apply_one_with_decay_rule.cc | 164 VectorRef assign0 = VectorRef({prim::kPrimAssign, input3_, sub0}); in DefinePattern()
166 VectorRef assign1 = VectorRef({prim::kPrimAssign, input2_, add0}); in DefinePattern()
168 VectorRef assign2 = VectorRef({prim::kPrimAssign, input1_, add1}); in DefinePattern()
189 VectorRef assign0 = VectorRef({prim::kPrimAssign, input3_, sub0}); in DefinePattern()
191 VectorRef assign1 = VectorRef({prim::kPrimAssign, input2_, add0}); in DefinePattern()
193 VectorRef assign2 = VectorRef({prim::kPrimAssign, input1_, add1}); in DefinePattern()
214 VectorRef assign0 = VectorRef({prim::kPrimAssign, input3_, sub0}); in DefinePattern()
216 VectorRef assign1 = VectorRef({prim::kPrimAssign, input2_, add0}); in DefinePattern()
218 VectorRef assign2 = VectorRef({prim::kPrimAssign, input1_, add1}); in DefinePattern()
239 VectorRef assign0 = VectorRef({prim::kPrimAssign, input3_, sub0}); in DefinePattern()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | updatestate_eliminate.cc | 400 if (IsPrimitiveCNode(a1, prim::kPrimAssign)) { in EliminateUpdateStateBetweenAssigns()
437 if (IsPrimitiveCNode(attach, prim::kPrimAssign)) { in EliminateLoadBeforeAssigns()
487 if (IsPrimitiveCNode(a1, prim::kPrimAssign) && IsPrimitiveCNode(a2, prim::kPrimAssign)) { in EliminateUpdateStateBetweenAssignMakeTuple()
546 if (IsPrimitiveCNode(a2, prim::kPrimAssign) && IsPrimitiveCNode(a3, prim::kPrimAssign)) { in EliminateUpdateStateBetweenMakeTupleAssign()
566 if (IsPrimitiveCNode(a1, prim::kPrimAssign)) { in EliminateUpdateStateBetweenMakeTupleAssign()
606 if (IsPrimitiveCNode(attach, prim::kPrimAssign)) { in EliminateUpdateStateForAssign()
615 if (IsPrimitiveCNode(u_attach, prim::kPrimAssign)) { in EliminateUpdateStateForAssign()
738 …if (IsPrimitiveCNode(attach, prim::kPrimAssign) || IsPrimitiveCNode(attach, prim::kPrimMakeTuple))… in operator ()()
|
| D | branch_culling.cc | 71 {prim::kPrimAssign, {1}}, in InConvertWhiteList()
91 {prim::kPrimAssign, {1}}, {prim::kPrimAssignAdd, {1}}, in InConvertWhiteList()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | adam_fusion.cc | 109 VectorRef assign_param = VectorRef({prim::kPrimAssign, param_, next_param, next_state}); in DefinePattern()
113 VectorRef assign_m = VectorRef({prim::kPrimAssign, m_, next_m, next_state}); in DefinePattern()
117 VectorRef assign_v = VectorRef({prim::kPrimAssign, v_, next_v, next_state}); in DefinePattern()
|
| D | adam_weight_decay_fusion.cc | 111 VectorRef assign_param = VectorRef({prim::kPrimAssign, param_, next_param, next_state}); in DefinePattern()
115 VectorRef assign_m = VectorRef({prim::kPrimAssign, m_, next_m, next_state}); in DefinePattern()
119 VectorRef assign_v = VectorRef({prim::kPrimAssign, v_, next_v, next_state}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | optimize_assign.cc | 60 if (IsPrimitiveCNode(out, prim::kPrimAssign)) { in FindAssignAndOutputVal()
167 if (!AnfAlgo::CheckPrimitiveType(n, prim::kPrimAssign)) continue; in ReplaceAssignByInplaceAssignInGraphkernel()
|
| D | split_umonad.cc | 38 bool CanSplit(const AnfNodePtr &node) { return IsPrimitiveCNode(node, prim::kPrimAssign); } in CanSplit()
|
| D | graph_kernel_cluster.cc | 68 {kGPUDevice, OpLevel_0, prim::kPrimAssign}, in GetClusterableOpList()
409 …if (AnfAlgo::IsGraphKernel(node) || IsPrimitiveCNode(node, prim::kPrimAssign) || !IsClusterableOp(… in Process()
|
| D | eliminate_redundant_output.cc | 101 std::vector<PrimitivePtr> side_effect_nodes = {prim::kPrimAssign, prim::kPrimInplaceAssign}; in IsSideEffectNode()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | assign.cc | 69 REGISTER_PRIMITIVE_EVAL_IMPL(Assign, prim::kPrimAssign, InferImplAssign, nullptr, true);
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_auto_monad.cc | 868 node_name == prim::kPrimAssign->name()) { in InsertStackOps()
877 node_name == prim::kPrimAssign->name()) { in InsertStackOps()
890 if (AnfAlgo::CheckPrimitiveType(node, prim::kPrimAssign)) { in FindInputNode()
1231 auto assign_prim = std::make_shared<Primitive>(prim::kPrimAssign->name()); in Assign()
1656 if (IsPrimitiveCNode(node, prim::kPrimAssign) && in RemoveSameInputsAssigns()
1813 if (!IsPrimitiveEquals(prim, prim::kPrimAssign)) { in IsOptimizableAssign()
|
| D | gpu_session.cc | 294 return output_cnode != nullptr && IsPrimitiveCNode(output_cnode, prim::kPrimAssign) && in UpdatedByAssign()
|
| D | kernel_graph.cc | 38 const std::set<std::string> kOpAssignKernelNameList = {prim::kPrimAssign->name(), prim::kPrimAssign…
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | elewise_calculation_ops_declare.cc | 26 REG_ADPT_DESC(Assign, prim::kPrimAssign->name(), ADPT_DESC(Assign))
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 559 auto assign_prim = prim::kPrimAssign; in CreateAssign()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 539 inline const PrimitivePtr kPrimAssign = std::make_shared<Primitive>("Assign"); variable
|