Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Ddp_algo_test.cc276 matmul8->AddPrevEdge(edge_m7_m8); in ConstructTwoLargeMatMul()
418 mm3_ptr->AddPrevEdge(edge_i0_m3); in ConstructBatmanGraph()
422 mm6_ptr->AddPrevEdge(edge_i0_m6); in ConstructBatmanGraph()
426 mm2_ptr->AddPrevEdge(edge_i1_m2); in ConstructBatmanGraph()
430 mm4_ptr->AddPrevEdge(edge_i1_m4); in ConstructBatmanGraph()
434 mm5_ptr->AddPrevEdge(edge_i2_m5); in ConstructBatmanGraph()
438 mm7_ptr->AddPrevEdge(edge_i2_m7); in ConstructBatmanGraph()
442 mm2_ptr->AddPrevEdge(edge_m1_m2); in ConstructBatmanGraph()
446 mm5_ptr->AddPrevEdge(edge_m1_m5); in ConstructBatmanGraph()
450 mm3_ptr->AddPrevEdge(edge_m2_m3); in ConstructBatmanGraph()
[all …]
Dgraph_costmodel_test.cc146 matmul2->AddPrevEdge(edge_m1_m2); in ConstructStarGraph2()
149 matmul2->AddPrevEdge(edge_m0_m2); in ConstructStarGraph2()
152 matmul3->AddPrevEdge(edge_m1_m3); in ConstructStarGraph2()
172 matmul2->AddPrevEdge(edge_m1_m2); in ConstructDiamondGraph()
174 matmul4->AddPrevEdge(edge_m2_m4); in ConstructDiamondGraph()
176 matmul3->AddPrevEdge(edge_m1_m3); in ConstructDiamondGraph()
178 matmul4->AddPrevEdge(edge_m3_m4); in ConstructDiamondGraph()
195 matmul2->AddPrevEdge(edge_m1_m2); in ConstructLinearGraph()
198 matmul4->AddPrevEdge(edge_m2_m4); in ConstructLinearGraph()
221 matmul2->AddPrevEdge(edge_m1_m2); in ConstructStarGraph()
[all …]
Dedge_costmodel_test.cc114 matmul2->AddPrevEdge(edge_m1_m2); in TEST_F()
127 matmul2->AddPrevEdge(edge_m1_m2); in TEST_F()
129 matmul4->AddPrevEdge(edge_m2_m4); in TEST_F()
147 matmul5->AddPrevEdge(edge_m1_m5); in TEST_F()
148 matmul5->AddPrevEdge(edge_m1_m5_2); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h130 void AddPrevEdge(const std::shared_ptr<Edge> &e) { prev_edges_.push_back(e); } in AddPrevEdge() function
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_auto_parallel.cc627 node_op_info->AddPrevEdge(edge_ptr); in CreateEdgeBetweenTwoOps()
837 target_op_info->AddPrevEdge(edge_ptr); in AugmentCostGraph()