Home
last modified time | relevance | path

Searched refs:instruction_schedule (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc379 cost_analysis.hlo_live_range().instruction_schedule().size(), 0), in CostAnalysisPrefetchIntervalPicker()
381 cost_analysis.hlo_live_range().instruction_schedule().size(), 0), in CostAnalysisPrefetchIntervalPicker()
388 &cost_analysis_.hlo_live_range().instruction_schedule(); in CostAnalysisPrefetchIntervalPicker()
754 instruction_schedule = hlo_live_range_.instruction_schedule(); in CreateAllocationValues() local
766 return instruction_schedule.at(pos1.instruction) < in CreateAllocationValues()
767 instruction_schedule.at(pos2.instruction); in CreateAllocationValues()
780 return instruction_schedule.at(use1.instruction) < in CreateAllocationValues()
781 instruction_schedule.at(use2.instruction); in CreateAllocationValues()
792 int64 use_time = instruction_schedule.at(use.instruction); in CreateAllocationValues()
799 instruction_schedule.at( in CreateAllocationValues()
[all …]
Dhlo_live_range.h72 instruction_schedule() const { in instruction_schedule() function
Dhlo_live_range_test.cc75 hlo_live_range_->instruction_schedule().size()); in CheckSchedule()
76 for (const auto& inst_and_time : hlo_live_range_->instruction_schedule()) { in CheckSchedule()
Dbuffer_assignment.cc818 const auto& instruction_schedule = live_ranges.instruction_schedule(); in BufferInfoString() local
841 uses.emplace_back(instruction_schedule.at(use.instruction), in BufferInfoString()
854 instruction_schedule.at(buffer.defining_position().instruction); in BufferInfoString()
Dheap_simulator.cc189 if (hlo_live_range->instruction_schedule().count( in RunComputation()