| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis.cc | 137 /* static */ StatusOr<std::unique_ptr<TuplePointsToAnalysis>>
138 TuplePointsToAnalysis::Run(const HloModule* module) { in Run()
140 std::unique_ptr<TuplePointsToAnalysis> analysis(new TuplePointsToAnalysis( in Run()
146 Status TuplePointsToAnalysis::Analyze() { in Analyze()
177 Status TuplePointsToAnalysis::PopulateDefinedBuffersAndAliases(const decltype(
198 Status TuplePointsToAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction()
219 Status TuplePointsToAnalysis::HandleGetTupleElement( in HandleGetTupleElement()
251 Status TuplePointsToAnalysis::HandleCopy(HloInstruction* copy) { in HandleCopy()
264 Status TuplePointsToAnalysis::HandleBitcast(HloInstruction* bitcast) { in HandleBitcast()
272 Status TuplePointsToAnalysis::HandleDomain(HloInstruction* domain) { in HandleDomain()
[all …]
|
| D | hlo_memory_scheduler.h | 39 HloComputation*, const TuplePointsToAnalysis&,
47 const TuplePointsToAnalysis& points_to_analysis,
55 const TuplePointsToAnalysis& points_to_analysis,
63 const TuplePointsToAnalysis& points_to_analysis,
73 const TuplePointsToAnalysis& points_to_analysis,
|
| D | hlo_memory_scheduler.cc | 79 const TuplePointsToAnalysis& points_to_analysis, in Run()
104 const TuplePointsToAnalysis& points_to_analysis, in ListScheduler()
371 const TuplePointsToAnalysis& points_to_analysis_;
392 const TuplePointsToAnalysis::BufferDefinitionVector& buffers, in SumLogicalBufferSizes()
403 const TuplePointsToAnalysis& points_to_analysis, in ScheduleComputationHelper()
421 const TuplePointsToAnalysis& points_to_analysis, in DFSMemoryScheduler()
492 const TuplePointsToAnalysis& points_to_analysis, in ListMemoryScheduler()
502 const TuplePointsToAnalysis& points_to_analysis, in PostOrderMemoryScheduler()
511 const TuplePointsToAnalysis& points_to_analysis, in DefaultMemoryScheduler()
574 TF_ASSIGN_OR_RETURN(std::unique_ptr<TuplePointsToAnalysis> points_to_analysis, in ScheduleModule()
[all …]
|
| D | tuple_points_to_analysis.h | 183 class TuplePointsToAnalysis : public DfsHloVisitorWithDefault {
186 static StatusOr<std::unique_ptr<TuplePointsToAnalysis>> Run(
277 explicit TuplePointsToAnalysis( in TuplePointsToAnalysis() function
359 TF_DISALLOW_COPY_AND_ASSIGN(TuplePointsToAnalysis);
|
| D | buffer_liveness.h | 61 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis()
109 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
|
| D | heap_simulator.h | 101 const TuplePointsToAnalysis& points_to_analysis,
118 const TuplePointsToAnalysis& points_to_analysis,
130 const TuplePointsToAnalysis& points_to_analysis,
149 const TuplePointsToAnalysis& points_to_analysis);
|
| D | layout_assignment.h | 132 LayoutConstraints(const TuplePointsToAnalysis& points_to_analysis,
138 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis()
226 const TuplePointsToAnalysis& points_to_analysis_;
399 const TuplePointsToAnalysis& points_to_analysis,
|
| D | heap_simulator.cc | 41 TF_ASSIGN_OR_RETURN(std::unique_ptr<TuplePointsToAnalysis> points_to_analysis, in MinimumMemoryForModule()
42 TuplePointsToAnalysis::Run(module)); in MinimumMemoryForModule()
59 const TuplePointsToAnalysis& points_to_analysis, in MinimumMemoryForComputation()
76 const TuplePointsToAnalysis& points_to_analysis, in Run()
91 const TuplePointsToAnalysis& points_to_analysis, in Run()
107 const TuplePointsToAnalysis& points_to_analysis) { in RunComputation()
175 const TuplePointsToAnalysis::BufferDefinitionVector& in RunComputation()
|
| D | hlo_module_group_metadata.h | 210 TuplePointsToAnalysis* points_to_analysis(HloModule* module) const { in points_to_analysis()
285 absl::flat_hash_map<HloModule*, std::unique_ptr<TuplePointsToAnalysis>>
|
| D | hlo_rematerialization.h | 121 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
|
| D | hlo_rematerialization.cc | 291 const TuplePointsToAnalysis& points_to_analysis, in GetUsers()
330 const TuplePointsToAnalysis& points_to_analysis,
415 const TuplePointsToAnalysis& points_to_analysis, in CreateBufferFromLogicalBuffer()
511 const TuplePointsToAnalysis& points_to_analysis, in MemoryUsageTracker()
1215 TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module)); in Run()
|
| D | hlo_memory_scheduler_test.cc | 298 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis = in TEST_F()
299 TuplePointsToAnalysis::Run(module.get()).ValueOrDie(); in TEST_F()
|
| D | buffer_liveness.cc | 47 TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module_)); in Analyze()
|
| D | hlo_module_group_metadata.cc | 136 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis, in Build()
137 TuplePointsToAnalysis::Run(module)); in Build()
|
| D | heap_simulator_test.cc | 209 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis = in TEST_F()
210 TuplePointsToAnalysis::Run(module.get()).ValueOrDie(); in TEST_F()
267 TuplePointsToAnalysis::Run(module_.get()).ConsumeValueOrDie(); in HeapSimulatorTracker()
292 TuplePointsToAnalysis::Run(module_.get()).ConsumeValueOrDie(); in RunWholeModule()
350 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
|
| D | layout_assignment.cc | 104 const TuplePointsToAnalysis& points_to_analysis, in LayoutConstraints()
879 TuplePointsToAnalysis::Run(module)); in CheckLayouts()
1170 const TuplePointsToAnalysis& points_to_analysis) { in GetArrayUsesOfBuffer()
1487 const TuplePointsToAnalysis& points_to_analysis, in InferArrayLayout()
1701 const TuplePointsToAnalysis& points_to_analysis, in RunOnComputation()
1951 TuplePointsToAnalysis::Run(module)); in Run()
|
| D | tuple_points_to_analysis_test.cc | 59 TuplePointsToAnalysis::Run(module_.get()).ConsumeValueOrDie(); in RunAnalysis()
120 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
835 TuplePointsToAnalysis::Run(module_.get()).ConsumeValueOrDie(); in RunAnalysis()
845 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
|
| D | buffer_assignment.h | 438 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis()
|
| D | buffer_assignment.cc | 67 const TuplePointsToAnalysis& points_to_analysis, in AddBufferToColocatedSet()
699 const TuplePointsToAnalysis& points_to_analysis = in ToProto()
1534 const TuplePointsToAnalysis& points_to_analysis = in BuildColocatedBufferSets()
|