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