Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.cc109 std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
113 hlo_live_range.get())); in Run()
130 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
134 alias_analysis, hlo_live_range.get())); in Run()
149 std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
152 alias_analysis, hlo_live_range.get())); in Run()
161 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation() argument
165 VLOG(1) << hlo_live_range->ToString(); in RunComputation()
173 hlo_live_range->schedule_end_time() + 1); in RunComputation()
175 hlo_live_range->schedule_end_time() + 1); in RunComputation()
[all …]
Dhlo_live_range.cc36 std::unique_ptr<HloLiveRange> hlo_live_range( in Run() local
38 hlo_live_range->FlattenSchedule(*computation); in Run()
39 hlo_live_range->CalculateBufferStartEndMap(); in Run()
40 hlo_live_range->NormalizeAliasedBuffers(); in Run()
41 return std::move(hlo_live_range); in Run()
Dmemory_space_assignment.h206 const HloLiveRange& hlo_live_range() const { return *hlo_live_range_; } in hlo_live_range() function
213 std::unique_ptr<HloLiveRange> hlo_live_range, in MemorySpaceAssignmentCostAnalysis() argument
218 hlo_live_range_(std::move(hlo_live_range)), in MemorySpaceAssignmentCostAnalysis()
910 HloModule* module, const HloLiveRange& hlo_live_range,
927 const HloLiveRange& hlo_live_range,
933 virtual Status FindAllocationSequence(const HloLiveRange& hlo_live_range,
939 const HloLiveRange& hlo_live_range) in MemorySpaceAssignment() argument
942 flattened_instructions_(hlo_live_range.flattened_instruction_sequence() in MemorySpaceAssignment()
945 hlo_live_range.flattened_instruction_sequence() in MemorySpaceAssignment()
951 hlo_live_range.computation_span_times()) { in MemorySpaceAssignment()
[all …]
Dmemory_space_assignment.cc137 const HloLiveRange& hlo_live_range, in FindCrossProgramPrefetchCandidate() argument
148 interval.end = hlo_live_range.schedule_end_time(); in FindCrossProgramPrefetchCandidate()
250 TF_ASSIGN_OR_RETURN(auto hlo_live_range, in Create()
256 std::move(hlo_live_range), std::move(call_graph))); in Create()
556 cost_analysis.hlo_live_range().instruction_schedule().size() + 1, 0), in CostAnalysisPrefetchIntervalPicker()
558 cost_analysis.hlo_live_range().instruction_schedule().size() + 1, 0), in CostAnalysisPrefetchIntervalPicker()
568 &cost_analysis_.hlo_live_range().instruction_schedule();
975 const HloLiveRange& hlo_live_range) in AlternateMemoryBestFitHeap() argument
980 hlo_live_range_(hlo_live_range), in AlternateMemoryBestFitHeap()
981 peak_memory_usage_(hlo_live_range.schedule_end_time() + 1) { in AlternateMemoryBestFitHeap()
[all …]
Dbuffer_assignment.h466 const HloLiveRange& hlo_live_range() const { return *hlo_live_range_; } in hlo_live_range() function
505 std::unique_ptr<HloLiveRange> hlo_live_range) in BufferAssignment() argument
511 hlo_live_range_(std::move(hlo_live_range)) { in BufferAssignment()
Dbuffer_assignment.cc864 const HloLiveRange& live_ranges = hlo_live_range(); in BufferInfoString()
974 CHECK((assignment->hlo_live_range().total_order_scheduled())); in LiveRangeInterferes()
975 const HloLiveRange& hlo_live_range = assignment->hlo_live_range(); in LiveRangeInterferes() local
977 const auto& buffer_live_ranges = hlo_live_range.buffer_live_ranges(); in LiveRangeInterferes()
1106 if (assignment->hlo_live_range().total_order_scheduled()) { in MaybeAssignBuffer()
1732 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloLiveRange> hlo_live_range, in CreateAssignment()
1748 std::move(hlo_live_range))); in CreateAssignment()
Dmemory_space_assignment_test.cc164 std::unique_ptr<HloLiveRange> hlo_live_range = in AssignMemorySpace() local
170 MemorySpaceAssignment::Run(module, *hlo_live_range, *alias_analysis, in AssignMemorySpace()
356 TF_ASSIGN_OR_RETURN(auto hlo_live_range, in Create()
362 std::move(hlo_live_range), std::move(call_graph))); in Create()
418 std::unique_ptr<HloLiveRange> hlo_live_range, in FakeMemorySpaceAssignmentCostAnalysis() argument
422 std::move(hlo_live_range), std::move(call_graph)) {} in FakeMemorySpaceAssignmentCostAnalysis()
3806 TF_ASSERT_OK_AND_ASSIGN(auto hlo_live_range, in TEST_P()
3811 hlo_live_range->flattened_instruction_sequence().size() + 1, 0); in TEST_P()
3824 hlo_live_range->buffer_live_ranges().at(value); in TEST_P()
Ddump.cc405 [&] { return buffer_assn->hlo_live_range().ToString(); }); in DumpHloModuleImpl()
DBUILD484 name = "hlo_live_range",
486 "hlo_live_range.cc",
489 "hlo_live_range.h",
515 ":hlo_live_range",
1537 ":hlo_live_range",
1646 ":hlo_live_range",