Home
last modified time | relevance | path

Searched refs:points_to_analysis (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc82 const TuplePointsToAnalysis& points_to_analysis, in Run() argument
86 ListScheduler scheduler(computation, points_to_analysis, size_function, in Run()
107 const TuplePointsToAnalysis& points_to_analysis, in ListScheduler() argument
112 points_to_analysis_(points_to_analysis), in ListScheduler()
122 points_to_analysis.GetPointsToSet(operand).ForEachElement( in ListScheduler()
134 unscheduled_use_count_.reserve(points_to_analysis.num_logical_buffers()); in ListScheduler()
137 points_to_analysis.GetBuffersDefinedByInstruction(instruction)) { in ListScheduler()
150 points_to_analysis.GetPointsToSet(computation->root_instruction()) in ListScheduler()
415 const TuplePointsToAnalysis& points_to_analysis, in ScheduleComputationHelper() argument
425 return algorithm(computation, points_to_analysis, alias_analysis, in ScheduleComputationHelper()
[all …]
Dhlo_memory_scheduler.h72 const TuplePointsToAnalysis& points_to_analysis,
82 const TuplePointsToAnalysis& points_to_analysis,
92 const TuplePointsToAnalysis& points_to_analysis,
105 const TuplePointsToAnalysis& points_to_analysis,
113 HloModule* module, const TuplePointsToAnalysis& points_to_analysis,
Dlayout_assignment.cc112 const TuplePointsToAnalysis& points_to_analysis, in LayoutConstraints() argument
114 : points_to_analysis_(points_to_analysis), computation_(computation) { in LayoutConstraints()
593 constraints->points_to_analysis().GetBufferDefinedAt(instruction, in AddMandatoryConstraints()
1019 TF_ASSIGN_OR_RETURN(auto points_to_analysis, in CheckLayouts()
1032 points_to_analysis->GetPointsToSet(instruction); in CheckLayouts()
1316 const TuplePointsToAnalysis& points_to_analysis) { in GetArrayUsesOfBuffer() argument
1319 for (const auto& buffer_alias : points_to_analysis.GetBufferAliases(buffer)) { in GetArrayUsesOfBuffer()
1346 constraints->points_to_analysis().GetPointsToSet(instruction); in PropagateUseConstraintToDefs()
1420 constraints->points_to_analysis().GetBufferDefinedAt( in PropagateOperandConstraint()
1511 if (!constraints->points_to_analysis() in PropagateOperandConstraint()
[all …]
Dhlo_rematerialization.cc448 const TuplePointsToAnalysis& points_to_analysis, in GetUsers() argument
455 points_to_analysis.GetBufferAliases(*logical_buffer)) { in GetUsers()
457 if (points_to_analysis.DoesNotUseOperandBuffer( in GetUsers()
498 const TuplePointsToAnalysis& points_to_analysis,
660 const TuplePointsToAnalysis& points_to_analysis, bool live_out) { in CreateBufferFromLogicalBuffer() argument
663 points_to_analysis, &has_indirect_uses); in CreateBufferFromLogicalBuffer()
790 const TuplePointsToAnalysis& points_to_analysis, in MemoryUsageTracker() argument
799 points_to_analysis.GetPointsToSet(computation_->root_instruction()) in MemoryUsageTracker()
807 points_to_analysis.GetBuffersDefinedByInstruction(instruction)) { in MemoryUsageTracker()
814 points_to_analysis.GetPointsToSet(instruction->operand(0)); in MemoryUsageTracker()
[all …]
Dlayout_assignment.h146 LayoutConstraints(const TuplePointsToAnalysis& points_to_analysis,
152 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc189 constraints->points_to_analysis().GetBufferDefinedAt( in AddBackendConstraintsToDnnConvCustomCall()
284 constraints->points_to_analysis().GetBufferDefinedAt(instruction, in AddBackendConstraints()
289 constraints->points_to_analysis().GetBufferDefinedAt(instruction, in AddBackendConstraints()
373 constraints->points_to_analysis().GetBufferDefinedAt( in PropagateOperandConstraint()
388 constraints->points_to_analysis().GetBufferDefinedAt( in PropagateOperandConstraint()