Home
last modified time | relevance | path

Searched refs:PointsToSet (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc48 bool PointsToSet::IsAmbiguous() const { in IsAmbiguous()
57 bool PointsToSet::IsDistinct() const { in IsDistinct()
71 size_t PointsToSet::size() const { in size()
77 PointsToSet::BufferSet PointsToSet::CreateFlattenedSet() const { in CreateFlattenedSet()
86 bool PointsToSet::ContainsBuffer(const LogicalBuffer& buffer) const { in ContainsBuffer()
97 bool PointsToSet::ContainsBufferAtIndex(const LogicalBuffer& buffer, in ContainsBufferAtIndex()
103 void PointsToSet::AddPointedToBuffer(const LogicalBuffer& buffer, in AddPointedToBuffer()
111 const PointsToSet::SourceSet& PointsToSet::tuple_sources( in tuple_sources()
116 void PointsToSet::add_tuple_source(const ShapeIndex& index, in add_tuple_source()
184 const PointsToSet& points_to_set = GetPointsToSet(instruction);
[all …]
Dtuple_points_to_analysis.h51 class PointsToSet {
56 explicit PointsToSet(const Shape* shape) : tree_(shape) {} in PointsToSet() function
152 TF_DISALLOW_COPY_AND_ASSIGN(PointsToSet);
191 const PointsToSet& GetPointsToSet(
294 PointsToSet& CreateEmptyPointsToSet(const HloInstruction* instruction);
298 PointsToSet& CreateCopiedPointsToSet(const HloInstruction* instruction,
311 std::unique_ptr<PointsToSet> points_to_set;
Dbuffer_liveness.h56 const PointsToSet::BufferSet& maybe_live_out_buffers() const { in maybe_live_out_buffers()
107 PointsToSet::BufferSet maybe_live_out_buffers_;
Dlayout_assignment.cc110 [&](const ShapeIndex&, const PointsToSet::BufferList& buffers) { in LayoutConstraints()
123 PointsToSet::BufferSet* LayoutConstraints::GetBufferSet( in GetBufferSet()
131 .emplace(instruction, absl::make_unique<PointsToSet::BufferSet>()) in GetBufferSet()
136 const PointsToSet::BufferList& buffers) { in GetBufferSet()
146 PointsToSet::BufferSet* output_buffers = GetBufferSet(instruction); in OperandBufferForwarded()
147 PointsToSet::BufferSet* operand_buffers = in OperandBufferForwarded()
890 const PointsToSet& points_to_set = in CheckLayouts()
894 const PointsToSet::BufferList& buffers) -> Status { in CheckLayouts()
1199 const PointsToSet& points_to_set = in PropagateUseConstraintToDefs()
1204 const PointsToSet::BufferList& buffers) -> Status { in PropagateUseConstraintToDefs()
Dlayout_assignment.h207 PointsToSet::BufferSet* GetBufferSet(const HloInstruction* instruction) const;
232 std::unique_ptr<PointsToSet::BufferSet>>
Dtuple_points_to_analysis_test.cc76 void ExpectHasBuffers(const PointsToSet::BufferList& points_to_set, in ExpectHasBuffers()
85 const PointsToSet::BufferList& points_to_set, in ExpectHasTopLevelBuffers()
87 PointsToSet::BufferList buffers; in ExpectHasTopLevelBuffers()
96 const PointsToSet::BufferSet& points_to_set, in ExpectHasTopLevelBuffers()
99 PointsToSet::BufferList(points_to_set.begin(), points_to_set.end()), in ExpectHasTopLevelBuffers()
161 const PointsToSet& tuple_points_to_set = in TEST_F()
Dbuffer_assignment.h410 const PointsToSet::BufferList& GetSourceBuffers( in GetSourceBuffers()
499 const PointsToSet& GetPointsToSet(const HloInstruction* instruction) const;
Dheap_simulator.cc141 const PointsToSet& points_to = in RunComputation()
143 const PointsToSet::BufferSet& buffer_set = points_to.CreateFlattenedSet(); in RunComputation()
157 const PointsToSet& gte_points_to = in RunComputation()
Dbuffer_assignment.cc336 const PointsToSet& BufferAssignment::GetPointsToSet( in GetPointsToSet()
377 const PointsToSet& points_to_set = points_to_analysis().GetPointsToSet(hlo); in GetInstructionAllocation()
770 const PointsToSet& init_set = in MayInterfereAcrossSubcomputations()
778 const PointsToSet& while_set = in MayInterfereAcrossSubcomputations()
Dhlo_rematerialization.cc514 PointsToSet::BufferSet live_out_set = in MemoryUsageTracker()
530 const PointsToSet& operand_points_to = in MemoryUsageTracker()
Dhlo_memory_scheduler.cc121 const PointsToSet::BufferList& buffers) { in ListScheduler()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.h101 const PointsToSet& GetRootPointsToSet() const;
Dgpu_executable.cc357 const PointsToSet& GpuExecutable::GetRootPointsToSet() const { in GetRootPointsToSet()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.h133 const PointsToSet& GetRootPointsToSet() const;
Dcpu_executable.cc356 const PointsToSet& CpuExecutable::GetRootPointsToSet() const { in GetRootPointsToSet()