Searched refs:device_to_host_streams_ (Results 1 – 2 of 2) sorted by relevance
52 device_to_host_streams_.reserve(kNumDeviceToHostStreams); in LocalDeviceState()56 device_to_host_streams_.push_back(std::move(stream)); in LocalDeviceState()90 for (auto& stream : device_to_host_streams_) { in SynchronizeAllActivity()135 (next_device_to_host_stream_ + 1) % device_to_host_streams_.size(); in GetDeviceToHostStream()136 return device_to_host_streams_.at(i).get(); in GetDeviceToHostStream()
178 std::vector<std::unique_ptr<se::Stream>> device_to_host_streams_; variable