Searched refs:total_sizes (Results 1 – 1 of 1) sorted by relevance
429 absl::flat_hash_map<const HloInstruction*, int64> total_sizes; in DFSMemoryScheduler() local433 total_sizes[hlo] = 0; in DFSMemoryScheduler()444 total_sizes[hlo] = logical_buffer_size; in DFSMemoryScheduler()450 total_sizes[hlo] += total_sizes[operand]; in DFSMemoryScheduler()461 total_sizes[hlo] = std::min(total_sizes[hlo], cumulative_total_size); in DFSMemoryScheduler()465 CHECK_EQ(total_sizes.size(), computation->instruction_count()); in DFSMemoryScheduler()476 &visitor, [&extra_users, &total_sizes](const HloInstruction* a, in DFSMemoryScheduler()481 if (total_sizes[a] != total_sizes[b]) { in DFSMemoryScheduler()482 return total_sizes[a] > total_sizes[b]; in DFSMemoryScheduler()