Searched refs:device_to_device_streams_ (Results 1 – 6 of 6) sorted by relevance
54 device_to_device_streams_.reserve(kNumDeviceToDeviceStreams); in LocalDeviceState()58 device_to_device_streams_.push_back(std::move(stream)); in LocalDeviceState()118 (next_device_to_device_stream_ + 1) % device_to_device_streams_.size(); in GetDeviceToDeviceStream()119 return device_to_device_streams_.at(i).get(); in GetDeviceToDeviceStream()
179 std::vector<std::unique_ptr<se::Stream>> device_to_device_streams_; variable
79 return device_to_device_streams_.at(index).get(); in device_to_device_stream()107 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams_; variable
87 device_to_device_streams_(std::move(device_to_device_streams)), in XlaDeviceContext()290 DCHECK_GT(device_to_device_streams_.size(), 0); in GetDeviceToDeviceStream()293 next_stream_ = (next_stream_ + 1) % device_to_device_streams_.size(); in GetDeviceToDeviceStream()
225 device_to_device_streams_.resize(kNumDeviceToDeviceStreams); in XlaDevice()309 for (std::shared_ptr<se::Stream>& stream : device_to_device_streams_) { in GetDeviceContextLocked()315 device_to_device_streams = device_to_device_streams_; in GetDeviceContextLocked()
242 std::vector<std::shared_ptr<se::Stream>> device_to_device_streams_