Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dpynative_dfunctor.cc151 std::vector<AnfNodePtr> used_forward_nodes; in RunOutputReplace() local
155 used_forward_nodes.push_back(cnode_morph); in RunOutputReplace()
159 return used_forward_nodes; in RunOutputReplace()
222 auto used_forward_nodes = RunOutputReplace(forward_node, bprop_graph, fprop_graph, cnode_morph); in ReplaceEquivdout() local
228 ms_func_graph->set_used_forward_nodes(used_forward_nodes); in ReplaceEquivdout()
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph.cc745 void FuncGraph::set_used_forward_nodes(const std::vector<AnfNodePtr> &used_forward_nodes) { in set_used_forward_nodes() argument
746 …(void)std::for_each(used_forward_nodes.begin(), used_forward_nodes.end(), [this](const AnfNodePtr … in set_used_forward_nodes()
Dfunc_graph.h410 const std::unordered_set<AnfNodePtr> &used_forward_nodes() const { return used_forward_nodes_; } in used_forward_nodes() function
411 void set_used_forward_nodes(const std::vector<AnfNodePtr> &used_forward_nodes);
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Daction.cc146 const auto &used_forward_nodes = func_graph->used_forward_nodes(); in ModifyOutputNode() local
157 std::for_each(used_forward_nodes.begin(), used_forward_nodes.end(), in ModifyOutputNode()