Searched refs:HloLiveRange (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_live_range.cc | 22 StatusOr<std::unique_ptr<HloLiveRange>> HloLiveRange::Run( in Run()
25 std::unique_ptr<HloLiveRange> hlo_live_range( in Run()
26 new HloLiveRange(schedule, alias_analysis, module_scoped_analysis)); in Run()
34 void HloLiveRange::NormalizeAliasedBuffers() { in NormalizeAliasedBuffers()
90 int64 HloLiveRange::FlattenSchedule(const HloComputation& computation, in FlattenSchedule()
130 void HloLiveRange::CalculateBufferStartEndMap() { in CalculateBufferStartEndMap()
210 std::string HloLiveRange::ToString() const { in ToString()
|
| D | hlo_live_range.h | 39 class HloLiveRange {
43 static StatusOr<std::unique_ptr<HloLiveRange>> Run(
100 explicit HloLiveRange(const HloSchedule& schedule, in HloLiveRange() function
|
| D | memory_space_assignment.h | 152 const HloLiveRange& hlo_live_range() const { return *hlo_live_range_; } in hlo_live_range()
160 std::unique_ptr<HloLiveRange> hlo_live_range, in MemorySpaceAssignmentCostAnalysis()
176 std::unique_ptr<HloLiveRange> hlo_live_range_;
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()
951 const HloLiveRange& hlo_live_range) in AlternateMemoryBestFitHeap()
1001 const HloLiveRange& hlo_live_range() { return hlo_live_range_; } in hlo_live_range()
1325 const HloLiveRange& hlo_live_range_;
|
| D | hlo_live_range_test.cc | 36 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()
|
| D | heap_simulator.cc | 106 std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
107 HloLiveRange::Run(schedule, alias_analysis, entry_computation)); in Run()
127 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
128 HloLiveRange::Run(schedule, alias_analysis, &computation, in Run()
146 std::unique_ptr<HloLiveRange> hlo_live_range, in Run()
147 HloLiveRange::Run(*schedule, alias_analysis, &computation)); in Run()
158 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation()
|
| D | buffer_assignment.h | 466 const HloLiveRange& hlo_live_range() const { return *hlo_live_range_; } in hlo_live_range()
502 std::unique_ptr<HloLiveRange> hlo_live_range) in BufferAssignment()
577 std::unique_ptr<HloLiveRange> hlo_live_range_;
|
| D | memory_space_assignment.cc | 94 const HloAliasAnalysis& alias_analysis, const HloLiveRange& hlo_live_range, in FindCrossProgramPrefetchCandidate()
140 HloLiveRange::Run(module.schedule(), *alias_analysis, in Create()
1325 const HloLiveRange::TimeBound& computation_span = in AllocateAllocationValues()
2648 const HloLiveRange& hlo_live_range, in Run()
2664 const HloLiveRange& hlo_live_range, in RunMemorySpaceAssignment()
2690 const HloLiveRange& hlo_live_range, in FindAllocationSequence()
3257 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloLiveRange> hlo_live_range, in VerifyAndExportHeapSimulatorTrace()
3258 HloLiveRange::Run(module_->schedule(), *alias_analysis, in VerifyAndExportHeapSimulatorTrace()
3331 const HloLiveRange::TimeBound& time_bound = in VerifyAndExportHeapSimulatorTrace()
|
| D | buffer_assignment.cc | 817 const HloLiveRange& live_ranges = hlo_live_range(); in BufferInfoString()
928 const HloLiveRange& hlo_live_range = assignment->hlo_live_range(); in LiveRangeInterferes()
1644 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloLiveRange> hlo_live_range, in CreateAssignment()
1645 HloLiveRange::Run(schedule, *alias_analysis, in CreateAssignment()
|
| D | heap_simulator.h | 188 HloLiveRange* live_range);
|
| D | memory_space_assignment_test.cc | 141 std::unique_ptr<HloLiveRange> hlo_live_range = in AssignMemorySpace()
142 HloLiveRange::Run(module->schedule(), *alias_analysis, in AssignMemorySpace()
324 HloLiveRange::Run(module.schedule(), *alias_analysis, in Create()
386 std::unique_ptr<HloLiveRange> hlo_live_range, in FakeMemorySpaceAssignmentCostAnalysis()
3846 HloLiveRange::Run(module->schedule(), *alias_analysis, in TEST_P()
3862 HloLiveRange::TimeBound time_bound = in TEST_P()
|