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