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.h77 const absl::flat_hash_map<const HloValue*, TimeBound>& buffer_live_ranges() in buffer_live_ranges() function
82 absl::flat_hash_map<const HloValue*, TimeBound>& buffer_live_ranges() { in buffer_live_ranges() function
Dheap_simulator.cc199 auto& buffer_live_ranges = hlo_live_range->buffer_live_ranges(); in RunComputation() local
203 const auto& live_range1 = buffer_live_ranges.at(value1); in RunComputation()
204 const auto& live_range2 = buffer_live_ranges.at(value2); in RunComputation()
214 auto live_range = buffer_live_ranges.at(value); in RunComputation()
259 auto it = buffer_live_ranges.find(operand_value); in RunComputation()
260 if (it == buffer_live_ranges.end()) { in RunComputation()
266 auto& user_live_range = buffer_live_ranges[value]; in RunComputation()
Dbuffer_assignment.cc819 const auto& buffer_live_ranges = live_ranges.buffer_live_ranges(); in BufferInfoString() local
834 if (!buffer_live_ranges.contains(&buffer)) { in BufferInfoString()
855 const int64 end_t = buffer_live_ranges.at(&buffer).end; in BufferInfoString()
930 const auto& buffer_live_ranges = hlo_live_range.buffer_live_ranges(); in LiveRangeInterferes() local
932 CHECK(buffer_live_ranges.contains(buffer1)) in LiveRangeInterferes()
935 CHECK(buffer_live_ranges.contains(buffer2)) in LiveRangeInterferes()
948 auto live_range_1 = buffer_live_ranges.at(buffer1); in LiveRangeInterferes()
949 auto live_range_2 = buffer_live_ranges.at(buffer2); in LiveRangeInterferes()
Dhlo_live_range_test.cc66 return hlo_live_range_->buffer_live_ranges().at(value);
Dmemory_space_assignment.cc3332 hlo_live_range->buffer_live_ranges().at(value); in VerifyAndExportHeapSimulatorTrace()
Dmemory_space_assignment_test.cc3863 hlo_live_range->buffer_live_ranges().at(value); in TEST_P()