Searched defs:gather_devices (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | all_reduce.py | 554 def build_shuffle_all_reduce(input_tensors, gather_devices, red_op, un_op=None): argument
588 def _build_shuffle_gather(input_tensors, gather_devices, red_op, un_op=None): argument
790 def build_nccl_then_shuffle(input_tensors, gather_devices, nccl_red_op, argument
799 def _build_shuffle_hybrid(input_tensors, gather_devices, red_op, upper_level_f): argument
841 def build_shuffle_then_ring(input_tensors, gather_devices, subdiv, argument
|