Home
last modified time | relevance | path

Searched refs:AddSuccEdge (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Ddp_algo_test.cc275 matmul7->AddSuccEdge(edge_m7_m8); in ConstructTwoLargeMatMul()
417 tmp_identity_ptr->AddSuccEdge(edge_i0_m3); in ConstructBatmanGraph()
421 tmp_identity_ptr->AddSuccEdge(edge_i0_m6); in ConstructBatmanGraph()
425 tmp_identity_ptr1->AddSuccEdge(edge_i1_m2); in ConstructBatmanGraph()
429 tmp_identity_ptr1->AddSuccEdge(edge_i1_m4); in ConstructBatmanGraph()
433 tmp_identity_ptr2->AddSuccEdge(edge_i2_m5); in ConstructBatmanGraph()
437 tmp_identity_ptr2->AddSuccEdge(edge_i2_m7); in ConstructBatmanGraph()
441 mm1_ptr->AddSuccEdge(edge_m1_m2); in ConstructBatmanGraph()
445 mm1_ptr->AddSuccEdge(edge_m1_m5); in ConstructBatmanGraph()
449 mm2_ptr->AddSuccEdge(edge_m2_m3); in ConstructBatmanGraph()
[all …]
Dgraph_costmodel_test.cc145 matmul1->AddSuccEdge(edge_m1_m2); in ConstructStarGraph2()
148 matmul0->AddSuccEdge(edge_m0_m2); in ConstructStarGraph2()
151 matmul1->AddSuccEdge(edge_m1_m3); in ConstructStarGraph2()
171 matmul1->AddSuccEdge(edge_m1_m2); in ConstructDiamondGraph()
173 matmul2->AddSuccEdge(edge_m2_m4); in ConstructDiamondGraph()
175 matmul1->AddSuccEdge(edge_m1_m3); in ConstructDiamondGraph()
177 matmul3->AddSuccEdge(edge_m3_m4); in ConstructDiamondGraph()
194 matmul1->AddSuccEdge(edge_m1_m2); in ConstructLinearGraph()
197 matmul2->AddSuccEdge(edge_m2_m4); in ConstructLinearGraph()
220 matmul1->AddSuccEdge(edge_m1_m2); in ConstructStarGraph()
[all …]
Dedge_costmodel_test.cc113 matmul1->AddSuccEdge(edge_m1_m2); in TEST_F()
126 matmul1->AddSuccEdge(edge_m1_m2); in TEST_F()
128 matmul2->AddSuccEdge(edge_m2_m4); in TEST_F()
145 matmul1->AddSuccEdge(edge_m1_m5); in TEST_F()
146 matmul1->AddSuccEdge(edge_m1_m5_2); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h129 void AddSuccEdge(const std::shared_ptr<Edge> &e) { succ_edges_.push_back(e); } in AddSuccEdge() function
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_auto_parallel.cc628 prev_op_info->AddSuccEdge(edge_ptr); in CreateEdgeBetweenTwoOps()
838 tmp_identity_ptr->AddSuccEdge(edge_ptr); in AugmentCostGraph()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dgraph_costmodel.cc878 op1->AddSuccEdge(new_edge); in UpdateEdgesIncidentToNodes()