Home
last modified time | relevance | path

Searched refs:HloLiveRange (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_live_range.cc33 StatusOr<std::unique_ptr<HloLiveRange>> HloLiveRange::Run( in Run()
36 std::unique_ptr<HloLiveRange> hlo_live_range( in Run()
37 new HloLiveRange(schedule, alias_analysis, module_scoped_analysis)); in Run()
44 void HloLiveRange::NormalizeAliasedBuffers() { in NormalizeAliasedBuffers()
72 void HloLiveRange::FlattenSchedule(const HloComputation& computation, in FlattenSchedule()
122 HloLiveRange::TimeBound HloLiveRange::GetLastPosition( in GetLastPosition()
124 HloLiveRange::LogicalTime definition_end_time) const { in GetLastPosition()
150 HloLiveRange::LogicalTime HloLiveRange::GetLastUsageTime( in GetLastUsageTime()
178 void HloLiveRange::CalculateBufferStartEndMap() { in CalculateBufferStartEndMap()
245 int64_t HloLiveRange::ComputePeakMemoryMoment() const { in ComputePeakMemoryMoment()
[all …]
Dhlo_live_range.h42 class HloLiveRange {
46 static StatusOr<std::unique_ptr<HloLiveRange>> Run(
109 explicit HloLiveRange(const HloSchedule& schedule, in HloLiveRange() function
Dmemory_space_assignment.h206 const HloLiveRange& hlo_live_range() const { return *hlo_live_range_; } in hlo_live_range()
213 std::unique_ptr<HloLiveRange> hlo_live_range, in MemorySpaceAssignmentCostAnalysis()
225 std::unique_ptr<HloLiveRange> hlo_live_range_;
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()
989 float ComputeEstimatedElapsedTime(const HloLiveRange& hlo_live_range,
1223 const HloLiveRange& hlo_live_range);
1262 const HloLiveRange& hlo_live_range() { return hlo_live_range_; } in hlo_live_range()
[all …]
Dhlo_live_range_test.cc36 using TimeBound = HloLiveRange::TimeBound;
44 hlo_live_range_ = HloLiveRange::Run(schedule, *alias_analysis_, in Analyze()
50 std::unique_ptr<HloLiveRange> hlo_live_range_;
63 HloLiveRange::TimeBound LiveRangeAt(const HloInstruction* instruction, in LiveRangeAt()
Dheap_simulator.cc109 std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
110 HloLiveRange::Run(schedule, alias_analysis, entry_computation)); in Run()
130 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
131 HloLiveRange::Run(schedule, alias_analysis, &computation, in Run()
149 std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
150 HloLiveRange::Run(*schedule, alias_analysis, &computation)); in Run()
161 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation()
Dbuffer_assignment.h466 const HloLiveRange& hlo_live_range() const { return *hlo_live_range_; } in hlo_live_range()
505 std::unique_ptr<HloLiveRange> hlo_live_range) in BufferAssignment()
581 std::unique_ptr<HloLiveRange> hlo_live_range_;
Dbuffer_assignment.cc864 const HloLiveRange& live_ranges = hlo_live_range(); in BufferInfoString()
975 const HloLiveRange& hlo_live_range = assignment->hlo_live_range(); in LiveRangeInterferes()
990 const HloLiveRange::TimeBound& operand_live_range) { in LiveRangeInterferes()
1732 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloLiveRange> hlo_live_range, in CreateAssignment()
1733 HloLiveRange::Run(schedule, *alias_analysis, in CreateAssignment()
Dmemory_space_assignment.cc137 const HloLiveRange& hlo_live_range, in FindCrossProgramPrefetchCandidate()
251 HloLiveRange::Run(module.schedule(), *alias_analysis, in Create()
975 const HloLiveRange& hlo_live_range) in AlternateMemoryBestFitHeap()
1711 const HloLiveRange::TimeBound& computation_span = in AllocateAllocationValues()
3449 const HloLiveRange& hlo_live_range, in Run()
3465 const HloLiveRange& hlo_live_range, in RunMemorySpaceAssignment()
3498 const HloLiveRange& hlo_live_range, in FindAllocationSequence()
3545 const HloLiveRange& hlo_live_range, const AllocationSequence& allocations) { in ComputeEstimatedElapsedTime()
4172 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloLiveRange> hlo_live_range, in VerifyAndExportHeapSimulatorTrace()
4173 HloLiveRange::Run(module_->schedule(), *alias_analysis, in VerifyAndExportHeapSimulatorTrace()
[all …]
Dheap_simulator.h194 HloLiveRange* live_range);
Dmemory_space_assignment_test.cc164 std::unique_ptr<HloLiveRange> hlo_live_range = in AssignMemorySpace()
165 HloLiveRange::Run(module->schedule(), *alias_analysis, in AssignMemorySpace()
357 HloLiveRange::Run(module.schedule(), *alias_analysis, in Create()
418 std::unique_ptr<HloLiveRange> hlo_live_range, in FakeMemorySpaceAssignmentCostAnalysis()
3807 HloLiveRange::Run(module->schedule(), *alias_analysis, in TEST_P()
3823 HloLiveRange::TimeBound time_bound = in TEST_P()