Searched refs:build_ring_all_reduce (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | all_reduce.py | 254 def build_ring_all_reduce(input_tensors, num_workers, num_subchunks, function
782 return build_ring_all_reduce(y, len(y), subdiv, [0], red_op, un_op)
849 return build_ring_all_reduce(tensors, len(tensors), subdiv, [0],
|
| D | all_reduce_test.py | 146 return lambda x, un_op: ar.build_ring_all_reduce(
|