Home
last modified time | relevance | path

Searched refs:device_to_host_stream_ (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_context.h39 device_to_host_stream_(device_to_host_stream), in PluggableDeviceContext()
46 se::Stream* device_to_host_stream() const { return device_to_host_stream_; } in device_to_host_stream()
80 se::Stream* device_to_host_stream_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgpu_device_context.h46 device_to_host_stream_(device_to_host_stream), in GPUDeviceContext()
58 se::Stream* device_to_host_stream() const { return device_to_host_stream_; } in device_to_host_stream()
97 se::Stream* device_to_host_stream_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device_context.cc97 device_to_host_stream_(std::move(device_to_host_stream)), in XlaDeviceContext()
227 if (device_to_host_stream_) { in CopyDeviceTensorToCPU()
228 device_to_host_stream = device_to_host_stream_; in CopyDeviceTensorToCPU()
H A Dxla_device_context.h107 std::shared_ptr<se::Stream> device_to_host_stream_; variable