Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_utils.py173 class CollectiveKeys(object): class
250 copied = CollectiveKeys()
Dcollective_all_reduce_strategy.py372 self._collective_keys = cross_device_utils.CollectiveKeys(
487 self._collective_keys = cross_device_utils.CollectiveKeys(
Dmirrored_strategy.py374 self._collective_keys = cross_device_utils.CollectiveKeys(
Dcross_device_ops.py1042 cross_device_utils.CollectiveKeys())
/external/tensorflow/tensorflow/python/distribute/v1/
Dcross_device_ops_test.py464 collective_keys = cross_device_utils.CollectiveKeys(