Searched refs:GetBufferSet (Results 1 – 2 of 2) sorted by relevance
207 PointsToSet::BufferSet* GetBufferSet(const HloInstruction* instruction) const;
123 PointsToSet::BufferSet* LayoutConstraints::GetBufferSet( in GetBufferSet() function in xla::LayoutConstraints146 PointsToSet::BufferSet* output_buffers = GetBufferSet(instruction); in OperandBufferForwarded()148 GetBufferSet(instruction->operand(operand_no)); in OperandBufferForwarded()