Searched refs:_choose_all_reduce_algorithm (Results 1 – 2 of 2) sorted by relevance
259 result = cross_device_ops_lib._choose_all_reduce_algorithm(device_links)266 result = cross_device_ops_lib._choose_all_reduce_algorithm(device_links)275 result = cross_device_ops_lib._choose_all_reduce_algorithm(device_links)283 result = cross_device_ops_lib._choose_all_reduce_algorithm(device_links)
1077 def _choose_all_reduce_algorithm(device_links): function1120 return _choose_all_reduce_algorithm(device_links)