Searched refs:forward_node (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | pynative_dfunctor.cc | 78 void PynativeDFunctor::GetForwardOutNodeAndBpropGraph(const CNodePtr &k_app, CNodePtr *forward_node, in GetForwardOutNodeAndBpropGraph() argument
121 MS_EXCEPTION_IF_NULL(forward_node); in GetForwardOutNodeAndBpropGraph()
123 *forward_node = output_node->cast<CNodePtr>(); in GetForwardOutNodeAndBpropGraph()
127 std::vector<AnfNodePtr> PynativeDFunctor::RunOutputReplace(const CNodePtr &forward_node, in RunOutputReplace() argument
141 MS_EXCEPTION_IF_NULL(forward_node); in RunOutputReplace()
142 auto ref_size = manager->node_users().at(forward_node).size(); in RunOutputReplace()
147 manager->Replace(forward_node, output_vnode); in RunOutputReplace()
148 …MS_LOG(DEBUG) << "Replace: " << forward_node->DebugString() << " with " << output_vnode->ToString(… in RunOutputReplace()
213 CNodePtr forward_node = nullptr; in ReplaceEquivdout() local
216 GetForwardOutNodeAndBpropGraph(k_app, &forward_node, &bprop_graph, &fprop_graph); in ReplaceEquivdout()
[all …]
|
| D | pynative_dfunctor.h | 30 static ValueNodePtr GenNewTensor(const CNodePtr &forward_node);
32 …static void GetForwardOutNodeAndBpropGraph(const CNodePtr &k_app, CNodePtr *forward_node, FuncGrap…
34 …static std::vector<AnfNodePtr> RunOutputReplace(const CNodePtr &forward_node, const FuncGraphPtr &…
|
| D | dfunctor.h | 104 ValueNodePtr GenNewTensor(const CNodePtr &forward_node);
106 …void GetForwardOutNodeAndBpropGraph(const CNodePtr &k_app, CNodePtr *forward_node, FuncGraphPtr *b…
108 …std::vector<AnfNodePtr> RunOutputReplace(const CNodePtr &forward_node, const FuncGraphPtr &bprop_g…
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_add_relu_grad_fusion.cc | 142 auto forward_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_getitem), 0); in PatternCheck() local
143 if (AnfAlgo::GetCNodeName(forward_node) != kBatchNormWithAddAndActivation) { in PatternCheck()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | program_specialize.cc | 366 const auto &forward_node = new_conf->node(); in BuildReplacedNode() local
367 MS_LOG(DEBUG) << "Replaced conf: node(" << forward_node->DebugString() << ")"; in BuildReplacedNode()
368 const auto &replicated_forward_node = ReplicateDisconnectedNode(forward_node); in BuildReplacedNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 296 … CNodePtr forward_node = func_graph->NewCNode(forward_input); // using NewCNode to create anfnode in ForwardCommunication() local
297 MS_EXCEPTION_IF_NULL(forward_node); in ForwardCommunication()
300 forward_node->set_scope(scope); in ForwardCommunication()
301 forward_node->set_in_forward_flag(true); in ForwardCommunication()
303 (void)manager->Replace(node_to_insert, forward_node); // using Replace function to insert node in ForwardCommunication()
|