Searched defs:forward_node (Results 1 – 9 of 9) sorted by relevance
134 void GetForwardOutNodeAndBpropGraph(const CNodePtr &k_app, CNodePtr *forward_node, FuncGraphPtr *bp… in GetForwardOutNodeAndBpropGraph()183 std::vector<AnfNodePtr> RunOutputReplace(const CNodePtr &forward_node, const FuncGraphPtr &bprop_gr… in RunOutputReplace()271 CNodePtr forward_node = nullptr; in ReplaceEquivOut() local
171 auto forward_node = forward_comm_nodes_map[pair.first]; in LabelMicroInterleavedIndexForBackwardCommNodes() local209 auto forward_node = forward_nodes_map[pair.first]; in LabelMicroInterleavedBranchTagForBackwardCommNodes() local
113 auto forward_node = forward_comm_nodes_map[pair.first]; in LabelMicroInterleavedIndexForBackwardCommNodes() local
102 for (auto &forward_node : forward_nodes) { in HasRecompute() local226 for (auto &forward_node : forward_nodes) { in GetSeqLengthAndAttentionHeads() local289 for (auto &forward_node : forward_nodes) { in GetPerBatch() local356 for (auto &forward_node : forward_nodes) { in GetFPFromNodes() local
151 for (auto &forward_node : forward_each_seg_first_recv_[segment_info]) { in IsChanged() local
161 auto forward_node = common::AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_getitem), 0); in PatternCheck() local
1094 const auto &forward_node = new_conf->node(); in BuildReplacedNode() local
174 … CNodePtr forward_node = func_graph->NewCNode(forward_input); // using NewCNode to create anfnode in ForwardCommunicationForMultiOut() local229 … CNodePtr forward_node = func_graph->NewCNode(forward_input); // using NewCNode to create anfnode in ForwardCommunication() local
638 for (const auto &forward_node : bprop_graph->used_forward_nodes()) { in ReplaceCNodeWithValueNode() local