Searched refs:TPUExtended (Results 1 – 1 of 1) sorted by relevance
146 super(TPUStrategy, self).__init__(TPUExtended(192 class TPUExtended(distribute_lib.DistributionStrategyExtended): class200 super(TPUExtended, self).__init__(container_strategy)