Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr_test.cc31 explicit TEST_EventMgrHelper(EventMgr* em) : em_(em) { in TEST_EventMgrHelper()
40 mutex_lock l(em_->mu_); in queue_size()
41 return em_->used_events_.size(); in queue_size()
45 mutex_lock l(em_->mu_); in free_size()
46 return em_->free_events_.size(); in free_size()
50 mutex_lock l(em_->mu_); in QueueTensors()
51 em_->QueueTensors(stream, tensors); in QueueTensors()
61 mutex_lock l(em_->mu_); in PollEvents()
62 em_->PollEvents(is_dedicated_poller, &to_free); in PollEvents()
64 em_->FreeMemory(to_free); in PollEvents()
[all …]
Dgpu_device.h155 std::unique_ptr<EventMgr> em_; variable
Dgpu_device.cc380 em_.reset(new EventMgr(executor_, options.config.gpu_options())); in Init()
397 gpu_device_info_->event_mgr = em_.get(); in Init()
589 em_->ThenExecute(stream, [op_kernel, tracker, queued_count]() { in ComputeHelper()
608 em_->ThenDeleteTensors(stream, tensor_refs); in ConsumeListOfAccessedTensors()