Searched refs:_all_reduce (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | nccl_ops.py | 217 def _all_reduce(): function
234 return def_function.function(_all_reduce)()
236 return _all_reduce()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 532 def _all_reduce(self, reduce_op, value, replica_id, options): member in CrossDeviceOps
1081 def _all_reduce(self, reduce_op, value, replica_id, options): member in CollectiveAllReduce
|
| D | cross_device_ops_test.py | 597 collective._all_reduce(reduce_op, value, replica_id, options))
614 collective._all_reduce(reduce_op, value, replica_id, options))
|
| D | collective_all_reduce_strategy.py | 793 return self._cross_device_ops._all_reduce( # pylint: disable=protected-access
|