Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc149 node1->set_in_forward_flag(true); in Make_Manager()
167 node2->set_in_forward_flag(true); in Make_Manager()
173 cnode_return->set_in_forward_flag(true); in Make_Manager()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Ddropout_do_mask_info.cc287 replace_node->set_in_forward_flag(true); in ReplaceOneOp()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc214 new_node->set_in_forward_flag(true); // mark forward flag in InsertNode()
248 new_node->set_in_forward_flag(true); // mark forward flag in ReplaceNode()
301 forward_node->set_in_forward_flag(true); in ForwardCommunication()
674 new_node->set_in_forward_flag(true); in SplitTensorList()
762 replace_node->set_in_forward_flag(true); in StepReplaceOp()
2730 cnode->set_in_forward_flag(true); in SetForwardFlag()
2746 cnode->set_in_forward_flag(true); in SetForwardFlag()
/third_party/mindspore/mindspore/core/ir/
Danf.h292 void set_in_forward_flag(bool flag) { in_forward_flag_ = flag; } in set_in_forward_flag() function