Home
last modified time | relevance | path

Searched refs:DirectEvent (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Ddirect_tpu_driver.cc62 class DirectEvent : public Event { class
64 explicit DirectEvent(::TpuDriverFn* driver_fn, ::TpuEvent* event) in DirectEvent() function in tpu_driver::__anon9d833a900111::DirectEvent
67 ~DirectEvent() override { driver_fn_->TpuDriver_FreeEvent(event_); } in ~DirectEvent()
119 : handle_(handle), event_(new DirectEvent(driver_fn, handle->event)) {} in DirectBufferHandle()
132 std::shared_ptr<DirectEvent> event_;
143 event_(new DirectEvent(driver_fn, handle->event)) {} in DirectCompiledProgramHandle()
169 std::shared_ptr<DirectEvent> event_;
178 : handle_(handle), event_(new DirectEvent(driver_fn, handle->event)) {} in DirectLoadedProgramHandle()
188 std::shared_ptr<DirectEvent> event_;
330 auto event = std::make_shared<DirectEvent>( in Deallocate()
[all …]