Searched refs:extra_users (Results 1 – 1 of 1) sorted by relevance
428 absl::flat_hash_map<const HloInstruction*, int64> extra_users; in DFSMemoryScheduler() local432 extra_users[hlo] = 0; in DFSMemoryScheduler()441 extra_users[hlo] = hlo->users().empty() ? 0 : hlo->users().size() - 1; in DFSMemoryScheduler()449 extra_users[hlo] += extra_users[operand]; in DFSMemoryScheduler()462 extra_users[hlo] = std::min(extra_users[hlo], total_hlos); in DFSMemoryScheduler()464 CHECK_EQ(extra_users.size(), computation->instruction_count()); in DFSMemoryScheduler()476 &visitor, [&extra_users, &total_sizes](const HloInstruction* a, in DFSMemoryScheduler()478 if (extra_users[a] != extra_users[b]) { in DFSMemoryScheduler()479 return extra_users[a] > extra_users[b]; in DFSMemoryScheduler()