Searched refs:companion_set (Results 1 – 2 of 2) sorted by relevance
395 auto companion_set = companion_sets_.back().get(); in AddCompanion() local396 companion_set->push_back(instruction1); in AddCompanion()397 companion_set->push_back(instruction2); in AddCompanion()537 for (auto& companion_set : companion_sets_) { in DumpCollectedStats() local539 for (HloInstruction* instruction : *companion_set) { in DumpCollectedStats()
184 return companion_set(companion_set_index_.at(instruction)); in Companions()188 const std::vector<HloInstruction*>& companion_set(int64 index) const { in companion_set() function