Searched refs:input_to_split_node (Results 1 – 1 of 1) sorted by relevance
778 Node* input_to_split_node = concat_node; in CreateConcatAndSplitNodesForInputTensor() local823 input_to_split_node = xla_sharding_op; in CreateConcatAndSplitNodesForInputTensor()830 .Input(input_to_split_node) in CreateConcatAndSplitNodesForInputTensor()