Home
last modified time | relevance | path

Searched defs:host_to_device_stream_ (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device_context.h103 std::shared_ptr<se::Stream> host_to_device_stream_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_context.h78 se::Stream* host_to_device_stream_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgpu_device_context.h95 se::Stream* host_to_device_stream_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dexecutable_run_options.h167 stream_executor::Stream* host_to_device_stream_ = nullptr; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dlocal_device_state.h177 std::unique_ptr<se::Stream> host_to_device_stream_; variable