Searched refs:device_to_host_stream_ (Results 1 – 3 of 3) sorted by relevance
39 device_to_host_stream_(device_to_host_stream), in GPUDeviceContext()46 se::Stream* device_to_host_stream() const { return device_to_host_stream_; } in device_to_host_stream()78 se::Stream* device_to_host_stream_; variable
81 device_to_host_stream_(std::move(device_to_host_stream)), in XlaDeviceContext()217 if (device_to_host_stream_) { in CopyDeviceTensorToCPU()218 device_to_host_stream = device_to_host_stream_; in CopyDeviceTensorToCPU()
101 std::shared_ptr<se::Stream> device_to_host_stream_; variable