Searched refs:buffers_to_assign (Results 1 – 3 of 3) sorted by relevance
100 buffers_to_assign(nullptr) {} in Options()109 const absl::flat_hash_set<const HloValue*>* buffers_to_assign; member
1443 const flat_hash_set<const HloValue*>& buffers_to_assign = pair.second; in AssignBuffersWithSequentialOrdering() local1448 all_buffers_to_assign.insert(buffers_to_assign.begin(), in AssignBuffersWithSequentialOrdering()1449 buffers_to_assign.end()); in AssignBuffersWithSequentialOrdering()1458 options.buffers_to_assign = &single_colored_set.second; in AssignBuffersWithSequentialOrdering()1475 const flat_hash_set<const HloValue*>& buffers_to_assign = pair.second; in AssignBuffersWithSequentialOrdering() local1479 auto color_map = SplitBuffersByColor(buffers_to_assign); in AssignBuffersWithSequentialOrdering()1485 options.buffers_to_assign = &single_colored_set.second; in AssignBuffersWithSequentialOrdering()
361 return options_.buffers_to_assign != nullptr && in IgnoreBuffer()362 !options_.buffers_to_assign->contains(buffer); in IgnoreBuffer()420 if (options_.buffers_to_assign != nullptr) { in Finish()421 CHECK_EQ(options_.buffers_to_assign->size(), total_chunk_count); in Finish()