Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc869 std::vector<const BufferInterval*> colocated_intervals; in GetSortedColocatedIntervals() local
874 colocated_intervals.push_back(item); in GetSortedColocatedIntervals()
880 absl::c_stable_sort(colocated_intervals, [&](const BufferInterval* x, in GetSortedColocatedIntervals()
884 return colocated_intervals; in GetSortedColocatedIntervals()
1117 auto colocated_intervals = GetSortedColocatedIntervals(interval); in Finish() local
1118 if (AreIntervalsReservedInAlternateMemory(colocated_intervals)) { in Finish()
1155 auto colocated_intervals = GetSortedColocatedIntervals(interval); in Finish() local
1157 if (AreIntervalsReservedInAlternateMemory(colocated_intervals)) { in Finish()
1160 for (const BufferInterval* colocated_interval : colocated_intervals) { in Finish()
1178 if (colocated_intervals.size() > 1 && in Finish()
[all …]
Dmemory_space_assignment.h988 colocated_intervals,
1211 colocated_intervals);
1235 absl::Span<const BufferInterval* const> colocated_intervals) const;