Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/cpp/dataset/
Dtree_modifying_function_test.cc66 std::shared_ptr<DatasetNode> node_to_insert = ds6->IRNode(); in TEST_F() local
67 rc = ds4_node->AppendChild(node_to_insert); in TEST_F()
69 EXPECT_TRUE( ds4_node->Children()[2] == node_to_insert); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc271 CNodePtr node_to_insert = node; in ForwardCommunication() local
284 node_to_insert = uses_cnode; in ForwardCommunication()
287 MS_EXCEPTION_IF_NULL(node_to_insert); in ForwardCommunication()
294 …std::vector<AnfNodePtr> forward_input = CreateInput(forward_op[index], node_to_insert, instance_na… in ForwardCommunication()
295 SetAllReduceRecomputeFlag(forward_input, node_to_insert); in ForwardCommunication()
303 (void)manager->Replace(node_to_insert, forward_node); // using Replace function to insert node in ForwardCommunication()