Searched refs:ThenExecute (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 48 Status GPUDeviceContext::ThenExecute(Device* device, se::Stream* stream, in ThenExecute() function in tensorflow::GPUDeviceContext
52 gpu_info->event_mgr->ThenExecute(stream, func); in ThenExecute()
|
| 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_event_mgr.h | 99 inline void ThenExecute(se::Stream* stream, std::function<void()> func) { in ThenExecute() function
|
| D | gpu_event_mgr_test.cc | 259 em.ThenExecute(stream.get(), [&hit, ¬e]() { in TEST()
|
| D | gpu_device.cc | 589 em_->ThenExecute(stream, [op_kernel, tracker, queued_count]() { in ComputeHelper()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gpu_device_context.h | 67 Status ThenExecute(Device* device, se::Stream* stream,
|
| D | ring_gatherer.cc | 143 Status s = gpu_info->default_context->ThenExecute( in RunAsyncParts()
|
| D | ring_reducer.cc | 179 Status s = gpu_info->default_context->ThenExecute( in RunAsyncParts()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_device_array.h | 89 context_->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in Finalize()
|
| 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/framework/ |
| D | device_base.h | 105 virtual Status ThenExecute(Device* device, stream_executor::Stream* stream, in ThenExecute() function
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 637 p->event_mgr->ThenExecute(comm_stream, done_callback); in LoopKernelLaunches()
|