Searched defs:event_mgr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| D | nccl_manager.cc | 119 EventMgr* const event_mgr; member
302 int gpu_device_id, EventMgr* event_mgr, in AddToAllReduce()
316 EventMgr* event_mgr, perftools::gputools::Stream* tensor_stream, in AddBroadcastSend()
329 EventMgr* event_mgr, perftools::gputools::Stream* tensor_stream, in AddBroadcastRecv()
341 int gpu_device_id, EventMgr* event_mgr, in AddReduceSend()
355 int gpu_device_id, EventMgr* event_mgr, in AddReduceRecv()
|
| D | nccl_manager_test.cc | 196 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; in TEST_F() local
261 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; in TEST_F() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 135 EventMgr* event_mgr = nullptr; member
|