Searched refs:build_ring_all_reduce (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce.py | 254 def build_ring_all_reduce(input_tensors, num_workers, num_subchunks, function
776 return build_ring_all_reduce(y, len(y), subdiv, [0], red_op, un_op)
843 return build_ring_all_reduce(tensors, len(tensors), subdiv, [0],
|
| D | all_reduce_test.py | 147 return lambda x, un_op: ar.build_ring_all_reduce(
|
| D | cross_device_utils.py | 393 summed_grads = all_reduce.build_ring_all_reduce(
|