Searched refs:_TPUReplicaContext (Results 1 – 1 of 1) sorted by relevance
169 with _TPUReplicaContext(self, replica_id_in_sync_group=replica_id):386 with _TPUReplicaContext(self._container_strategy()):624 class _TPUReplicaContext(distribute_lib.ReplicaContext): class