Searched refs:gpu_device_context (Results 1 – 3 of 3) sorted by relevance
653 GPUDeviceContext* gpu_device_context = device_context_; in Compute() local655 gpu_device_context = in Compute()658 se::Stream* stream = gpu_device_context->stream(); in Compute()659 const auto stream_id = gpu_device_context->stream_id(); in Compute()744 GPUDeviceContext* gpu_device_context = device_context_; in ComputeAsync() local746 gpu_device_context = in ComputeAsync()749 se::Stream* stream = gpu_device_context->stream(); in ComputeAsync()750 const auto stream_id = gpu_device_context->stream_id(); in ComputeAsync()
408 auto* gpu_device_context = in SetOutputForConstant() local410 gpu_device_context->stream()->ThenWaitFor( in SetOutputForConstant()411 gpu_device_context->host_to_device_stream()); in SetOutputForConstant()
1971 "gpu_device_context.h",