Searched defs:batch_all_reduce (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | cross_device_utils.py | 390 def batch_all_reduce(self, member in CollectiveReplicaLauncher |
| D | distribute_lib.py | 3242 def batch_all_reduce(strategy, *value_flat): function |