Searched refs:colocated_buffers (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.cc | 898 const flat_hash_set<const LogicalBuffer*>& colocated_buffers, in AssignBuffersForComputation() argument
984 if (colocated_buffers.contains(buffer)) { in AssignBuffersForComputation()
1696 flat_hash_set<const LogicalBuffer*>* colocated_buffers, in AssignColocatedBufferSets() argument
1739 colocated_buffers->insert(buffer); in AssignColocatedBufferSets()
1776 flat_hash_set<const LogicalBuffer*> colocated_buffers; in CreateAssignment() local
1786 &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()
|
| D | buffer_assignment.h | 578 const absl::flat_hash_set<const LogicalBuffer*>& colocated_buffers,
627 absl::flat_hash_set<const LogicalBuffer*>* colocated_buffers,
|