Searched refs:first_gather_devices (Results 1 – 1 of 1) sorted by relevance
857 def build_shuffle_then_shuffle(input_tensors, first_gather_devices, argument866 input_tensors, first_gather_devices, red_op, upper_level_f)