Searched refs:InsertSplitForInput (Results 1 – 2 of 2) sorted by relevance
22 std::vector<AnfNodePtr> SplitInputsForReduceScatter::InsertSplitForInput(const FuncGraphPtr &func_g… in InsertSplitForInput() function in mindspore::opt::SplitInputsForReduceScatter109 std::vector<AnfNodePtr> split_outputs = InsertSplitForInput(func_graph, cnode, rank_size); in Process()
38 std::vector<AnfNodePtr> InsertSplitForInput(const FuncGraphPtr &func_graph, const CNodePtr &node,