Searched refs:EventMgr (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr_test.cc | 31 explicit TEST_EventMgrHelper(EventMgr* em) : em_(em) { in TEST_EventMgrHelper()
59 EventMgr::ToFreeVector to_free; in PollEvents()
73 EventMgr* em_;
99 TEST(EventMgr, Empty) { in TEST() argument
101 EventMgr em(stream_exec, GPUOptions()); in TEST()
116 TEST(EventMgr, DelayedPolling) { in TEST() argument
118 EventMgr em(stream_exec, GPUOptions()); in TEST()
149 TEST(EventMgr, FlushLargeTensorImmediately) { in TEST() argument
151 EventMgr em(stream_exec, GPUOptions()); in TEST()
166 TEST(EventMgr, ManySmallTensorsFlushedImmediately) { in TEST() argument
[all …]
|
| D | gpu_event_mgr.cc | 92 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr() function in tensorflow::EventMgr
108 EventMgr::~EventMgr() { in ~EventMgr()
141 void EventMgr::StartPollingLoop() { in StartPollingLoop()
151 void EventMgr::StopPollingLoop() { in StopPollingLoop()
163 void EventMgr::ThenDeleteTensors(se::Stream* stream, in ThenDeleteTensors()
183 void EventMgr::FlushAccumulatedTensors() { in FlushAccumulatedTensors()
196 void EventMgr::PollLoop() { in PollLoop()
221 void EventMgr::QueueInUse(se::Stream* stream, InUse iu) { in QueueInUse()
258 void EventMgr::PollEvents(bool is_dedicated_poller, in PollEvents()
|
| D | gpu_event_mgr.h | 64 class EventMgr {
66 EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
68 ~EventMgr();
|
| D | gpu_device.h | 155 std::unique_ptr<EventMgr> em_;
|
| D | gpu_device.cc | 380 em_.reset(new EventMgr(executor_, options.config.gpu_options())); in Init()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 59 EventMgr* event_mgr, int gpu_device_id, const Tensor* input, in Participant()
89 EventMgr* const event_mgr;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 47 class EventMgr; variable
154 EventMgr* event_mgr = nullptr;
|