Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Ddp_algo_costmodel.cc143 auto decision_op = e->selected_cost()->decision_ptr_->cast<OpEliminationDecisionPtr>(); in RecoverStrategy()
152 … auto decision_edge = new_edge->selected_cost()->decision_ptr_->cast<EdgeEliminationDecisionPtr>(); in RecoverStrategy()
164 MS_EXCEPTION_IF_NULL(target_node_m->selected_cost()->decision_ptr_); in RecoverStrategy()
165 …auto decision = target_node_m->selected_cost()->decision_ptr_->cast<MergeEliminationDecisionPtr>(); in RecoverStrategy()
175 …auto decision_cont = target_node->selected_cost()->decision_ptr_->cast<ContractEliminationDecision… in RecoverStrategy()
188 …auto decision_tri = left_node->selected_cost()->decision_ptr_->cast<TriangleEliminationDecisionPtr… in RecoverStrategy()
211 …auto decision_star = succ_nodes[0]->selected_cost()->decision_ptr_->cast<StarEliminationDecisionPt… in RecoverStrategy()
Dcostmodel.h49 …: computation_cost_(computation), communication_cost_(communication), decision_ptr_(std::move(deci… in computation_cost_()
73 std::shared_ptr<Decision> decision_ptr_; member
Dgraph_costmodel.cc530 auto decision_f = selected_cost->decision_ptr_->cast<FinalSingleDecisionPtr>(); in SearchStrategyForMultiNodeFinalGraph()
552 MS_EXCEPTION_IF_NULL(selected_cost->decision_ptr_); in SearchStrategyForMultiNodeFinalGraph()
553 auto decision = selected_cost->decision_ptr_->cast<FinalDecisionPtr>(); in SearchStrategyForMultiNodeFinalGraph()
618 MS_EXCEPTION_IF_NULL(selected_cost->decision_ptr_); in SearchStrategyForTwoNodeFinalGraph()
619 auto decision = selected_cost->decision_ptr_->cast<FinalSingleDecisionPtr>(); in SearchStrategyForTwoNodeFinalGraph()
647 MS_EXCEPTION_IF_NULL(cost->decision_ptr_); in SearchStrategyForTwoNodeFinalGraph()
648 auto f_decision = cost->decision_ptr_->cast<FinalDecisionPtr>(); in SearchStrategyForTwoNodeFinalGraph()
696 MS_EXCEPTION_IF_NULL(cost->decision_ptr_); in SearchStrategy()
697 auto decision = cost->decision_ptr_->cast<FinalSingleDecisionPtr>(); in SearchStrategy()
Dedge_costmodel.cc189 new_cost->decision_ptr_ = decision; in CreateEdgeEliminationCostList()
/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Dgraph_costmodel_test.cc443 auto decision = matmul2->selected_cost()->decision_ptr_->cast<MergeEliminationDecisionPtr>(); in TEST_F()