Searched refs:extra_users (Results 1 – 1 of 1) sorted by relevance
446 absl::flat_hash_map<const HloInstruction*, int64> extra_users; in DFSMemoryScheduler() local450 extra_users[hlo] = 0; in DFSMemoryScheduler()459 extra_users[hlo] = hlo->users().empty() ? 0 : hlo->users().size() - 1; in DFSMemoryScheduler()467 extra_users[hlo] += extra_users[operand]; in DFSMemoryScheduler()480 extra_users[hlo] = std::min(extra_users[hlo], total_hlos); in DFSMemoryScheduler()482 CHECK_EQ(extra_users.size(), computation->instruction_count()); in DFSMemoryScheduler()495 &visitor, [&extra_users, &total_sizes](const HloInstruction* a, in DFSMemoryScheduler()497 if (extra_users[a] != extra_users[b]) { in DFSMemoryScheduler()498 return extra_users[a] > extra_users[b]; in DFSMemoryScheduler()