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