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.cc304 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams; in GetDeviceContextLocked() local
315 device_to_device_streams = device_to_device_streams_; in GetDeviceContextLocked()
323 device_to_device_streams = {stream_}; in GetDeviceContextLocked()
345 device_to_device_streams, client, shape_representation_fn_, in GetDeviceContextLocked()
352 std::move(device_to_host_stream), std::move(device_to_device_streams), in GetDeviceContextLocked()
Dxla_device_context.cc80 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams, in XlaDeviceContext() argument
87 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,