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.cc528 x_end = std::max(x_end, buffer_intervals_.at(colocation).end); in GetTemporalBufferIntervalCompare()
554 auto emplace_result = buffer_intervals_.emplace( in Alloc()
568 DCHECK_NE(buffer_intervals_.count(share_with), 0); in ShareWith()
569 buffer_intervals_[share_with].colocations.push_back(buffer); in ShareWith()
570 auto emplace_result = buffer_intervals_.emplace( in ShareWith()
587 worklist.push_back(&buffer_intervals_.at(buffer_colocated)); in GetTransitiveColocations()
602 BufferInterval& buffer_interval = FindOrDie(buffer_intervals_, buffer); in Free()
831 sorted_buffer_intervals.reserve(buffer_intervals_.size()); in GetSortedBufferIntervals()
832 for (auto& entry : buffer_intervals_) { in GetSortedBufferIntervals()
881 std::max(max_colocation_size, buffer_intervals_.at(colocation).size); in FindChunkCandidate()
[all …]
Dheap_simulator.h437 absl::flat_hash_map<const BufferType*, BufferInterval> buffer_intervals_; variable
Dmemory_space_assignment.cc1159 worklist.push_back(&buffer_intervals_.at(buffer_colocated)); in GetSortedColocatedIntervals()
1403 CHECK_EQ((*options_.autotuning_config).size(), buffer_intervals_.size()); in Finish()