Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc1364 std::vector<ColocatedBufferSet>* colocated_buffer_sets) { in AddSetToColocatedBufferSets() argument
1374 for (size_t index = 0; index < colocated_buffer_sets->size(); ++index) { in AddSetToColocatedBufferSets()
1376 if ((*colocated_buffer_sets)[index].contains(buffer)) { in AddSetToColocatedBufferSets()
1379 << ColocatedBufferSetsToString((*colocated_buffer_sets)[index], in AddSetToColocatedBufferSets()
1389 colocated_buffer_sets->emplace_back(); in AddSetToColocatedBufferSets()
1390 colocated_buffer_sets->back().insert(colocated_set.begin(), in AddSetToColocatedBufferSets()
1397 ColocatedBufferSet* first = &(*colocated_buffer_sets)[overlap_set_indices[0]]; in AddSetToColocatedBufferSets()
1400 (*colocated_buffer_sets)[overlap_set_indices[index]]; in AddSetToColocatedBufferSets()
1412 colocated_buffer_sets->erase(colocated_buffer_sets->begin() + offset); in AddSetToColocatedBufferSets()
1418 const std::vector<ColocatedBufferSet>& colocated_buffer_sets, in MergeColocatedBufferSets() argument
[all …]
Dbuffer_assignment.h620 std::vector<ColocatedBufferSet>* colocated_buffer_sets);
625 const std::vector<ColocatedBufferSet>& colocated_buffer_sets,
634 std::vector<ColocatedBufferSet>* colocated_buffer_sets);
640 const std::vector<ColocatedBufferSet>& colocated_buffer_sets,