Searched refs:colocated_allocations (Results 1 – 2 of 2) sorted by relevance
899 const flat_hash_set<BufferAllocation::Index>& colocated_allocations, in AssignBuffersForComputation() argument1056 if (!colocated_allocations.contains(allocation->index())) { in AssignBuffersForComputation()1087 if (!colocated_allocations.contains(allocation->index())) { in AssignBuffersForComputation()1697 flat_hash_set<BufferAllocation::Index>* colocated_allocations) { in AssignColocatedBufferSets() argument1731 colocated_allocations->insert(allocation->index()); in AssignColocatedBufferSets()1777 flat_hash_set<BufferAllocation::Index> colocated_allocations; in CreateAssignment() local1786 &colocated_buffers, &colocated_allocations); in CreateAssignment()1800 /*is_thread_local=*/false, colocated_buffers, colocated_allocations, in CreateAssignment()1821 /*is_thread_local=*/true, colocated_buffers, colocated_allocations, in CreateAssignment()
579 const absl::flat_hash_set<BufferAllocation::Index>& colocated_allocations,628 absl::flat_hash_set<BufferAllocation::Index>* colocated_allocations);