Home
last modified time | relevance | path

Searched refs:buffer_intervals_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_best_fit_repacker.cc44 return buffer_intervals_.contains(search); in ImportAllocationBlocks()
47 buffer_intervals_[*aliased_it].colocations.push_back(allocation_block); in ImportAllocationBlocks()
50 buffer_intervals_[allocation_block] = {allocation_block, in ImportAllocationBlocks()
Dheap_simulator.cc525 x_end = std::max(x_end, buffer_intervals_.at(colocation).end); in GetTemporalBufferIntervalCompare()
530 y_end = std::max(y_end, buffer_intervals_.at(colocation).end); in GetTemporalBufferIntervalCompare()
568 auto emplace_result = buffer_intervals_.emplace( in Alloc()
582 DCHECK_NE(buffer_intervals_.count(share_with), 0); in ShareWith()
583 buffer_intervals_[share_with].colocations.push_back(buffer); in ShareWith()
584 auto emplace_result = buffer_intervals_.emplace( in ShareWith()
601 worklist.push_back(&buffer_intervals_.at(buffer_colocated)); in GetTransitiveColocations()
615 BufferInterval& buffer_interval = FindOrDie(buffer_intervals_, buffer); in Free()
844 for (auto& entry : buffer_intervals_) { in GetSortedBufferIntervals()
883 auto colocation_interval = buffer_intervals_.at(colocation); in FindChunkCandidate()
[all …]
Dheap_simulator.h440 absl::flat_hash_map<const BufferType*, BufferInterval> buffer_intervals_; variable
Dmemory_space_assignment.cc876 worklist.push_back(&buffer_intervals_.at(buffer_colocated)); in GetSortedColocatedIntervals()