Searched refs:_batch_all_reduce (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 843 return self._batch_all_reduce(reduce_op, [per_replica_value])[0]
851 return self._batch_all_reduce(reduce_op,
859 def _batch_all_reduce(self, reduce_op, per_replica_values): member in AllReduceCrossDeviceOps
1133 all_reduced = self._batch_all_reduce(reduce_op, [per_replica_value],
1167 return self._batch_all_reduce(reduce_op,
1181 def _batch_all_reduce(self, reduce_op, per_replica_values, options): member in CollectiveAllReduce
|