| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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,
|
| buffer_liveness.h | 61 const TuplePointsToAnalysis& points_to_analysis() const { function in class:xla::BufferLiveness
75 id < buffer_liveness.points_to_analysis().num_logical_buffers();
77 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id);
|
| hlo_memory_scheduler.cc | 79 const TuplePointsToAnalysis& points_to_analysis,
83 ListScheduler scheduler(computation, points_to_analysis, size_function,
104 const TuplePointsToAnalysis& points_to_analysis,
109 points_to_analysis_(points_to_analysis),
119 points_to_analysis.GetPointsToSet(operand).ForEachElement(
133 points_to_analysis.GetBuffersDefinedByInstruction(instruction)) {
146 points_to_analysis.GetPointsToSet(computation->root_instruction())
403 const TuplePointsToAnalysis& points_to_analysis,
410 return algorithm(computation, points_to_analysis, size_function,
413 return DefaultMemoryScheduler(computation, points_to_analysis, size_function
[all...] |
| buffer_assignment.cc | 67 const TuplePointsToAnalysis& points_to_analysis,
70 const auto& points_to = points_to_analysis.GetPointsToSet(instruction);
338 return points_to_analysis().GetPointsToSet(instruction);
342 TF_CHECK_OK(points_to_analysis().VerifyBuffer(buffer));
377 const PointsToSet& points_to_set = points_to_analysis().GetPointsToSet(hlo);
520 TF_CHECK_OK(points_to_analysis().VerifyBuffer(buffer));
699 const TuplePointsToAnalysis& points_to_analysis = local
700 liveness_->points_to_analysis();
701 for (LogicalBuffer::Id id = 0; id < points_to_analysis.num_logical_buffers();
703 auto& buffer = points_to_analysis.logical_buffer(id)
1534 const TuplePointsToAnalysis& points_to_analysis = local
[all...] |
| 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);
|
| heap_simulator.cc | 41 TF_ASSIGN_OR_RETURN(std::unique_ptr<TuplePointsToAnalysis> points_to_analysis,
52 schedule, *points_to_analysis, size_function));
59 const TuplePointsToAnalysis& points_to_analysis,
66 computation, sequence, points_to_analysis,
76 const TuplePointsToAnalysis& points_to_analysis,
83 *entry_computation, instruction_sequence, points_to_analysis));
91 const TuplePointsToAnalysis& points_to_analysis,
98 points_to_analysis));
107 const TuplePointsToAnalysis& points_to_analysis) {
142 points_to_analysis.GetPointsToSet(instruction)
[all...] |
| buffer_liveness.cc | 105 if (points_to_analysis().DoesNotUseOperandBuffer(alias.instruction(),
132 !points_to_analysis().CanShareOperandBufferWithUser(
|
| hlo_rematerialization.cc | 291 const TuplePointsToAnalysis& points_to_analysis,
298 points_to_analysis.GetBufferAliases(*logical_buffer)) {
300 if (points_to_analysis.DoesNotUseOperandBuffer(
330 const TuplePointsToAnalysis& points_to_analysis,
415 const TuplePointsToAnalysis& points_to_analysis,
420 points_to_analysis, &has_indirect_uses);
511 const TuplePointsToAnalysis& points_to_analysis,
515 points_to_analysis.GetPointsToSet(computation_->root_instruction())
524 points_to_analysis.GetBuffersDefinedByInstruction(instruction)) {
531 points_to_analysis.GetPointsToSet(instruction->operand(0))
[all...] |
| layout_assignment.h | 132 LayoutConstraints(const TuplePointsToAnalysis& points_to_analysis,
138 const TuplePointsToAnalysis& points_to_analysis() const { function in class:xla::LayoutConstraints
399 const TuplePointsToAnalysis& points_to_analysis,
|
| layout_assignment.cc | 104 const TuplePointsToAnalysis& points_to_analysis,
106 : points_to_analysis_(points_to_analysis), computation_(computation) {
501 constraints->points_to_analysis().GetBufferDefinedAt(instruction,
[all...] |
| hlo_memory_scheduler_test.cc | 298 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis = local
304 *points_to_analysis, size_fn)
310 *points_to_analysis, size_fn, &memory_by_computation)
|
| hlo_module_group_metadata.h | 210 TuplePointsToAnalysis* points_to_analysis(HloModule* module) const { function in class:xla::HloModuleGroupMetadata
|
| buffer_liveness_test.cc | 42 const auto& pointed_to = liveness.points_to_analysis()
75 auto& points_to_analysis = liveness.points_to_analysis(); local
77 points_to_analysis.GetPointsToSet(a).element(index);
79 points_to_analysis.GetPointsToSet(b).element(index);
[all...] |
| buffer_assignment.h | 438 const TuplePointsToAnalysis& points_to_analysis() const { function in class:xla::BufferAssignment
439 return liveness_->points_to_analysis();
|
| hlo_module_group_metadata.cc | 136 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis,
138 points_to_analyses_[module] = std::move(points_to_analysis);
|
| buffer_assignment_test.cc | 583 id < buffer_liveness.points_to_analysis().num_logical_buffers();
585 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id);
642 id < buffer_liveness.points_to_analysis().num_logical_buffers();
644 auto& buffer = buffer_liveness.points_to_analysis().logical_buffer(id);
646 buffer_liveness.points_to_analysis().GetBufferAliases(buffer);
[all...] |
| heap_simulator_test.cc | 209 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis = local
216 *points_to_analysis, size_fn, &memory_by_computation)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_layout_assignment.cc | 148 constraints->points_to_analysis().GetBufferDefinedAt(
232 constraints->points_to_analysis().GetBufferDefinedAt(instruction,
237 constraints->points_to_analysis().GetBufferDefinedAt(instruction,
298 constraints->points_to_analysis().GetBufferDefinedAt(
313 constraints->points_to_analysis().GetBufferDefinedAt(
|
| gpu_executable.cc | 358 return assignment_->points_to_analysis().GetPointsToSet(
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_executable.cc | 357 return assignment_->points_to_analysis().GetPointsToSet(
|
