Searched defs:host_to_device_stream (Results 1 – 5 of 5) sorted by relevance
33 se::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
75 se::Stream* host_to_device_stream() const { in host_to_device_stream() function
73 std::shared_ptr<se::Stream> host_to_device_stream, in XlaDeviceContext()
286 std::shared_ptr<se::Stream> host_to_device_stream; in GetDeviceContextLocked() local
54 stream_executor::Stream* ExecutableRunOptions::host_to_device_stream() const { in host_to_device_stream() function in xla::ExecutableRunOptions