Searched refs:BufferSet (Results 1 – 8 of 8) sorted by relevance
56 const PointsToSet::BufferSet& maybe_live_out_buffers() const { in maybe_live_out_buffers()107 PointsToSet::BufferSet maybe_live_out_buffers_;
72 using BufferSet = tensorflow::gtl::CompactPointerSet<const LogicalBuffer*>; variable73 BufferSet CreateFlattenedSet() const;
207 PointsToSet::BufferSet* GetBufferSet(const HloInstruction* instruction) const;232 std::unique_ptr<PointsToSet::BufferSet>>
77 PointsToSet::BufferSet PointsToSet::CreateFlattenedSet() const { in CreateFlattenedSet()78 BufferSet flat_set; in CreateFlattenedSet()
123 PointsToSet::BufferSet* LayoutConstraints::GetBufferSet( in GetBufferSet()131 .emplace(instruction, absl::make_unique<PointsToSet::BufferSet>()) in GetBufferSet()146 PointsToSet::BufferSet* output_buffers = GetBufferSet(instruction); in OperandBufferForwarded()147 PointsToSet::BufferSet* operand_buffers = in OperandBufferForwarded()
143 const PointsToSet::BufferSet& buffer_set = points_to.CreateFlattenedSet(); in RunComputation()
514 PointsToSet::BufferSet live_out_set = in MemoryUsageTracker()
96 const PointsToSet::BufferSet& points_to_set, in ExpectHasTopLevelBuffers()