Home
last modified time | relevance | path

Searched refs:SetSelectedStrategyAndCost (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Ddp_algo_costmodel.cc144 w->SetSelectedStrategyAndCost(decision_op->op_strategy_, decision_op->middle_cost_); in RecoverStrategy()
166 … merged_node->SetSelectedStrategyAndCost(decision->merged_op_strategy_, decision->merged_op_cost_); in RecoverStrategy()
168 …target_node_m->SetSelectedStrategyAndCost(decision->target_op_strategy_, decision->target_op_cost_… in RecoverStrategy()
176 contracted_node->SetSelectedStrategyAndCost(decision_cont->contracted_op_strategy_, in RecoverStrategy()
179 …target_node->SetSelectedStrategyAndCost(decision_cont->target_op_strategy_, decision_cont->target_… in RecoverStrategy()
190 eliminated_node->SetSelectedStrategyAndCost(decision_tri->eliminated_op_strategy_, in RecoverStrategy()
195 …left_node->SetSelectedStrategyAndCost(decision_tri->left_node_strategy_, decision_tri->left_node_c… in RecoverStrategy()
199 …right_node->SetSelectedStrategyAndCost(decision_tri->right_node_strategy_, decision_tri->right_nod… in RecoverStrategy()
212 merged_node_star->SetSelectedStrategyAndCost(decision_star->eliminated_op_strategy_, in RecoverStrategy()
221 succ_nodes[0]->SetSelectedStrategyAndCost(decision_star->succ_ops_stra_list_[0], in RecoverStrategy()
[all …]
Dgraph_costmodel.cc531 u->SetSelectedStrategyAndCost(decision_f->u_strategy_, decision_f->u_cost_); in SearchStrategyForMultiNodeFinalGraph()
555 u->SetSelectedStrategyAndCost(decision->u_strategy_, decision->left_cost_); in SearchStrategyForMultiNodeFinalGraph()
556 v->SetSelectedStrategyAndCost(decision->v_strategy_, decision->right_cost_); in SearchStrategyForMultiNodeFinalGraph()
621 one_operator->SetSelectedStrategyAndCost(decision->u_strategy_, decision->u_cost_); in SearchStrategyForTwoNodeFinalGraph()
650 u->SetSelectedStrategyAndCost(f_decision->u_strategy_, f_decision->left_cost_); in SearchStrategyForTwoNodeFinalGraph()
651 v->SetSelectedStrategyAndCost(f_decision->v_strategy_, f_decision->right_cost_); in SearchStrategyForTwoNodeFinalGraph()
699 u->SetSelectedStrategyAndCost(decision->u_strategy_, decision->u_cost_); in SearchStrategy()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_generate_strategy.cc54 op->SetSelectedStrategyAndCost(user_defined_stra, op->selected_cost()); in GenerateStrategy()
577 op->SetSelectedStrategyAndCost(sp, op->selected_cost()); in SetBackToRawStrategy()
620 ops[iter_ops]->SetSelectedStrategyAndCost(sp, ops[iter_ops]->selected_cost()); in GeneratePartitionedOperatorStrategy()
1142 ops[iter_ops]->SetSelectedStrategyAndCost(sp, ops[iter_ops]->selected_cost()); in GenerateEliminatedOperatorStrategyForward()
1237 ops[iter_ops]->SetSelectedStrategyAndCost(sp, ops[iter_ops]->selected_cost()); in GenerateEliminatedOperatorStrategyBackward()
1279 ops[iter_ops]->SetSelectedStrategyAndCost(sp, ops[iter_ops]->selected_cost()); in GenerateRemainingOperatorStrategy()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h140 void SetSelectedStrategyAndCost(const StrategyPtr &s_strategy, const CostPtr &cost) { in SetSelectedStrategyAndCost() function
/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Dgraph_costmodel_test.cc445 matmul1->SetSelectedStrategyAndCost(decision->merged_op_strategy_, decision->merged_op_cost_); in TEST_F()