Home
last modified time | relevance | path

Searched refs:ColocatedBufferSet (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h612 using ColocatedBufferSet = absl::flat_hash_set<const LogicalBuffer*>; variable
620 std::vector<ColocatedBufferSet>* colocated_buffer_sets);
625 const std::vector<ColocatedBufferSet>& colocated_buffer_sets,
634 std::vector<ColocatedBufferSet>* colocated_buffer_sets);
639 std::vector<ColocatedBufferSet> MergeColocatedBufferSets(
640 const std::vector<ColocatedBufferSet>& colocated_buffer_sets,
Dbuffer_assignment.cc1364 std::vector<ColocatedBufferSet>* colocated_buffer_sets) { in AddSetToColocatedBufferSets()
1397 ColocatedBufferSet* first = &(*colocated_buffer_sets)[overlap_set_indices[0]]; in AddSetToColocatedBufferSets()
1399 const ColocatedBufferSet& overlap_set = in AddSetToColocatedBufferSets()
1416 std::vector<BufferAssigner::ColocatedBufferSet>
1418 const std::vector<ColocatedBufferSet>& colocated_buffer_sets, in MergeColocatedBufferSets()
1515 std::vector<ColocatedBufferSet> new_colocated_buffer_sets(num_sets); in MergeColocatedBufferSets()
1533 std::vector<ColocatedBufferSet>* colocated_buffer_sets) { in BuildColocatedBufferSets()
1685 std::vector<ColocatedBufferSet> new_colocated_buffer_sets = in BuildColocatedBufferSets()
1694 const std::vector<ColocatedBufferSet>& colocated_buffer_sets, in AssignColocatedBufferSets()
1698 for (const ColocatedBufferSet& colocated_buffer_set : colocated_buffer_sets) { in AssignColocatedBufferSets()
[all …]