Searched refs:_collective_key_base (Results 1 – 5 of 5) sorted by relevance
165 _collective_key_base = 0 variable in CollectiveAllReduceStrategy277 _collective_key_base = 0 variable in CollectiveAllReduceStrategyV1327 …self._collective_key_base = container_strategy._collective_key_base # pylint: disable=protected-a…373 group_key_start=1 + self._collective_key_base)488 group_key_start=1 + self._collective_key_base)
275 _collective_key_base = 0 variable in MirroredStrategy291 _collective_key_base = 0 variable in MirroredStrategyV1336 self._collective_key_base = container_strategy._collective_key_base375 group_key_start=1 + self._collective_key_base) # pylint: disable=protected-access
123 mirrored_lib.MirroredStrategyV1._collective_key_base += 100000124 mirrored_lib.MirroredStrategy._collective_key_base += 100000
102 CollectiveAllReduceStrategy._collective_key_base += 100000
454 mwms_lib.CollectiveAllReduceStrategy._collective_key_base = (