Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc288 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams; in GetDeviceContextLocked() local
299 device_to_device_streams = device_to_device_streams_; in GetDeviceContextLocked()
307 device_to_device_streams = {stream_}; in GetDeviceContextLocked()
326 std::move(device_to_host_stream), std::move(device_to_device_streams), in GetDeviceContextLocked()
Dxla_device_context.cc75 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams, in XlaDeviceContext() argument
82 device_to_device_streams_(std::move(device_to_device_streams)), in XlaDeviceContext()
Dxla_device_context.h58 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams,