Home
last modified time | relevance | path

Searched refs:CostPtrList (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dgraph_costmodel.h69 …CostPtrList CreateFinalCostList(const OperatorInfoPtr &u, const EdgePtr &e, const OperatorInfoPtr …
70 CostPtrList CreateFinalSingleCostList(const OperatorInfoPtr &u);
71 CostPtr SelectCostWithMinInferenceTime(const CostPtrList &cost_list, double memory);
72 CostPtr SelectCostWithMinTrainingTime(const CostPtrList &cost_list, double memory);
73 …CostPtrList SelectCostListWithMinTrainingTimeMultiple(const std::vector<CostPtrList> &all_costlist,
125 void CreateMergeEliminationSubCostList(StrategyPtr op_strategy, const CostPtrList &op_cost_list,
126 … const CostPtrList &edge_cost_list, StrategyPtr tar_op_strategy,
127 … const CostPtrList &tar_cost_list, CostPtrList *tar_cost_list_new);
130 …void CreateContractEliminationSubCostList(StrategyPtr, const CostPtrList &, const CostPtrList &, S…
131 const CostPtrList &, CostPtrList *);
[all …]
Dedge_costmodel.cc56 CostPtrList cl; in InitEdgeCost()
85 CostPtrList cl; in InitEdgeCost()
155 CostPtrList Edge::GetCostList(StrategyPtr output_str, StrategyPtr input_str) { in GetCostList()
157 CostPtrList result; in GetCostList()
164 CostPtrList Edge::CreateEdgeEliminationCostList(const StrategyPtr &output_st_ptr, const std::vector… in CreateEdgeEliminationCostList()
166 std::function<CostPtrList(EdgePtr)> LocalGetCostList = [&](const EdgePtr &edge) { in CreateEdgeEliminationCostList()
170 CostPtrList result; in CreateEdgeEliminationCostList()
171 std::vector<CostPtrList> all_cost_list; in CreateEdgeEliminationCostList()
175 CostPtrList selected_cost_list(all_cost_list.size(), nullptr); in CreateEdgeEliminationCostList()
213 CostPtrList clist = CreateEdgeEliminationCostList(output_st_ptr, edges, input_st_ptr); in EdgeEliminationSetNewCost()
[all …]
Dedge_costmodel.h83 std::map<CostPtrKey, CostPtrList> GetCostMap() { return cost_map_; } in GetCostMap()
87 void SetCostMapAndInputOutput(std::map<CostPtrKey, CostPtrList> &);
102 CostPtrList GetCostList(StrategyPtr output_str, StrategyPtr input_str);
117 CostPtrList CreateEdgeEliminationCostList(const StrategyPtr &output_st_ptr,
124 void CreateOpEliminationSubCostList(StrategyPtr op_strategy, const CostPtrList &left_cost_list,
125 … const CostPtrList &middle_cost_list, const CostPtrList &right_cost_list,
126 CostPtrList *ret_cost_list);
128 …CostPtrList CreateOpEliminationCostList(const std::shared_ptr<Edge> &e1, const StrategyPtr &output…
152 std::map<CostPtrKey, CostPtrList> cost_map_;
Dcostmodel.h77 using CostPtrList = std::vector<std::shared_ptr<Cost>>; variable
83 StrategyWithCost(StrategyPtr strategy, CostPtrList c_list) in StrategyWithCost()
97 CostPtrList cost_list;
144 explicit EdgeEliminationDecision(CostPtrList cost_list) : edges_cost_list_(std::move(cost_list)) { in EdgeEliminationDecision()
148 CostPtrList edges_cost_list_;
282 …StarEliminationDecision(StrategyPtr elimi_op_stra, CostPtr elimi_op_cost, CostPtrList succ_edges_c… in StarEliminationDecision()
283 std::vector<StrategyPtr> succ_ops_stra_list, CostPtrList succ_ops_clist) in StarEliminationDecision()
294 CostPtrList succ_edges_cost_list_;
296 CostPtrList succ_ops_cost_list_;
343 void Simplify(CostPtrList *clist);
[all …]
Dgraph_costmodel.cc234 CostPtrList CostGraph::CreateFinalCostList(const OperatorInfoPtr &u, const std::shared_ptr<Edge> &e, in CreateFinalCostList()
239 CostPtrList ret; in CreateFinalCostList()
246 CostPtrList clist1 = u_strategy->cost_list; in CreateFinalCostList()
247 CostPtrList clist2 = e->GetCostList(u_strategy_ptr, v_strategy_ptr); in CreateFinalCostList()
248 CostPtrList clist3 = v_strategy->cost_list; in CreateFinalCostList()
285 CostPtrList CostGraph::CreateFinalSingleCostList(const OperatorInfoPtr &u) { in CreateFinalSingleCostList()
287 CostPtrList ret; in CreateFinalSingleCostList()
291 CostPtrList clist1 = u_strategy->cost_list; in CreateFinalSingleCostList()
312 CostPtr CostGraph::SelectCostWithMinInferenceTime(const CostPtrList &cost_list, double memory) { in SelectCostWithMinInferenceTime()
319 CostPtrList after_mem_filter; in SelectCostWithMinInferenceTime()
[all …]
Dcostmodel.cc25 void Simplify(CostPtrList *clist_ptrs) { in Simplify()
35 void SimplifyForDecreasingCommunicationForward(CostPtrList *clist_ptrs) { in SimplifyForDecreasingCommunicationForward()
49 CostPtrList ret; in SimplifyForDecreasingCommunicationForward()
59 void SimplifyForDecreasingCommunicationWithPartialPara(CostPtrList *clist_ptrs) { in SimplifyForDecreasingCommunicationWithPartialPara()
72 CostPtrList ret; in SimplifyForDecreasingCommunicationWithPartialPara()
/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Dgraph_costmodel_test.cc308 CostPtrList clist_1, clist_2; in TEST_F()
309 std::vector<CostPtrList> all_list; in TEST_F()