Method Await

tensorflow/compiler/xla/python/tpu_driver/grpc_tpu_driver.cc:445–448

GrpcEvent::~GrpcEvent() { stream_->DeleteEvent(id_); }

Status GrpcEvent::Await() {
  auto opt_status = stream_->WaitForEvent(id_, absl::InfiniteDuration());
  return opt_status.value();
}

absl::optional<Status> GrpcEvent::AwaitWithTimeout(absl::Duration duration) {
  return stream_->WaitForEvent(id_, duration);