Searched refs:orig_src_output (Results 1 – 1 of 1) sorted by relevance
714 const int orig_src_output, DataType dtype, in CreateSplitNode() argument720 int to_split_index = orig_src_output; in CreateSplitNode()725 orig_src, orig_src_output, graph)); in CreateSplitNode()794 int orig_src_output, Node* orig_src, Node* control_predecessor, in CreateOrGetSplitNodesForInputSharding() argument839 orig_src_output, dtype, in CreateOrGetSplitNodesForInputSharding()