Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc48 explicit TEST_EventMgrHelper(EventMgr* em) : em_(em) { in TEST_EventMgrHelper()
57 mutex_lock l(em_->mu_); in queue_size()
58 return em_->used_events_.size(); in queue_size()
62 mutex_lock l(em_->mu_); in free_size()
63 return em_->free_events_.size(); in free_size()
73 mutex_lock l(em_->mu_); in PollEvents()
74 em_->PollEvents(true, &to_free); in PollEvents()
76 em_->FreeMemory(to_free); in PollEvents()
80 void StopPollingLoop() { return em_->StopPollingLoop(); } in StopPollingLoop()
82 void StartPollingLoop() { return em_->StartPollingLoop(); } in StartPollingLoop()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h151 EventMgr* em_ = nullptr; variable
210 em_(event_manager), in GPUKernelTracker()
275 EventMgr* em_ = nullptr; variable
Dgpu_device.cc444 em_ = EventMgrFactory::Singleton()->GetEventMgr(executor_, in Init()
469 timestamped_allocator_ ? gpu_allocator_ : nullptr, em_)); in Init()
475 gpu_device_info_->event_mgr = em_; in Init()
590 em_->ThenExecute(stream, [tracker, queued_count]() { in Compute()
2011 em_->ThenExecute(stream_, in MaybeQueueProgressEvent()