Home
last modified time | relevance | path

Searched refs:event_mgr (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.h73 event_mgr(info->event_mgr), in Participant()
84 DCHECK(event_mgr != nullptr); in Participant()
102 EventMgr* const event_mgr; member
Dnccl_manager.cc848 p->event_mgr->ThenExecute(comm_stream, done_callback); in LoopKernelLaunches()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr.cc259 auto event_mgr = new EventMgr(se, gpu_options); in GetEventMgr() local
260 event_mgr_map_[se] = event_mgr; in GetEventMgr()
261 return event_mgr; in GetEventMgr()
Dgpu_util.cc165 dev_info->event_mgr->ThenExecute( in SetProtoFromGPU()
237 dev_info->event_mgr->ThenExecute( in DeviceToDeviceCopy()
287 dev_info->event_mgr->ThenExecute( in CopyGPUTensorToCPU()
335 dev_info->event_mgr->ThenExecute( in CopyCPUTensorToGPU()
Dgpu_util_platform_specific.cc54 gpu_info->event_mgr->ThenExecute(stream, func); in ThenExecute()
Dgpu_device.cc475 gpu_device_info_->event_mgr = em_; in Init()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr.cc260 auto event_mgr = new EventMgr(se, gpu_options); in GetEventMgr() local
261 event_mgr_map_[se] = event_mgr; in GetEventMgr()
262 return event_mgr; in GetEventMgr()
Ddevice_event_mgr_test.cc163 EventMgr* event_mgr() { return gpu_->em_; } in event_mgr() function in tensorflow::GPUDeviceTestHelper
396 gpu_helper_->event_mgr()->ThenExecute( in DoAddChain()
407 gpu_helper_->event_mgr()->ThenExecute(gpu_helper_->compute_stream(), in DoAddChain()
413 gpu_helper_->event_mgr()->ThenExecute( in DoAddChain()
424 gpu_helper_->event_mgr()->ThenExecute(gpu_helper_->d2h_stream(), in DoAddChain()
427 gpu_helper_->event_mgr()->ThenExecute( in DoAddChain()
/external/pdfium/third_party/libopenjpeg20/
Devent.h89 OPJ_BOOL opj_event_msg(opj_event_mgr_t* event_mgr, OPJ_INT32 event_type,
/external/tensorflow/tensorflow/core/kernels/
Dgpu_device_array.h91 context_->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in Finalize()
Ddebug_ops.h753 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute(
780 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute(
812 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute(
845 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute(
871 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute(
Dcheck_numerics_op.cc274 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsync()
Dwhere_op.cc368 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsyncType()
Ddynamic_partition_op_gpu.cu.cc329 c->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsync()
Dsegment_reduction_ops_impl.h305 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsync()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h157 EventMgr* event_mgr = nullptr; member
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_tensor_to_csr_sparse_matrix_op.cc284 c->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsync()
Ddense_to_csr_sparse_matrix_op.cc351 c->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/image/
Dcrop_and_resize_op.cc865 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in RunIfBoxIndexIsValid()
/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc268 ->event_mgr->ThenExecute(stream, std::move(cb)); in CheckLapackInfoAndDeleteSolverAsync()