Searched refs:new_node_input (Results 1 – 3 of 3) sorted by relevance
163 void SetCommunicationOpGroupLabel(std::vector<AnfNodePtr> new_node_input) { in SetCommunicationOpGroupLabel() argument164 if (new_node_input.empty()) { in SetCommunicationOpGroupLabel()168 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetCommunicationOpGroupLabel()
32 void SetCommunicationOpGroupLabel(std::vector<AnfNodePtr> new_node_input);
67 void SetMiniStepOpDoMirrorLabel(std::vector<AnfNodePtr> new_node_input, bool accu_flag) { in SetMiniStepOpDoMirrorLabel() argument68 if (new_node_input.empty()) { in SetMiniStepOpDoMirrorLabel()71 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetMiniStepOpDoMirrorLabel()80 void SetAllReduceRecomputeFlag(const std::vector<AnfNodePtr> &new_node_input, const CNodePtr &node)… in SetAllReduceRecomputeFlag() argument81 if (new_node_input.empty()) { in SetAllReduceRecomputeFlag()85 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetAllReduceRecomputeFlag()108 std::vector<AnfNodePtr> new_node_input = {NewValueNode(pyop_instance), node}; in CreateInput() local114 (void)new_node_input.insert(new_node_input.begin() + position, val); in CreateInput()119 SetCommunicationOpGroupLabel(new_node_input); in CreateInput()120 return new_node_input; in CreateInput()[all …]