CollectiveKeys (reference) in projects: external

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

/external/tensorflow/tensorflow/python/distribute/
D	collective_all_reduce_strategy.py	`125 self._collective_keys = cross_device_utils.CollectiveKeys() 190 self._collective_keys = cross_device_utils.CollectiveKeys()`
D	cross_device_utils.py	`237 class CollectiveKeys(object): class`
D	cross_device_ops.py	`970 cross_device_utils.CollectiveKeys())`
/external/tensorflow/tensorflow/contrib/distribute/python/
D	cross_device_ops_test.py	`459 collective_keys = cross_device_utils.CollectiveKeys(`
D	collective_all_reduce_strategy_test.py	`135 collective_keys = cross_device_utils.CollectiveKeys(`