Searched refs:OperandLayout (Results 1 – 3 of 3) sorted by relevance
132 if (constraints->OperandLayout(instruction, operand_no) != nullptr) { in AddBackendConstraints()
161 const ShapeLayout* OperandLayout(const HloInstruction* instruction,
345 const ShapeLayout* LayoutConstraints::OperandLayout( in OperandLayout() function in xla::LayoutConstraints371 if (OperandLayout(instruction, i) != nullptr) { in ToString()373 "): ", OperandLayout(instruction, i)->ToString(), "\n"); in ToString()1574 if (constraints->OperandLayout(user, operand_no) == nullptr && in PropagateBufferConstraintToUses()1780 constraints.OperandLayout(instruction, operand_no); in AssignLayouts()