Home
last modified time | relevance | path

Searched refs:buffer_live_ranges (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_live_range.h84 const absl::flat_hash_map<const HloValue*, TimeBound>& buffer_live_ranges() in buffer_live_ranges() function
89 absl::flat_hash_map<const HloValue*, TimeBound>& buffer_live_ranges() { in buffer_live_ranges() function
Dheap_simulator.cc190 auto& buffer_live_ranges = hlo_live_range->buffer_live_ranges(); in RunComputation() local
194 if (!buffer_live_ranges.contains(value)) { in RunComputation()
206 const auto& live_range1 = buffer_live_ranges.at(value1); in RunComputation()
207 const auto& live_range2 = buffer_live_ranges.at(value2); in RunComputation()
217 auto live_range = buffer_live_ranges.at(value); in RunComputation()
262 auto it = buffer_live_ranges.find(operand_value); in RunComputation()
263 if (it == buffer_live_ranges.end()) { in RunComputation()
269 auto& user_live_range = buffer_live_ranges[value]; in RunComputation()
Dbuffer_assignment.cc866 const auto& buffer_live_ranges = live_ranges.buffer_live_ranges(); in BufferInfoString() local
881 if (!buffer_live_ranges.contains(&buffer)) { in BufferInfoString()
902 const int64_t end_t = buffer_live_ranges.at(&buffer).end; in BufferInfoString()
977 const auto& buffer_live_ranges = hlo_live_range.buffer_live_ranges(); in LiveRangeInterferes() local
979 auto live_range_it1 = buffer_live_ranges.find(buffer1); in LiveRangeInterferes()
980 CHECK(live_range_it1 != buffer_live_ranges.end()) in LiveRangeInterferes()
983 auto live_range_it2 = buffer_live_ranges.find(buffer2); in LiveRangeInterferes()
984 CHECK(live_range_it2 != buffer_live_ranges.end()) in LiveRangeInterferes()
Dhlo_live_range_test.cc66 return hlo_live_range_->buffer_live_ranges().at(value);
Dmemory_space_assignment.cc4247 hlo_live_range->buffer_live_ranges().at(value); in VerifyAndExportHeapSimulatorTrace()
Dmemory_space_assignment_test.cc3824 hlo_live_range->buffer_live_ranges().at(value); in TEST_P()