| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_memory_scheduler.cc | 79 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 …]
|
| D | buffer_assignment.cc | 67 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 …]
|
| D | hlo_memory_scheduler.h | 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 | buffer_liveness.h | 61 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()
|
| D | heap_simulator.cc | 41 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 …]
|
| D | hlo_rematerialization.cc | 291 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 …]
|
| D | layout_assignment.cc | 104 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 …]
|
| 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 | buffer_liveness.cc | 105 if (points_to_analysis().DoesNotUseOperandBuffer(alias.instruction(), in live_range_strictly_before()
132 !points_to_analysis().CanShareOperandBufferWithUser( in live_range_strictly_before()
|
| D | layout_assignment.h | 132 LayoutConstraints(const TuplePointsToAnalysis& points_to_analysis,
138 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis() function
399 const TuplePointsToAnalysis& points_to_analysis,
|
| D | hlo_memory_scheduler_test.cc | 298 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()
|
| D | buffer_liveness_test.cc | 42 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()
|
| D | buffer_assignment.h | 438 const TuplePointsToAnalysis& points_to_analysis() const { in points_to_analysis() function
439 return liveness_->points_to_analysis(); in points_to_analysis()
|
| D | hlo_module_group_metadata.h | 210 TuplePointsToAnalysis* points_to_analysis(HloModule* module) const { in points_to_analysis() function
|
| D | hlo_module_group_metadata.cc | 136 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis, in Build()
138 points_to_analyses_[module] = std::move(points_to_analysis); in Build()
|
| D | buffer_assignment_test.cc | 583 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()
|
| D | heap_simulator_test.cc | 209 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/ |
| D | gpu_layout_assignment.cc | 148 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()
|
| D | gpu_executable.cc | 358 return assignment_->points_to_analysis().GetPointsToSet( in GetRootPointsToSet()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 357 return assignment_->points_to_analysis().GetPointsToSet( in GetRootPointsToSet()
|