Searched refs:shared_event_ (Results 1 – 1 of 1) sorted by relevance
48 : shared_event_(std::move(event)), id_(id_counter++) {} in RecordingEvent()51 : shared_event_(event), id_(id) {} in RecordingEvent()55 xla::Status Await() override { return shared_event_->Await(); } in Await()59 return shared_event_->AwaitWithTimeout(duration); in AwaitWithTimeout()63 return shared_event_->AddCallback(callback); in AddCallback()67 std::shared_ptr<Event> shared_event_; member in tpu_driver::__anone63593b20111::RecordingEvent529 static_cast<RecordingEvent*>(event)->shared_event_.get(); in UnwrapWaitFor()