Home
last modified time | relevance | path

Searched refs:hlo_live_range (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_live_range.cc25 std::unique_ptr<HloLiveRange> hlo_live_range( in Run() local
27 hlo_live_range->schedule_end_time_ = in Run()
28 hlo_live_range->FlattenSchedule(*computation, 0); in Run()
29 hlo_live_range->CalculateBufferStartEndMap(); in Run()
30 hlo_live_range->NormalizeAliasedBuffers(); in Run()
31 return std::move(hlo_live_range); in Run()
Dheap_simulator.cc106 std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
110 hlo_live_range.get())); in Run()
127 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
131 alias_analysis, hlo_live_range.get())); in Run()
146 std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
149 alias_analysis, hlo_live_range.get())); in Run()
158 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation() argument
162 VLOG(1) << hlo_live_range->ToString(); in RunComputation()
170 hlo_live_range->schedule_end_time() + 1); in RunComputation()
172 hlo_live_range->schedule_end_time() + 1); in RunComputation()
[all …]
Dmemory_space_assignment.h152 const HloLiveRange& hlo_live_range() const { return *hlo_live_range_; } in hlo_live_range() function
160 std::unique_ptr<HloLiveRange> hlo_live_range, in MemorySpaceAssignmentCostAnalysis() argument
168 hlo_live_range_(std::move(hlo_live_range)), in MemorySpaceAssignmentCostAnalysis()
795 HloModule* module, const HloLiveRange& hlo_live_range,
812 const HloLiveRange& hlo_live_range,
818 virtual Status FindAllocationSequence(const HloLiveRange& hlo_live_range,
824 const HloLiveRange& hlo_live_range) in MemorySpaceAssignment() argument
827 flattened_instructions_(hlo_live_range.flattened_instruction_sequence() in MemorySpaceAssignment()
830 hlo_live_range.flattened_instruction_sequence() in MemorySpaceAssignment()
836 hlo_live_range.computation_span_times()) { in MemorySpaceAssignment()
[all …]
Dmemory_space_assignment.cc94 const HloAliasAnalysis& alias_analysis, const HloLiveRange& hlo_live_range, in FindCrossProgramPrefetchCandidate() argument
105 interval.end = hlo_live_range.schedule_end_time(); in FindCrossProgramPrefetchCandidate()
139 TF_ASSIGN_OR_RETURN(auto hlo_live_range, in Create()
146 std::move(hlo_live_range), std::move(call_graph))); in Create()
379 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()
2648 const HloLiveRange& hlo_live_range, in Run() argument
2656 hlo_live_range); in Run()
2658 return memory_space_assignment.RunMemorySpaceAssignment(hlo_live_range, in Run()
[all …]
Dbuffer_assignment.h466 const HloLiveRange& hlo_live_range() const { return *hlo_live_range_; } in hlo_live_range() function
502 std::unique_ptr<HloLiveRange> hlo_live_range) in BufferAssignment() argument
508 hlo_live_range_(std::move(hlo_live_range)) { in BufferAssignment()
Dbuffer_assignment.cc817 const HloLiveRange& live_ranges = hlo_live_range(); in BufferInfoString()
927 CHECK((assignment->hlo_live_range().total_order_scheduled())); in LiveRangeInterferes()
928 const HloLiveRange& hlo_live_range = assignment->hlo_live_range(); in LiveRangeInterferes() local
930 const auto& buffer_live_ranges = hlo_live_range.buffer_live_ranges(); in LiveRangeInterferes()
1051 if (assignment->hlo_live_range().total_order_scheduled()) { in MaybeAssignBuffer()
1644 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloLiveRange> hlo_live_range, in CreateAssignment()
1660 std::move(hlo_live_range))); in CreateAssignment()
Dmemory_space_assignment_test.cc141 std::unique_ptr<HloLiveRange> hlo_live_range = in AssignMemorySpace() local
147 MemorySpaceAssignment::Run(module, *hlo_live_range, *alias_analysis, in AssignMemorySpace()
323 TF_ASSIGN_OR_RETURN(auto hlo_live_range, in Create()
330 std::move(alias_analysis), std::move(hlo_live_range), in Create()
386 std::unique_ptr<HloLiveRange> hlo_live_range, in FakeMemorySpaceAssignmentCostAnalysis() argument
391 std::move(hlo_live_range), std::move(call_graph)) {} in FakeMemorySpaceAssignmentCostAnalysis()
3845 TF_ASSERT_OK_AND_ASSIGN(auto hlo_live_range, in TEST_P()
3850 hlo_live_range->flattened_instruction_sequence().size() + 1, 0); in TEST_P()
3863 hlo_live_range->buffer_live_ranges().at(value); in TEST_P()
DBUILD320 name = "hlo_live_range",
322 "hlo_live_range.cc",
325 "hlo_live_range.h",
353 ":hlo_live_range",
1318 ":hlo_live_range",
1423 ":hlo_live_range",