Searched refs:gpu_device_context (Results 1 – 3 of 3) sorted by relevance
545 GPUDeviceContext* gpu_device_context = device_context_; in Compute() local547 gpu_device_context = in Compute()550 se::Stream* stream = gpu_device_context->stream(); in Compute()551 const auto stream_id = gpu_device_context->stream_id(); in Compute()618 GPUDeviceContext* gpu_device_context = device_context_; in ComputeAsync() local620 gpu_device_context = in ComputeAsync()623 se::Stream* stream = gpu_device_context->stream(); in ComputeAsync()624 const auto stream_id = gpu_device_context->stream_id(); in ComputeAsync()
373 auto* gpu_device_context = in SetOutputForConstant() local375 gpu_device_context->stream()->ThenWaitFor( in SetOutputForConstant()376 gpu_device_context->host_to_device_stream()); in SetOutputForConstant()
1747 "gpu_device_context.h",