Method BlockUntilDone

tensorflow/compiler/xla/service/execution_tracker.cc:40–45

}

Status AsyncExecution::BlockUntilDone() const {
  for (auto& stream : streams_) {
    TF_RETURN_IF_ERROR(stream->BlockHostUntilDone());
  }
  return Status::OK();
}

ExecutionTracker::ExecutionTracker() : next_handle_(1) {}