Searched refs:EventMgr (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr.cc | 92 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr() function in tensorflow::EventMgr
102 EventMgr::~EventMgr() { in ~EventMgr()
117 void EventMgr::StartPollingLoop() { in StartPollingLoop()
127 void EventMgr::StopPollingLoop() { in StopPollingLoop()
143 void EventMgr::PollLoop() { in PollLoop()
168 void EventMgr::QueueInUse(se::Stream* stream, InUse in_use) { in QueueInUse()
205 void EventMgr::PollEvents(bool is_dedicated_poller, in PollEvents()
250 EventMgr* EventMgrFactory::GetEventMgr(se::StreamExecutor* se, in GetEventMgr()
259 auto event_mgr = new EventMgr(se, gpu_options); in GetEventMgr()
|
| D | gpu_device.h | 151 EventMgr* em_ = nullptr;
204 EventMgr* event_manager) in GPUKernelTracker()
275 EventMgr* em_ = nullptr;
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr.cc | 93 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr() function in tensorflow::EventMgr
103 EventMgr::~EventMgr() { in ~EventMgr()
118 void EventMgr::StartPollingLoop() { in StartPollingLoop()
128 void EventMgr::StopPollingLoop() { in StopPollingLoop()
144 void EventMgr::PollLoop() { in PollLoop()
169 void EventMgr::QueueInUse(se::Stream* stream, InUse in_use) { in QueueInUse()
206 void EventMgr::PollEvents(bool is_dedicated_poller, in PollEvents()
251 EventMgr* EventMgrFactory::GetEventMgr(se::StreamExecutor* se, in GetEventMgr()
260 auto event_mgr = new EventMgr(se, gpu_options); in GetEventMgr()
|
| D | device_event_mgr.h | 66 class EventMgr {
68 virtual ~EventMgr();
99 EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
153 EventMgr* GetEventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
160 std::map<se::StreamExecutor*, EventMgr*> event_mgr_map_ TF_GUARDED_BY(mu_);
|
| D | device_event_mgr_test.cc | 40 class TEST_EventMgr : public EventMgr {
43 : EventMgr(se, gpu_options) {} in TEST_EventMgr()
48 explicit TEST_EventMgrHelper(EventMgr* em) : em_(em) { in TEST_EventMgrHelper()
71 EventMgr::ToFreeVector to_free; in PollEvents()
85 EventMgr* em_;
111 TEST(EventMgr, Empty) { in TEST() argument
120 TEST(EventMgr, WarnIfInCallback) { in TEST() argument
163 EventMgr* event_mgr() { return gpu_->em_; } in event_mgr()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 46 class EventMgr; variable
157 EventMgr* event_mgr = nullptr;
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 102 EventMgr* const event_mgr;
|