Searched refs:device_to_device_streams_ (Results 1 – 4 of 4) sorted by relevance
79 return device_to_device_streams_.at(index).get(); in device_to_device_stream()104 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams_; variable
82 device_to_device_streams_(std::move(device_to_device_streams)), in XlaDeviceContext()260 DCHECK_GT(device_to_device_streams_.size(), 0); in GetDeviceToDeviceStream()263 next_stream_ = (next_stream_ + 1) % device_to_device_streams_.size(); in GetDeviceToDeviceStream()
217 device_to_device_streams_.resize(kNumDeviceToDeviceStreams); in XlaDevice()293 for (std::shared_ptr<se::Stream>& stream : device_to_device_streams_) { in GetDeviceContextLocked()299 device_to_device_streams = device_to_device_streams_; in GetDeviceContextLocked()
223 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams_