Home
last modified time | relevance | path

Searched refs:HloAliasAnalysis (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.h50 HloComputation*, const TuplePointsToAnalysis&, const HloAliasAnalysis&,
59 const HloModule*, const TuplePointsToAnalysis&, const HloAliasAnalysis&,
74 const HloAliasAnalysis& alias_analysis,
84 const HloAliasAnalysis& alias_analysis,
94 const HloAliasAnalysis& alias_analysis,
107 const HloAliasAnalysis& alias_analysis,
115 const HloAliasAnalysis& alias_analysis,
Dhlo_alias_analysis.cc286 HloAliasAnalysis::HloAliasAnalysis(const HloModule* module) : module_(module) {} in HloAliasAnalysis() function in xla::HloAliasAnalysis
288 const HloBuffer& HloAliasAnalysis::GetUniqueBufferAt( in GetUniqueBufferAt()
295 HloBuffer& HloAliasAnalysis::GetUniqueBufferAt( in GetUniqueBufferAt()
297 return GetBuffer(const_cast<const HloAliasAnalysis*>(this) in GetUniqueBufferAt()
302 std::vector<const HloBuffer*> HloAliasAnalysis::ComputeBuffersAt( in ComputeBuffersAt()
319 Status HloAliasAnalysis::Verify() const { in Verify()
345 std::string HloAliasAnalysis::ToString() const { in ToString()
384 StatusOr<std::unique_ptr<HloAliasAnalysis>> HloAliasAnalysis::Run( in Run()
390 auto alias_analysis = absl::WrapUnique(new HloAliasAnalysis(module)); in Run()
Dhlo_alias_analysis_test.cc49 HloAliasAnalysis& RunAnalysis() { in RunAnalysis()
50 analysis_ = HloAliasAnalysis::Run(module_.get(), in RunAnalysis()
145 std::unique_ptr<HloAliasAnalysis> analysis_;
162 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
197 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
241 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
285 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
328 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
412 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
457 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
[all …]
Dhlo_memory_scheduler.cc416 const HloAliasAnalysis& alias_analysis, in ScheduleComputationHelper()
439 const HloAliasAnalysis& alias_analysis, in DFSMemoryScheduler()
532 const HloAliasAnalysis& alias_analysis, in ComputationSchedulerToModuleScheduler()
561 const HloAliasAnalysis& alias_analysis, in ListMemoryScheduler()
584 const HloAliasAnalysis& alias_analysis, in PostOrderMemoryScheduler()
605 const HloAliasAnalysis& alias_analysis, in DefaultMemoryScheduler()
665 const HloAliasAnalysis& alias_analysis, in DefaultModuleScheduler()
730 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in ScheduleModule()
731 HloAliasAnalysis::Run(module)); in ScheduleModule()
749 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in ScheduleComputation()
[all …]
Dhlo_alias_analysis.h38 class HloAliasAnalysis {
42 static StatusOr<std::unique_ptr<HloAliasAnalysis>> Run(
103 explicit HloAliasAnalysis(const HloModule* module);
Dhlo_live_range.h47 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis,
110 const HloAliasAnalysis& alias_analysis, in HloLiveRange()
220 const HloAliasAnalysis& alias_analysis_;
Dheap_simulator.h129 const HloAliasAnalysis& alias_analysis,
136 const HloAliasAnalysis& alias_analysis,
152 const HloAliasAnalysis& alias_analysis,
164 const HloAliasAnalysis& alias_analysis,
176 const HloAliasAnalysis& alias_analysis,
193 const HloAliasAnalysis& alias_analysis,
Dloop_schedule_linearizer.cc71 HloInstruction* xla_while, HloAliasAnalysis& alias_analysis) { in AddControlEdgesForLoopWrites()
146 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in Run()
147 HloAliasAnalysis::Run(module, can_share_buffer_)); in Run()
Dbuffer_assignment.h461 HloAliasAnalysis& alias_analysis() const { return *alias_analysis_; } in alias_analysis()
504 std::unique_ptr<HloAliasAnalysis> alias_analysis, in BufferAssignment()
579 std::unique_ptr<HloAliasAnalysis> alias_analysis_;
592 using Colorer = std::function<Status(HloAliasAnalysis*, const HloOrdering&)>;
597 return [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in DefaultColorer()
Dheap_simulator.cc56 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in MinimumMemoryForModule()
57 HloAliasAnalysis::Run(module)); in MinimumMemoryForModule()
74 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
88 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
102 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
122 const HloAliasAnalysis& alias_analysis, in Run()
143 const HloAliasAnalysis& alias_analysis, in Run()
161 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation()
Dhlo_module_group_metadata.h209 HloAliasAnalysis* alias_analysis(HloModule* module) const { in alias_analysis()
284 absl::flat_hash_map<HloModule*, std::unique_ptr<HloAliasAnalysis>>
Dmemory_space_assignment.h212 std::unique_ptr<HloAliasAnalysis> alias_analysis, in MemorySpaceAssignmentCostAnalysis()
224 std::unique_ptr<HloAliasAnalysis> alias_analysis_;
911 const HloAliasAnalysis& alias_analysis, const Options& options);
928 const HloAliasAnalysis& alias_analysis);
934 const HloAliasAnalysis& alias_analysis);
1222 const Options& options, const HloAliasAnalysis& alias_analysis,
1261 const HloAliasAnalysis& alias_analysis() { return alias_analysis_; } in alias_analysis()
1596 const HloAliasAnalysis& alias_analysis_;
Dcopy_insertion.cc288 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile()
346 Status AddCopiesForInPlaceOperation(const HloAliasAnalysis& alias_analysis, in AddCopiesForInPlaceOperation()
1146 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover()
1784 const HloAliasAnalysis& alias_analysis, HloInstruction* conditional) { in AddCopiesForConditional()
1816 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in AddCopiesToResolveInterference()
1817 HloAliasAnalysis::Run(module, can_share_buffer_)); in AddCopiesToResolveInterference()
1861 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in AddSpecialCaseCopies()
1862 HloAliasAnalysis::Run(module, can_share_buffer_)); in AddSpecialCaseCopies()
2028 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in RemoveUnnecessaryCopies()
2029 HloAliasAnalysis::Run(module, can_share_buffer_)); in RemoveUnnecessaryCopies()
Dcopy_insertion.h102 virtual Status AddCopiesForConditional(const HloAliasAnalysis& alias_analysis,
Dhlo_memory_scheduler_test.cc46 std::unique_ptr<HloAliasAnalysis> alias_analysis = in PeakMemoryUseOfEntryComputation()
47 HloAliasAnalysis::Run(module).value(); in PeakMemoryUseOfEntryComputation()
Dhlo_live_range_test.cc43 alias_analysis_ = HloAliasAnalysis::Run(module_.get()).ValueOrDie(); in Analyze()
51 std::unique_ptr<HloAliasAnalysis> alias_analysis_;
Dhlo_module_group_metadata.cc154 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in Build()
155 HloAliasAnalysis::Run(module)); in Build()
Dheap_simulator_test.cc210 std::unique_ptr<HloAliasAnalysis> alias_analysis = in TEST_F()
211 HloAliasAnalysis::Run(module.get()).ValueOrDie(); in TEST_F()
305 alias_analysis_ = HloAliasAnalysis::Run(module_.get()).value(); in RunWholeModule()
389 HloAliasAnalysis::Run(module_.get(), can_share_buffer).ValueOrDie(); in Init()
401 std::unique_ptr<HloAliasAnalysis> alias_analysis_;
Dhlo_live_range.cc34 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
Dbuffer_assignment.cc1310 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignBuffersForComputations()
1436 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignPresetBuffers()
1718 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in CreateAssignment()
1719 HloAliasAnalysis::Run(module, can_share_buffer)); in CreateAssignment()
Dmemory_space_assignment.cc136 FindCrossProgramPrefetchCandidate(const HloAliasAnalysis& alias_analysis, in FindCrossProgramPrefetchCandidate()
249 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(&module)); in Create()
974 const Options& options, const HloAliasAnalysis& alias_analysis, in AlternateMemoryBestFitHeap()
3450 const HloAliasAnalysis& alias_analysis, in Run()
3466 const HloAliasAnalysis& alias_analysis) { in RunMemorySpaceAssignment()
3499 const HloAliasAnalysis& alias_analysis) { in FindAllocationSequence()
3855 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(module_)); in ExportAndColorBuffers()
4170 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in VerifyAndExportHeapSimulatorTrace()
4171 HloAliasAnalysis::Run(module_)); in VerifyAndExportHeapSimulatorTrace()
Dmemory_space_assignment_test.cc67 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpaceUsingCostAnalysis()
163 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpace()
273 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie();
355 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(&module)); in Create()
417 std::unique_ptr<HloAliasAnalysis> alias_analysis, in FakeMemorySpaceAssignmentCostAnalysis()
3805 HloAliasAnalysis::Run(module.get())); in TEST_P()
Dbuffer_assignment_test.cc603 auto colorer = [&](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
670 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
Dlayout_assignment.cc2325 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(module)); in PropagateMemorySpace()