Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_auto_parallel_test.cc123 std::vector<AnfNodePtr> MatMul_2_inputs; in Create_two_nodes() local
124 MatMul_2_inputs.push_back(NewValueNode(prim::kPrimMatMul)); in Create_two_nodes()
125 MatMul_2_inputs.push_back(MatMul_1_node); in Create_two_nodes()
126 MatMul_2_inputs.push_back(paramW); in Create_two_nodes()
127 CNodePtr MatMul_2_node = func_graph->NewCNode(MatMul_2_inputs); in Create_two_nodes()