Searched refs:host_to_device_stream (Results 1 – 8 of 8) sorted by relevance
33 se::Stream* host_to_device_stream, in GPUDeviceContext() argument38 host_to_device_stream_(host_to_device_stream), in GPUDeviceContext()45 se::Stream* host_to_device_stream() const { return host_to_device_stream_; } in host_to_device_stream() function
56 std::shared_ptr<se::Stream> host_to_device_stream,75 se::Stream* host_to_device_stream() const { in host_to_device_stream() function
286 std::shared_ptr<se::Stream> host_to_device_stream; in GetDeviceContextLocked() local298 host_to_device_stream = host_to_device_stream_; in GetDeviceContextLocked()305 host_to_device_stream = stream_; in GetDeviceContextLocked()325 stream_, std::move(host_to_device_stream), in GetDeviceContextLocked()
73 std::shared_ptr<se::Stream> host_to_device_stream, in XlaDeviceContext() argument80 host_to_device_stream_(std::move(host_to_device_stream)), in XlaDeviceContext()
334 gpu_device_context->host_to_device_stream()); in PopulateOutputs()
54 stream_executor::Stream* ExecutableRunOptions::host_to_device_stream() const { in host_to_device_stream() function in xla::ExecutableRunOptions
64 stream_executor::Stream* host_to_device_stream() const;
316 ->host_to_device_stream(); in CopyCPUTensorToGPU()