Home
last modified time | relevance | path

Searched defs:device_to_device_streams_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device_context.h110 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dlocal_device_state.h179 std::vector<std::unique_ptr<se::Stream>> device_to_device_streams_; variable