Searched defs:new_node_input (Results 1 – 4 of 4) sorted by relevance
156 std::vector<AnfNodePtr> new_node_input; in CreateMirrorInput() local521 … const FuncGraphPtr &func_graph, std::vector<AnfNodePtr> *new_node_input) { in ConvertStridedSliceInputs()587 … const FuncGraphPtr &func_graph, std::vector<AnfNodePtr> *new_node_input) { in ConvertReshapeInputs()654 std::vector<AnfNodePtr> *new_node_input) { in ConvertSplitInputs()697 … const FuncGraphPtr &func_graph, std::vector<AnfNodePtr> *new_node_input) { in ConvertParamsToInputs()728 std::vector<AnfNodePtr> new_node_input = {pre_node}; in CreateInput() local
879 std::vector<AnfNodePtr> new_node_input = {NewValueNode(cloned_prim), input}; in GenNewNodeFromOld() local
786 void SetCommunicationOpGroupLabel(std::vector<AnfNodePtr> new_node_input) { in SetCommunicationOpGroupLabel()
82 static void SetAllReduceRecomputeFlag(const std::vector<AnfNodePtr> &new_node_input, const CNodePtr… in SetAllReduceRecomputeFlag()