HomeSort by relevance Sort by last modified time
    Searched defs:event_mgr (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/nccl/kernels/
nccl_manager_test.cc 196 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; local
200 device->gpu_id(), event_mgr, stream, &test_case->ins[device_num],
261 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr;
266 device->executor(), device->gpu_id(), event_mgr, stream,
nccl_manager.cc 95 Participant(const Tensor* in_t, Tensor* out_t, EventMgr* event_mgr,
101 event_mgr(event_mgr),
107 DCHECK(event_mgr != nullptr);
119 EventMgr* const event_mgr; member in struct:tensorflow::NcclManager::Participant
302 int gpu_device_id, EventMgr* event_mgr,
307 new Participant(in_t, out_t, event_mgr, tensor_stream, executor,
316 EventMgr* event_mgr, perftools::gputools::Stream* tensor_stream,
319 new Participant(in_t, nullptr /* out_t */, event_mgr, tensor_stream,
329 EventMgr* event_mgr, perftools::gputools::Stream* tensor_stream
    [all...]
  /external/tensorflow/tensorflow/core/framework/
device_base.h 129 // "event_mgr" is used to delay deallocation of temporary GPU buffers.
135 EventMgr* event_mgr = nullptr; member in struct:tensorflow::DeviceBase::GpuDeviceInfo

Completed in 79 milliseconds