Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc68 std::vector<const LogicalBuffer*>* colocated_set) { in AddBufferToColocatedSet() argument
72 colocated_set->push_back(points_to.element(index)[0]); in AddBufferToColocatedSet()
73 return colocated_set->back(); in AddBufferToColocatedSet()
1363 const std::vector<const LogicalBuffer*>& colocated_set, in AddSetToColocatedBufferSets() argument
1365 if (colocated_set.empty()) { in AddSetToColocatedBufferSets()
1368 VLOG(5) << ColocatedBufferSetsToString(colocated_set, in AddSetToColocatedBufferSets()
1375 for (const LogicalBuffer* buffer : colocated_set) { in AddSetToColocatedBufferSets()
1390 colocated_buffer_sets->back().insert(colocated_set.begin(), in AddSetToColocatedBufferSets()
1391 colocated_set.end()); in AddSetToColocatedBufferSets()
1403 first->insert(colocated_set.begin(), colocated_set.end()); in AddSetToColocatedBufferSets()
[all …]
Dbuffer_assignment.h633 const std::vector<const LogicalBuffer*>& colocated_set,