Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc79 const TuplePointsToAnalysis& points_to_analysis, in Run() argument
83 ListScheduler scheduler(computation, points_to_analysis, size_function, in Run()
104 const TuplePointsToAnalysis& points_to_analysis, in ListScheduler() argument
109 points_to_analysis_(points_to_analysis), in ListScheduler()
119 points_to_analysis.GetPointsToSet(operand).ForEachElement( in ListScheduler()
133 points_to_analysis.GetBuffersDefinedByInstruction(instruction)) { in ListScheduler()
146 points_to_analysis.GetPointsToSet(computation->root_instruction()) in ListScheduler()
403 const TuplePointsToAnalysis& points_to_analysis, in ScheduleComputationHelper() argument
410 return algorithm(computation, points_to_analysis, size_function, in ScheduleComputationHelper()
413 return DefaultMemoryScheduler(computation, points_to_analysis, size_function, in ScheduleComputationHelper()
[all …]
Dbuffer_assignment.cc67 const TuplePointsToAnalysis& points_to_analysis, in AddBufferToColocatedSet() argument
70 const auto& points_to = points_to_analysis.GetPointsToSet(instruction); in AddBufferToColocatedSet()
338 return points_to_analysis().GetPointsToSet(instruction); in GetPointsToSet()
342 TF_CHECK_OK(points_to_analysis().VerifyBuffer(buffer)); in HasAllocation()
377 const PointsToSet& points_to_set = points_to_analysis().GetPointsToSet(hlo); in GetInstructionAllocation()
520 TF_CHECK_OK(points_to_analysis().VerifyBuffer(buffer)); in AddAssignment()
699 const TuplePointsToAnalysis& points_to_analysis = in ToProto() local
700 liveness_->points_to_analysis(); in ToProto()
701 for (LogicalBuffer::Id id = 0; id < points_to_analysis.num_logical_buffers(); in ToProto()
703 auto& buffer = points_to_analysis.logical_buffer(id); in ToProto()
[all …]
Dhlo_memory_scheduler.h47 const TuplePointsToAnalysis& points_to_analysis,
55 const TuplePointsToAnalysis& points_to_analysis,
63 const TuplePointsToAnalysis& points_to_analysis,
73 const TuplePointsToAnalysis& points_to_analysis,
Dbuffer_liveness.h61 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis() function
75 id < buffer_liveness.points_to_analysis().num_logical_buffers(); in DefaultColorer()
77 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id); in DefaultColorer()
Dheap_simulator.cc41 TF_ASSIGN_OR_RETURN(std::unique_ptr<TuplePointsToAnalysis> points_to_analysis, in MinimumMemoryForModule()
52 schedule, *points_to_analysis, size_function)); in MinimumMemoryForModule()
59 const TuplePointsToAnalysis& points_to_analysis, in MinimumMemoryForComputation() argument
66 computation, sequence, points_to_analysis, in MinimumMemoryForComputation()
76 const TuplePointsToAnalysis& points_to_analysis, in Run() argument
83 *entry_computation, instruction_sequence, points_to_analysis)); in Run()
91 const TuplePointsToAnalysis& points_to_analysis, in Run() argument
98 points_to_analysis)); in Run()
107 const TuplePointsToAnalysis& points_to_analysis) { in RunComputation() argument
142 points_to_analysis.GetPointsToSet(instruction); in RunComputation()
[all …]
Dhlo_rematerialization.cc291 const TuplePointsToAnalysis& points_to_analysis, in GetUsers() argument
298 points_to_analysis.GetBufferAliases(*logical_buffer)) { in GetUsers()
300 if (points_to_analysis.DoesNotUseOperandBuffer( in GetUsers()
330 const TuplePointsToAnalysis& points_to_analysis,
415 const TuplePointsToAnalysis& points_to_analysis, in CreateBufferFromLogicalBuffer() argument
420 points_to_analysis, &has_indirect_uses); in CreateBufferFromLogicalBuffer()
511 const TuplePointsToAnalysis& points_to_analysis, in MemoryUsageTracker() argument
515 points_to_analysis.GetPointsToSet(computation_->root_instruction()) in MemoryUsageTracker()
524 points_to_analysis.GetBuffersDefinedByInstruction(instruction)) { in MemoryUsageTracker()
531 points_to_analysis.GetPointsToSet(instruction->operand(0)); in MemoryUsageTracker()
[all …]
Dlayout_assignment.cc104 const TuplePointsToAnalysis& points_to_analysis, in LayoutConstraints() argument
106 : points_to_analysis_(points_to_analysis), computation_(computation) { in LayoutConstraints()
501 constraints->points_to_analysis().GetBufferDefinedAt(instruction, in AddMandatoryConstraints()
878 TF_ASSIGN_OR_RETURN(auto points_to_analysis, in CheckLayouts()
891 points_to_analysis->GetPointsToSet(instruction); in CheckLayouts()
1170 const TuplePointsToAnalysis& points_to_analysis) { in GetArrayUsesOfBuffer() argument
1173 for (const auto& buffer_alias : points_to_analysis.GetBufferAliases(buffer)) { in GetArrayUsesOfBuffer()
1200 constraints->points_to_analysis().GetPointsToSet(instruction); in PropagateUseConstraintToDefs()
1330 constraints->points_to_analysis().GetBufferDefinedAt( in PropagateOperandConstraint()
1359 constraints->points_to_analysis().GetBufferDefinedAt(user, in PropagateOperandConstraint()
[all …]
Dheap_simulator.h101 const TuplePointsToAnalysis& points_to_analysis,
118 const TuplePointsToAnalysis& points_to_analysis,
130 const TuplePointsToAnalysis& points_to_analysis,
149 const TuplePointsToAnalysis& points_to_analysis);
Dbuffer_liveness.cc105 if (points_to_analysis().DoesNotUseOperandBuffer(alias.instruction(), in live_range_strictly_before()
132 !points_to_analysis().CanShareOperandBufferWithUser( in live_range_strictly_before()
Dlayout_assignment.h132 LayoutConstraints(const TuplePointsToAnalysis& points_to_analysis,
138 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis() function
399 const TuplePointsToAnalysis& points_to_analysis,
Dhlo_memory_scheduler_test.cc298 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis = in TEST_F() local
304 *points_to_analysis, size_fn) in TEST_F()
310 *points_to_analysis, size_fn, &memory_by_computation) in TEST_F()
Dbuffer_liveness_test.cc42 const auto& pointed_to = liveness.points_to_analysis() in GetBuffer()
75 auto& points_to_analysis = liveness.points_to_analysis(); in TupleElementsMayInterfere() local
77 points_to_analysis.GetPointsToSet(a).element(index); in TupleElementsMayInterfere()
79 points_to_analysis.GetPointsToSet(b).element(index); in TupleElementsMayInterfere()
Dbuffer_assignment.h438 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis() function
439 return liveness_->points_to_analysis(); in points_to_analysis()
Dhlo_module_group_metadata.h210 TuplePointsToAnalysis* points_to_analysis(HloModule* module) const { in points_to_analysis() function
Dhlo_module_group_metadata.cc136 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis, in Build()
138 points_to_analyses_[module] = std::move(points_to_analysis); in Build()
Dbuffer_assignment_test.cc583 id < buffer_liveness.points_to_analysis().num_logical_buffers(); in TEST_F()
585 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id); in TEST_F()
642 id < buffer_liveness.points_to_analysis().num_logical_buffers(); in TEST_F()
644 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id); in TEST_F()
646 buffer_liveness.points_to_analysis().GetBufferAliases(buffer); in TEST_F()
Dheap_simulator_test.cc209 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis = in TEST_F() local
216 *points_to_analysis, size_fn, &memory_by_computation) in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc148 constraints->points_to_analysis().GetBufferDefinedAt( in AddBackendConstraintsToDnnConvCustomCall()
232 constraints->points_to_analysis().GetBufferDefinedAt(instruction, in AddBackendConstraints()
237 constraints->points_to_analysis().GetBufferDefinedAt(instruction, in AddBackendConstraints()
298 constraints->points_to_analysis().GetBufferDefinedAt( in PropagateOperandConstraint()
313 constraints->points_to_analysis().GetBufferDefinedAt( in PropagateOperandConstraint()
Dgpu_executable.cc358 return assignment_->points_to_analysis().GetPointsToSet( in GetRootPointsToSet()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc357 return assignment_->points_to_analysis().GetPointsToSet( in GetRootPointsToSet()