Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc140 /* static */ StatusOr<std::unique_ptr<TuplePointsToAnalysis>>
141 TuplePointsToAnalysis::Run(const HloModule* module) { in Run()
143 std::unique_ptr<TuplePointsToAnalysis> analysis(new TuplePointsToAnalysis( in Run()
149 Status TuplePointsToAnalysis::Analyze() { in Analyze()
180 Status TuplePointsToAnalysis::PopulateDefinedBuffersAndAliases(
202 Status TuplePointsToAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction()
223 Status TuplePointsToAnalysis::HandleGetTupleElement( in HandleGetTupleElement()
255 Status TuplePointsToAnalysis::HandleCopy(HloInstruction* copy) { in HandleCopy()
268 Status TuplePointsToAnalysis::HandleBitcast(HloInstruction* bitcast) { in HandleBitcast()
276 Status TuplePointsToAnalysis::HandleDomain(HloInstruction* domain) { in HandleDomain()
[all …]
Dhlo_memory_scheduler.h50 HloComputation*, const TuplePointsToAnalysis&, const HloAliasAnalysis&,
59 const HloModule*, const TuplePointsToAnalysis&, const HloAliasAnalysis&,
73 const TuplePointsToAnalysis& points_to_analysis,
83 const TuplePointsToAnalysis& points_to_analysis,
93 const TuplePointsToAnalysis& points_to_analysis,
106 const TuplePointsToAnalysis& points_to_analysis,
114 const HloModule* module, const TuplePointsToAnalysis& points_to_analysis,
Dhlo_memory_scheduler.cc82 const TuplePointsToAnalysis& points_to_analysis, in Run()
107 const TuplePointsToAnalysis& points_to_analysis, in ListScheduler()
383 const TuplePointsToAnalysis& points_to_analysis_;
404 const TuplePointsToAnalysis::BufferDefinitionVector& buffers, in SumLogicalBufferSizes()
415 const TuplePointsToAnalysis& points_to_analysis, in ScheduleComputationHelper()
438 const TuplePointsToAnalysis& points_to_analysis, in DFSMemoryScheduler()
531 const TuplePointsToAnalysis& points_to_analysis, in ComputationSchedulerToModuleScheduler()
560 const TuplePointsToAnalysis& points_to_analysis, in ListMemoryScheduler()
583 const TuplePointsToAnalysis& points_to_analysis, in PostOrderMemoryScheduler()
604 const TuplePointsToAnalysis& points_to_analysis, in DefaultMemoryScheduler()
[all …]
Dtuple_points_to_analysis.h183 class TuplePointsToAnalysis : public DfsHloVisitorWithDefault {
186 static StatusOr<std::unique_ptr<TuplePointsToAnalysis>> Run(
274 explicit TuplePointsToAnalysis( in TuplePointsToAnalysis() function
358 TuplePointsToAnalysis(const TuplePointsToAnalysis&) = delete;
359 TuplePointsToAnalysis& operator=(const TuplePointsToAnalysis&) = delete;
Dtuple_points_to_analysis_test.cc61 points_to_analysis_ = TuplePointsToAnalysis::Run(module_.get()).value(); in RunAnalysis()
123 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
Dhlo_rematerialization.h203 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
Dlayout_assignment.h252 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis()
614 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
Dhlo_rematerialization.cc451 const TuplePointsToAnalysis& points_to_analysis, in GetUsers()
501 const TuplePointsToAnalysis& points_to_analysis,
666 const TuplePointsToAnalysis& points_to_analysis, bool live_out) { in CreateBufferFromLogicalBuffer()
796 const TuplePointsToAnalysis& points_to_analysis, in MemoryUsageTracker()
2099 TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module)); in Run()
Dlayout_assignment.cc1133 TuplePointsToAnalysis::Run(module)); in CheckLayouts()
1458 const TuplePointsToAnalysis::BufferAliasVector& aliases) { in GetArrayUsesOfBuffer()
2484 TuplePointsToAnalysis::Run(module)); in Run()