Searched refs:BufferInterval (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | heap_simulator.h | 370 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;
|
D | heap_simulator.cc | 522 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 …]
|
D | memory_space_assignment.h | 109 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 …]
|
D | memory_space_assignment.cc | 92 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 …]
|
D | memory_space_assignment_utils.h | 29 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval&
|
D | memory_space_assignment_utils.cc | 110 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval) { in IsIntervalAllowedInAlternateMemory()
|
D | memory_space_assignment_test.cc | 1410 [](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 …]
|
D | heap_simulator_test.cc | 1047 BufferInterval buffer_interval_;
|