Searched refs:MirroredExtended (Results 1 – 3 of 3) sorted by relevance
31 CoreMirroredExtended = mirrored_strategy.MirroredExtended100 extended = MirroredExtended(self, devices, num_gpus,161 class MirroredExtended(CoreMirroredExtended): class175 super(MirroredExtended, self).__init__(container_strategy, devices,
437 extended = MirroredExtended(442 class MirroredExtended(distribute_lib.DistributionStrategyExtended): class446 super(MirroredExtended, self).__init__(container_strategy)
83 class CollectiveAllReduceExtended(mirrored_strategy.MirroredExtended):