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(const decltype(
201 Status TuplePointsToAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction()
222 Status TuplePointsToAnalysis::HandleGetTupleElement( in HandleGetTupleElement()
254 Status TuplePointsToAnalysis::HandleCopy(HloInstruction* copy) { in HandleCopy()
267 Status TuplePointsToAnalysis::HandleBitcast(HloInstruction* bitcast) { in HandleBitcast()
275 Status TuplePointsToAnalysis::HandleDomain(HloInstruction* domain) { in HandleDomain()
[all …]
Dhlo_memory_scheduler.h44 HloComputation*, const TuplePointsToAnalysis&, const HloAliasAnalysis&,
52 HloModule*, const TuplePointsToAnalysis&, const HloAliasAnalysis&,
65 const TuplePointsToAnalysis& points_to_analysis,
75 const TuplePointsToAnalysis& points_to_analysis,
85 const TuplePointsToAnalysis& points_to_analysis,
98 const TuplePointsToAnalysis& points_to_analysis,
106 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()
437 const TuplePointsToAnalysis& points_to_analysis, in DFSMemoryScheduler()
518 HloModule* module, const TuplePointsToAnalysis& points_to_analysis, in ComputationSchedulerToModuleScheduler()
544 const TuplePointsToAnalysis& points_to_analysis, in ListMemoryScheduler()
564 const TuplePointsToAnalysis& points_to_analysis, in PostOrderMemoryScheduler()
582 const TuplePointsToAnalysis& points_to_analysis, in DefaultMemoryScheduler()
[all …]
Dtuple_points_to_analysis.h184 class TuplePointsToAnalysis : public DfsHloVisitorWithDefault {
187 static StatusOr<std::unique_ptr<TuplePointsToAnalysis>> Run(
272 explicit TuplePointsToAnalysis( in TuplePointsToAnalysis() function
354 TF_DISALLOW_COPY_AND_ASSIGN(TuplePointsToAnalysis);
Dlayout_assignment.h134 LayoutConstraints(const TuplePointsToAnalysis& points_to_analysis,
140 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis()
228 const TuplePointsToAnalysis& points_to_analysis_;
536 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
Dhlo_rematerialization.h169 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
Dtuple_points_to_analysis_test.cc61 TuplePointsToAnalysis::Run(module_.get()).ConsumeValueOrDie(); in RunAnalysis()
122 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
886 TuplePointsToAnalysis::Run(module_.get()).ConsumeValueOrDie(); in RunAnalysis()
896 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
Dhlo_rematerialization.cc447 const TuplePointsToAnalysis& points_to_analysis, in GetUsers()
497 const TuplePointsToAnalysis& points_to_analysis,
655 const TuplePointsToAnalysis& points_to_analysis, bool live_out) { in CreateBufferFromLogicalBuffer()
785 const TuplePointsToAnalysis& points_to_analysis, in MemoryUsageTracker()
2017 TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module)); in Run()
Dlayout_assignment.cc109 const TuplePointsToAnalysis& points_to_analysis, in LayoutConstraints()
932 TuplePointsToAnalysis::Run(module)); in CheckLayouts()
1228 const TuplePointsToAnalysis& points_to_analysis) { in GetArrayUsesOfBuffer()
1628 const TuplePointsToAnalysis& points_to_analysis, in InferArrayLayout()
2172 TuplePointsToAnalysis::Run(module)); in Run()