Home
last modified time | relevance | path

Searched defs: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.h34 se::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() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgpu_device_context.h37 se::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() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device_context.cc90 std::shared_ptr<se::Stream> device_to_host_stream, in XlaDeviceContext()
226 std::shared_ptr<se::Stream> device_to_host_stream; in CopyDeviceTensorToCPU() local
H A Dxla_device.cc332 std::shared_ptr<se::Stream> device_to_host_stream; in GetDeviceContextLocked() local