Searched refs:gpu_device_context (Results 1 – 3 of 3) sorted by relevance
516 GPUDeviceContext* gpu_device_context = device_contexts_[0]; in ComputeHelper() local518 gpu_device_context = in ComputeHelper()521 se::Stream* stream = gpu_device_context->stream(); in ComputeHelper()522 const auto stream_id = gpu_device_context->stream_id(); in ComputeHelper()603 GPUDeviceContext* gpu_device_context = device_contexts_[0]; in ConsumeListOfAccessedTensors() local605 gpu_device_context = static_cast<GPUDeviceContext*>(device_context); in ConsumeListOfAccessedTensors()607 se::Stream* stream = gpu_device_context->stream(); in ConsumeListOfAccessedTensors()618 GPUDeviceContext* gpu_device_context = device_contexts_[0]; in ComputeAsync() local620 gpu_device_context = in ComputeAsync()623 se::Stream* stream = gpu_device_context->stream(); in ComputeAsync()[all …]
331 auto* gpu_device_context = in PopulateOutputs() local333 gpu_device_context->stream()->ThenWaitFor( in PopulateOutputs()334 gpu_device_context->host_to_device_stream()); in PopulateOutputs()
3331 "common_runtime/gpu_device_context.h",