Searched refs:companion_sets_ (Results 1 – 2 of 2) sorted by relevance
145 for (const auto& companions : companion_sets_) { in VerifyCompanionSets()393 companion_sets_.push_back( in AddCompanion()395 auto companion_set = companion_sets_.back().get(); in AddCompanion()398 companion_set_index_[instruction1] = companion_sets_.size() - 1; in AddCompanion()399 companion_set_index_[instruction2] = companion_sets_.size() - 1; in AddCompanion()401 companion_sets_[companion_set_index_[instruction2]]->push_back( in AddCompanion()405 companion_sets_[companion_set_index_[instruction1]]->push_back( in AddCompanion()415 *companion_sets_[companion_set_index_[instruction1]])); in AddCompanion()423 companion_sets_[index_to_remove].reset(nullptr); in AddCompanion()537 for (auto& companion_set : companion_sets_) { in DumpCollectedStats()
189 CHECK_LT(index, companion_sets_.size()); in companion_set()190 return *companion_sets_[index]; in companion_set()201 return companion_sets_; in companion_sets()256 std::vector<std::unique_ptr<std::vector<HloInstruction*>>> companion_sets_; variable