get_group_key (reference) in projects: external

Searched refs:get_group_key (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
D	cross_device_utils.py	`199 def get_group_key(self, devices): member in CollectiveKeys`
D	collective_all_reduce_strategy.py	`548 group_key = self._collective_keys.get_group_key([device])`
D	cross_device_ops.py	`1059 group_key = self._collective_keys.get_group_key(self._devices)`