Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc51 bool PointsToSet::IsAmbiguous() const { in IsAmbiguous()
60 bool PointsToSet::IsDistinct() const { in IsDistinct()
74 size_t PointsToSet::size() const { in size()
80 PointsToSet::BufferSet PointsToSet::CreateFlattenedSet() const { in CreateFlattenedSet()
89 bool PointsToSet::ContainsBuffer(const LogicalBuffer& buffer) const { in ContainsBuffer()
100 bool PointsToSet::ContainsBufferAtIndex(const LogicalBuffer& buffer, in ContainsBufferAtIndex()
106 void PointsToSet::AddPointedToBuffer(const LogicalBuffer& buffer, in AddPointedToBuffer()
114 const PointsToSet::SourceSet& PointsToSet::tuple_sources( in tuple_sources()
119 void PointsToSet::add_tuple_source(const ShapeIndex& index, in add_tuple_source()
187 const PointsToSet& points_to_set = GetPointsToSet(instruction);
[all …]
Dtuple_points_to_analysis.h52 class PointsToSet {
57 explicit PointsToSet(const Shape* shape) : tree_(shape) {} in PointsToSet() function
153 TF_DISALLOW_COPY_AND_ASSIGN(PointsToSet);
192 const PointsToSet& GetPointsToSet(
289 PointsToSet& CreateEmptyPointsToSet(const HloInstruction* instruction);
293 PointsToSet& CreateCopiedPointsToSet(const HloInstruction* instruction,
306 std::unique_ptr<PointsToSet> points_to_set;
Dtuple_points_to_analysis_test.cc78 void ExpectHasBuffers(const PointsToSet::BufferList& points_to_set, in ExpectHasBuffers()
87 const PointsToSet::BufferList& points_to_set, in ExpectHasTopLevelBuffers()
89 PointsToSet::BufferList buffers; in ExpectHasTopLevelBuffers()
98 const PointsToSet::BufferSet& points_to_set, in ExpectHasTopLevelBuffers()
101 PointsToSet::BufferList(points_to_set.begin(), points_to_set.end()), in ExpectHasTopLevelBuffers()
163 const PointsToSet& tuple_points_to_set = in TEST_F()
Dlayout_assignment.h209 PointsToSet::BufferSet* GetBufferSet(const HloInstruction* instruction) const;
234 std::unique_ptr<PointsToSet::BufferSet>>
Dlayout_assignment.cc115 [&](const ShapeIndex&, const PointsToSet::BufferList& buffers) { in LayoutConstraints()
128 PointsToSet::BufferSet* LayoutConstraints::GetBufferSet( in GetBufferSet()
136 .emplace(instruction, absl::make_unique<PointsToSet::BufferSet>()) in GetBufferSet()
141 const PointsToSet::BufferList& buffers) { in GetBufferSet()
151 PointsToSet::BufferSet* output_buffers = GetBufferSet(instruction); in OperandBufferForwarded()
152 PointsToSet::BufferSet* operand_buffers = in OperandBufferForwarded()
943 const PointsToSet& points_to_set = in CheckLayouts()
947 const PointsToSet::BufferList& buffers) -> Status { in CheckLayouts()
1257 const PointsToSet& points_to_set = in PropagateUseConstraintToDefs()
1262 const PointsToSet::BufferList& buffers) -> Status { in PropagateUseConstraintToDefs()
Dhlo_memory_scheduler.cc124 const PointsToSet::BufferList& buffers) { in ListScheduler()
Dhlo_rematerialization.cc793 PointsToSet::BufferSet live_out_set = in MemoryUsageTracker()
808 const PointsToSet& operand_points_to = in MemoryUsageTracker()