Searched refs:em_ (Results 1 – 3 of 3) sorted by relevance
31 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 …]
155 std::unique_ptr<EventMgr> em_; variable
380 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()