Home
last modified time | relevance | path

Searched refs:device_to_device_streams (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device.cc333 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams; in GetDeviceContextLocked() local
344 device_to_device_streams = device_to_device_streams_; in GetDeviceContextLocked()
352 device_to_device_streams = {stream_}; in GetDeviceContextLocked()
372 device_to_device_streams, client, iter, thread_pool_.get()); in GetDeviceContextLocked()
H A Dxla_device_context.cc91 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams, in XlaDeviceContext() argument
98 device_to_device_streams_(std::move(device_to_device_streams)), in XlaDeviceContext()
H A Dxla_device_context.h60 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams,