Searched refs:second_gather_devices (Results 1 – 2 of 2) sorted by relevance
405 second_gather_devices = aux_devices[:num_shards]407 scaled_grads, aux_devices, second_gather_devices, math_ops.add_n)
852 second_gather_devices, red_op, un_op=None): argument855 return build_shuffle_all_reduce(tensors, second_gather_devices,