Home
last modified time | relevance | path

Searched refs:EventMgr (Results 1 – 7 of 7) 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()
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 …]
Dgpu_event_mgr.cc92 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()
Dgpu_event_mgr.h64 class EventMgr {
66 EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
68 ~EventMgr();
Dgpu_device.h155 std::unique_ptr<EventMgr> em_;
Dgpu_device.cc380 em_.reset(new EventMgr(executor_, options.config.gpu_options())); in Init()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.h59 EventMgr* event_mgr, int gpu_device_id, const Tensor* input, in Participant()
89 EventMgr* const event_mgr;
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h47 class EventMgr; variable
154 EventMgr* event_mgr = nullptr;