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.h44 HloComputation*, const TuplePointsToAnalysis&, const HloAliasAnalysis&,
52 HloModule*, const TuplePointsToAnalysis&, const HloAliasAnalysis&,
66 const HloAliasAnalysis& alias_analysis,
76 const HloAliasAnalysis& alias_analysis,
86 const HloAliasAnalysis& alias_analysis,
99 const HloAliasAnalysis& alias_analysis,
107 const HloAliasAnalysis& alias_analysis,
Dhlo_alias_analysis.cc379 HloAliasAnalysis::HloAliasAnalysis(const HloModule* module) : module_(module) {} in HloAliasAnalysis() function in xla::HloAliasAnalysis
381 const HloBuffer& HloAliasAnalysis::GetUniqueBufferAt( in GetUniqueBufferAt()
388 HloBuffer& HloAliasAnalysis::GetUniqueBufferAt( in GetUniqueBufferAt()
390 return GetBuffer(static_cast<const HloAliasAnalysis*>(this) in GetUniqueBufferAt()
395 std::vector<const HloBuffer*> HloAliasAnalysis::ComputeBuffersAt( in ComputeBuffersAt()
410 bool HloAliasAnalysis::InstructionBuffersAreAmbiguous( in InstructionBuffersAreAmbiguous()
427 bool HloAliasAnalysis::InstructionBuffersAreDistinct( in InstructionBuffersAreDistinct()
461 Status HloAliasAnalysis::Verify() const { in Verify()
487 string HloAliasAnalysis::ToString() const { in ToString()
525 StatusOr<std::unique_ptr<HloAliasAnalysis>> HloAliasAnalysis::Run( in Run()
[all …]
Dhlo_alias_analysis_test.cc49 HloAliasAnalysis& RunAnalysis() { in RunAnalysis()
50 analysis_ = HloAliasAnalysis::Run(module_.get(), in RunAnalysis()
112 std::unique_ptr<HloAliasAnalysis> analysis_;
129 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
164 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
208 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
252 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
295 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
379 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
425 const HloAliasAnalysis& analysis = RunAnalysis(); in TEST_F()
[all …]
Dhlo_live_range.h44 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis,
101 const HloAliasAnalysis& alias_analysis, in HloLiveRange()
199 const HloAliasAnalysis& alias_analysis_;
Dhlo_alias_analysis.h39 class HloAliasAnalysis {
43 static StatusOr<std::unique_ptr<HloAliasAnalysis>> Run(
124 explicit HloAliasAnalysis(const HloModule* module);
Dhlo_memory_scheduler.cc416 const HloAliasAnalysis& alias_analysis, in ScheduleComputationHelper()
438 const HloAliasAnalysis& alias_analysis, in DFSMemoryScheduler()
519 const HloAliasAnalysis& alias_analysis, in ComputationSchedulerToModuleScheduler()
545 const HloAliasAnalysis& alias_analysis, in ListMemoryScheduler()
565 const HloAliasAnalysis& alias_analysis, in PostOrderMemoryScheduler()
583 const HloAliasAnalysis& alias_analysis, in DefaultMemoryScheduler()
641 const HloAliasAnalysis& alias_analysis, in DefaultModuleScheduler()
701 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in ScheduleModule()
702 HloAliasAnalysis::Run(module)); in ScheduleModule()
720 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in ScheduleComputation()
[all …]
Dheap_simulator.h123 const HloAliasAnalysis& alias_analysis,
130 const HloAliasAnalysis& alias_analysis,
146 const HloAliasAnalysis& alias_analysis,
158 const HloAliasAnalysis& alias_analysis,
170 const HloAliasAnalysis& alias_analysis,
187 const HloAliasAnalysis& alias_analysis,
Dloop_schedule_linearizer.cc71 HloInstruction* xla_while, HloAliasAnalysis& alias_analysis) { in AddControlEdgesForLoopWrites()
144 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in Run()
145 HloAliasAnalysis::Run(module, can_share_buffer_)); in Run()
Dcopy_insertion.cc279 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile()
337 Status AddCopiesForInPlaceOperation(const HloAliasAnalysis& alias_analysis, in AddCopiesForInPlaceOperation()
487 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover()
984 const HloAliasAnalysis& alias_analysis, HloInstruction* conditional) { in AddCopiesForConditional()
1014 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in AddCopiesToResolveInterference()
1015 HloAliasAnalysis::Run(module, can_share_buffer_)); in AddCopiesToResolveInterference()
1049 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in AddSpecialCaseCopies()
1050 HloAliasAnalysis::Run(module, can_share_buffer_)); in AddSpecialCaseCopies()
1180 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in RemoveUnnecessaryCopies()
1181 HloAliasAnalysis::Run(module, can_share_buffer_)); in RemoveUnnecessaryCopies()
Dbuffer_assignment.h461 HloAliasAnalysis& alias_analysis() const { return *alias_analysis_; } in alias_analysis()
501 std::unique_ptr<HloAliasAnalysis> alias_analysis, in BufferAssignment()
575 std::unique_ptr<HloAliasAnalysis> alias_analysis_;
587 using Colorer = std::function<Status(HloAliasAnalysis*, const HloOrdering&)>;
590 return [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in DefaultColorer()
Dmemory_space_assignment.h159 std::unique_ptr<HloAliasAnalysis> alias_analysis, in MemorySpaceAssignmentCostAnalysis()
175 std::unique_ptr<HloAliasAnalysis> alias_analysis_;
796 const HloAliasAnalysis& alias_analysis, const Options& options);
813 const HloAliasAnalysis& alias_analysis);
819 const HloAliasAnalysis& alias_analysis);
950 const HloAliasAnalysis& alias_analysis, in AlternateMemoryBestFitHeap()
1000 const HloAliasAnalysis& alias_analysis() { return alias_analysis_; } in alias_analysis()
1324 const HloAliasAnalysis& alias_analysis_;
Dhlo_module_group_metadata.h210 HloAliasAnalysis* alias_analysis(HloModule* module) const { in alias_analysis()
285 absl::flat_hash_map<HloModule*, std::unique_ptr<HloAliasAnalysis>>
Dheap_simulator.cc50 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in MinimumMemoryForModule()
51 HloAliasAnalysis::Run(module)); in MinimumMemoryForModule()
69 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
84 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
99 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
119 const HloAliasAnalysis& alias_analysis, in Run()
140 const HloAliasAnalysis& alias_analysis, in Run()
158 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation()
Dcopy_insertion.h87 virtual Status AddCopiesForConditional(const HloAliasAnalysis& alias_analysis,
Dhlo_memory_scheduler_test.cc46 std::unique_ptr<HloAliasAnalysis> alias_analysis = in PeakMemoryUseOfEntryComputation()
47 HloAliasAnalysis::Run(module).ConsumeValueOrDie(); in PeakMemoryUseOfEntryComputation()
Dhlo_live_range_test.cc43 alias_analysis_ = HloAliasAnalysis::Run(module_.get()).ValueOrDie(); in Analyze()
51 std::unique_ptr<HloAliasAnalysis> alias_analysis_;
Dhlo_live_range.cc23 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
Dhlo_module_group_metadata.cc153 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in Build()
154 HloAliasAnalysis::Run(module)); in Build()
Dheap_simulator_test.cc211 std::unique_ptr<HloAliasAnalysis> alias_analysis = in TEST_F()
212 HloAliasAnalysis::Run(module.get()).ValueOrDie(); in TEST_F()
305 alias_analysis_ = HloAliasAnalysis::Run(module_.get()).ConsumeValueOrDie(); in RunWholeModule()
389 HloAliasAnalysis::Run(module_.get(), can_share_buffer).ValueOrDie(); in Init()
401 std::unique_ptr<HloAliasAnalysis> alias_analysis_;
Dmemory_space_assignment.cc94 const HloAliasAnalysis& alias_analysis, const HloLiveRange& hlo_live_range, in FindCrossProgramPrefetchCandidate()
138 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(&module)); in Create()
2649 const HloAliasAnalysis& alias_analysis, in Run()
2665 const HloAliasAnalysis& alias_analysis) { in RunMemorySpaceAssignment()
2691 const HloAliasAnalysis& alias_analysis) { in FindAllocationSequence()
2937 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(module_)); in ExportAndColorBuffers()
3255 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in VerifyAndExportHeapSimulatorTrace()
3256 HloAliasAnalysis::Run(module_)); in VerifyAndExportHeapSimulatorTrace()
Dbuffer_assignment.cc1254 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignBuffersForComputations()
1380 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignPresetBuffers()
1630 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in CreateAssignment()
1631 HloAliasAnalysis::Run(module, can_share_buffer)); in CreateAssignment()
Dmemory_space_assignment_test.cc55 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpaceUsingCostAnalysis()
140 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpace()
241 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie();
322 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(&module)); in Create()
385 std::unique_ptr<HloAliasAnalysis> alias_analysis, in FakeMemorySpaceAssignmentCostAnalysis()
3844 HloAliasAnalysis::Run(module.get())); in TEST_P()
Dbuffer_assignment_test.cc585 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
651 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
Dlayout_assignment.cc2012 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(module)); in PropagateMemorySpace()