| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 270 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; in TYPED_TEST() local
273 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
338 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; in TYPED_TEST() local
342 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
382 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; in TYPED_TEST() local
385 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
436 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; in TYPED_TEST() local
440 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
473 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; in TYPED_TEST() local
476 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
[all …]
|
| D | nccl_manager.h | 59 EventMgr* event_mgr, int gpu_device_id, const Tensor* input, in Participant()
63 event_mgr(event_mgr), in Participant()
71 DCHECK(event_mgr != nullptr); in Participant()
89 EventMgr* const event_mgr; member
|
| D | nccl_manager.cc | 637 p->event_mgr->ThenExecute(comm_stream, done_callback); in LoopKernelLaunches()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nccl_ops.cc | 107 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in ComputeAsync()
139 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in ComputeAsync()
176 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in ComputeAsync()
211 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in ComputeAsync()
248 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in ComputeAsync()
|
| D | gpu_device_array.h | 89 context_->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in Finalize()
|
| D | collective_nccl_reducer.cc | 145 compute_stream->parent(), compute_stream, gpu_info->event_mgr, in Run()
|
| D | check_numerics_op.cc | 212 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsync()
|
| D | where_op.cc | 360 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsyncType()
|
| D | dynamic_partition_op_gpu.cu.cc | 319 c->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsync()
|
| D | crop_and_resize_op.cc | 823 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in RunIfBoxIndexIsValid()
|
| D | cuda_solvers.cc | 247 ->event_mgr->ThenExecute(stream, std::move(cb)); in CheckLapackInfoAndDeleteSolverAsync()
|
| D | segment_reduction_ops.cc | 294 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 166 dev_info->event_mgr->ThenExecute( in SetProtoFromGPU()
238 dev_info->event_mgr->ThenExecute( in DeviceToDeviceCopy()
288 dev_info->event_mgr->ThenExecute( in CopyGPUTensorToCPU()
334 dev_info->event_mgr->ThenExecute( in CopyCPUTensorToGPU()
|
| D | gpu_util_platform_specific.cc | 52 gpu_info->event_mgr->ThenExecute(stream, func); in ThenExecute()
|
| D | gpu_device.cc | 397 gpu_device_info_->event_mgr = em_.get(); in Init()
|
| /external/pdfium/third_party/libopenjpeg20/ |
| D | event.h | 89 OPJ_BOOL opj_event_msg(opj_event_mgr_t* event_mgr, OPJ_INT32 event_type,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 154 EventMgr* event_mgr = nullptr; member
|