Searched refs:device_to_host_stream_ (Results 1 – 3 of 3) sorted by relevance
45 device_to_host_stream_(device_to_host_stream), in GPUDeviceContext()56 se::Stream* device_to_host_stream() const { return device_to_host_stream_; } in device_to_host_stream()92 se::Stream* device_to_host_stream_; variable
86 device_to_host_stream_(std::move(device_to_host_stream)), in XlaDeviceContext()225 if (device_to_host_stream_) { in CopyDeviceTensorToCPU()226 device_to_host_stream = device_to_host_stream_; in CopyDeviceTensorToCPU()
104 std::shared_ptr<se::Stream> device_to_host_stream_; variable