Searched refs:_TPUReplicaContext (Results 1 – 1 of 1) sorted by relevance
1089 with _TPUReplicaContext(self._container_strategy()):1449 with _TPUReplicaContext(strategy, replica_id_in_sync_group=replica_id):1527 class _TPUReplicaContext(distribute_lib.ReplicaContext): class