Searched defs:host_to_device_stream (Results 1 – 8 of 8) sorted by relevance
36 se::Stream* host_to_device_stream, in GPUDeviceContext()55 se::Stream* host_to_device_stream() const { return host_to_device_stream_; } in host_to_device_stream() function
75 se::Stream* host_to_device_stream() const { in host_to_device_stream() function
78 std::shared_ptr<se::Stream> host_to_device_stream, in XlaDeviceContext()
302 std::shared_ptr<se::Stream> host_to_device_stream; in GetDeviceContextLocked() local
107 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
421 stream_executor::Stream* host_to_device_stream, in TPUExecute()
166 SE_Stream* host_to_device_stream; member