Searched refs:group_key_start (Results 1 – 4 of 4) sorted by relevance
188 def __init__(self, group_key_start=1): argument194 self._group_key = group_key_start
373 group_key_start=1 + self._collective_key_base)488 group_key_start=1 + self._collective_key_base)
375 group_key_start=1 + self._collective_key_base) # pylint: disable=protected-access
465 group_key_start=10 + CollectiveAllReduceTest.collective_key_base)