Home
last modified time | relevance | path

Searched refs:BufferInterval (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.h370 struct BufferInterval { struct
388 std::function<bool(const BufferInterval&, const BufferInterval&)>; argument
415 std::vector<BufferInterval> GetSortedBufferIntervals() const;
425 ChunkCandidate FindChunkCandidate(const BufferInterval& buffer_interval,
427 void CommitChunk(const BufferInterval& buffer_interval,
440 absl::flat_hash_map<const BufferType*, BufferInterval> buffer_intervals_;
456 const BufferInterval& interval) const;
Dheap_simulator.cc522 return [&](const BufferInterval& x, const BufferInterval& y) { in GetTemporalBufferIntervalCompare()
548 return [&](const BufferInterval& x, const BufferInterval& y) { in GetSpatialBufferIntervalCompare()
569 buffer, BufferInterval{buffer, size, current_time_, -1, {}, true}); in Alloc()
585 buffer, BufferInterval{buffer, size, current_time_, -1, {}, false}); in ShareWith()
593 const BufferInterval& interval) const { in GetTransitiveColocations()
595 std::vector<const BufferInterval*> worklist = {&interval}; in GetTransitiveColocations()
597 const BufferInterval* item = worklist.back(); in GetTransitiveColocations()
615 BufferInterval& buffer_interval = FindOrDie(buffer_intervals_, buffer); in Free()
819 std::vector<BufferInterval> sorted_buffer_intervals = in Finish()
841 typename GlobalDecreasingSizeBestFitHeap<BufferType>::BufferInterval>
[all …]
Dmemory_space_assignment.h109 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval,
241 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval) in BufferIntervalAlternateMemoryBenefit()
344 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval)
391 using BufferInterval = variable
392 GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval;
967 HloModule* module, absl::optional<BufferInterval> prefetch_candidate);
975 std::vector<const BufferInterval*> GetSortedColocatedIntervals(
976 const BufferInterval& interval) const;
981 const BufferInterval& buffer_interval,
987 absl::Span<const AlternateMemoryBestFitHeap::BufferInterval* const>
[all …]
Dmemory_space_assignment.cc92 absl::optional<MemorySpaceAssignment::BufferInterval>
96 std::vector<MemorySpaceAssignment::BufferInterval> candidates; in FindCrossProgramPrefetchCandidate()
101 MemorySpaceAssignment::BufferInterval interval; in FindCrossProgramPrefetchCandidate()
186 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval, in GetMemoryBoundedness()
701 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval) in BufferIntervalAlternateMemoryBenefit()
742 const AlternateMemoryBestFitHeap::BufferInterval& buffer_interval, in CreateAllocationValues()
866 std::vector<const AlternateMemoryBestFitHeap::BufferInterval*>
868 const AlternateMemoryBestFitHeap::BufferInterval& interval) const { in GetSortedColocatedIntervals()
869 std::vector<const BufferInterval*> colocated_intervals; in GetSortedColocatedIntervals()
870 std::vector<const BufferInterval*> worklist = {&interval}; in GetSortedColocatedIntervals()
[all …]
Dmemory_space_assignment_utils.h29 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval&
Dmemory_space_assignment_utils.cc110 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval) { in IsIntervalAllowedInAlternateMemory()
Dmemory_space_assignment_test.cc1410 [](const MemorySpaceAssignment::BufferInterval& a, in TEST_P()
1411 const MemorySpaceAssignment::BufferInterval& b) { in TEST_P()
4012 [](const MemorySpaceAssignment::BufferInterval& a, in TEST_P()
4013 const MemorySpaceAssignment::BufferInterval& b) { in TEST_P()
4081 [](const MemorySpaceAssignment::BufferInterval& a, in TEST_P()
4082 const MemorySpaceAssignment::BufferInterval& b) { in TEST_P()
4228 [](const MemorySpaceAssignment::BufferInterval& a, in TEST_P()
4229 const MemorySpaceAssignment::BufferInterval& b) { in TEST_P()
4506 [](const MemorySpaceAssignment::BufferInterval& a, in TEST_P()
4507 const MemorySpaceAssignment::BufferInterval& b) { in TEST_P()
[all …]
Dheap_simulator_test.cc1047 BufferInterval buffer_interval_;