Searched refs:PointsToSet (Results 1 – 15 of 15) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | tuple_points_to_analysis.cc | 48 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 …]
|
D | tuple_points_to_analysis.h | 51 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;
|
D | buffer_liveness.h | 56 const PointsToSet::BufferSet& maybe_live_out_buffers() const { in maybe_live_out_buffers() 107 PointsToSet::BufferSet maybe_live_out_buffers_;
|
D | layout_assignment.cc | 110 [&](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()
|
D | layout_assignment.h | 207 PointsToSet::BufferSet* GetBufferSet(const HloInstruction* instruction) const; 232 std::unique_ptr<PointsToSet::BufferSet>>
|
D | tuple_points_to_analysis_test.cc | 76 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()
|
D | buffer_assignment.h | 410 const PointsToSet::BufferList& GetSourceBuffers( in GetSourceBuffers() 499 const PointsToSet& GetPointsToSet(const HloInstruction* instruction) const;
|
D | heap_simulator.cc | 141 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()
|
D | buffer_assignment.cc | 336 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()
|
D | hlo_rematerialization.cc | 514 PointsToSet::BufferSet live_out_set = in MemoryUsageTracker() 530 const PointsToSet& operand_points_to = in MemoryUsageTracker()
|
D | hlo_memory_scheduler.cc | 121 const PointsToSet::BufferList& buffers) { in ListScheduler()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_executable.h | 101 const PointsToSet& GetRootPointsToSet() const;
|
D | gpu_executable.cc | 357 const PointsToSet& GpuExecutable::GetRootPointsToSet() const { in GetRootPointsToSet()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_executable.h | 133 const PointsToSet& GetRootPointsToSet() const;
|
D | cpu_executable.cc | 356 const PointsToSet& CpuExecutable::GetRootPointsToSet() const { in GetRootPointsToSet()
|