Home
last modified time | relevance | path

Searched refs:CollectiveKeys (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcross_device_utils.py173 class CollectiveKeys(object): class
249 copied = CollectiveKeys()
262 collective_keys: CollectiveKeys, device: str, argument
H A Dcollective_all_reduce_strategy.py413 self._collective_keys = cross_device_utils.CollectiveKeys(
538 self._collective_keys = cross_device_utils.CollectiveKeys(
H A Dmirrored_strategy.py379 self._collective_keys = cross_device_utils.CollectiveKeys(
H A Dcross_device_ops.py1071 cross_device_utils.CollectiveKeys())
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/
H A Dcross_device_ops_test.py461 collective_keys = cross_device_utils.CollectiveKeys(