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