Home
last modified time | relevance | path

Searched refs:group_key_start (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_utils.py188 def __init__(self, group_key_start=1): argument
194 self._group_key = group_key_start
Dcollective_all_reduce_strategy.py373 group_key_start=1 + self._collective_key_base)
488 group_key_start=1 + self._collective_key_base)
Dmirrored_strategy.py375 group_key_start=1 + self._collective_key_base) # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/distribute/v1/
Dcross_device_ops_test.py465 group_key_start=10 + CollectiveAllReduceTest.collective_key_base)