Searched defs:device_to_host_stream (Results 1 – 3 of 3) sorted by relevance
34 se::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() function
74 std::shared_ptr<se::Stream> device_to_host_stream, in XlaDeviceContext()216 std::shared_ptr<se::Stream> device_to_host_stream; in CopyDeviceTensorToCPU() local
287 std::shared_ptr<se::Stream> device_to_host_stream; in GetDeviceContextLocked() local