Searched defs:host_to_device_stream (Results 1 – 9 of 9) sorted by relevance
33 int stream_id, se::Stream* stream, se::Stream* host_to_device_stream, in PluggableDeviceContext()45 se::Stream* host_to_device_stream() const { return host_to_device_stream_; } in host_to_device_stream() function
36 se::Stream* host_to_device_stream, in GPUDeviceContext()57 se::Stream* host_to_device_stream() const { return host_to_device_stream_; } in host_to_device_stream() function
77 se::Stream* host_to_device_stream() const { in host_to_device_stream() function
89 std::shared_ptr<se::Stream> host_to_device_stream, in XlaDeviceContext()
331 std::shared_ptr<se::Stream> host_to_device_stream; in GetDeviceContextLocked() local
108 se::Stream* host_to_device_stream() const { in host_to_device_stream() function
70 stream_executor::Stream* ExecutableRunOptions::host_to_device_stream() const { in host_to_device_stream() function in xla::ExecutableRunOptions
425 stream_executor::Stream* host_to_device_stream, in TPUExecute()
163 SE_Stream* host_to_device_stream; member