Searched refs:succ_edges (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | dp_algo_costmodel.cc | 106 auto succ_edges = graph->EliminationStar(star_center); in GetStrategy() local
108 for (size_t i = 0; i < succ_edges.size(); ++i) { in GetStrategy()
109 MS_EXCEPTION_IF_NULL(succ_edges[i]); in GetStrategy()
110 succ_nodes.push_back(succ_edges[i]->next_operator()); in GetStrategy()
112 auto elimi_star = std::make_shared<StarElimination>(star_center, succ_edges, succ_nodes); in GetStrategy()
208 auto succ_edges = elimination_star->succ_edges_; in RecoverStrategy() local
214 for (size_t i = 0; i < succ_edges.size(); ++i) { in RecoverStrategy()
215 succ_edges[i]->set_selected_cost(decision_star->succ_edges_cost_list_[i]); in RecoverStrategy()
|
| D | graph_costmodel.cc | 128 for (auto &edge : curr_op->succ_edges()) { in BFS()
214 for (auto &edge : current_op->succ_edges()) { in DFS()
1464 void CostGraph::CreateStarEliminationCostList(std::vector<std::shared_ptr<Edge>> &succ_edges, in CreateStarEliminationCostList() argument
1470 std::vector<StrategyPtr> succ_nodes_stras(succ_edges.size(), nullptr); in CreateStarEliminationCostList()
1471 …CostPtrList succ_edges_costs(succ_edges.size(), nullptr), succ_nodes_costs(succ_edges.size(), null… in CreateStarEliminationCostList()
1474 … &succ_nodes_costs, &first_succ_node_clist_new, &succ_edges, &recursive, in CreateStarEliminationCostList()
1476 if (k == succ_edges.size()) { in CreateStarEliminationCostList()
1486 auto succ_edge = succ_edges[k]; in CreateStarEliminationCostList()
1514 auto succ_edges = merged_op->GetAliveSuccEdges(); in EliminationStar() local
1516 for (auto &succ_edge : succ_edges) { in EliminationStar()
[all …]
|
| /third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/ |
| D | graph_costmodel_test.cc | 379 ASSERT_EQ(new_edge.get(), matmul1->succ_edges()[0].get()); in TEST_F()
406 ASSERT_EQ(new_edge.get(), matmul1->succ_edges()[0].get()); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.h | 131 std::vector<std::shared_ptr<Edge>> succ_edges() const { return succ_edges_; } in succ_edges() function
|
| D | operator_info.cc | 1458 for (auto &next_edge : succ_edges()) { in ExactStrategiesAndRelatedEdges()
|