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;
|