Searched refs:OperandLayoutConstraint (Results 1 – 2 of 2) sorted by relevance
108 class OperandLayoutConstraint : public LayoutConstraint {110 OperandLayoutConstraint(const ShapeLayout& shape_layout,278 const OperandLayoutConstraint* GetOperandLayoutConstraint(281 OperandLayoutConstraint* InsertOperandLayoutConstraint(283 const OperandLayoutConstraint& constraint);300 std::map<OperandConstraintKey, OperandLayoutConstraint>401 const OperandLayoutConstraint& operand_constraint,
104 OperandLayoutConstraint::OperandLayoutConstraint( in OperandLayoutConstraint() function in xla::OperandLayoutConstraint119 std::string OperandLayoutConstraint::ToString() const { in ToString()261 const OperandLayoutConstraint* curr_shape_layout = in SetOperandLayout()289 OperandLayoutConstraint new_constraint(ShapeLayout(shape_with_layout), in SetOperandLayout()298 OperandLayoutConstraint*301 const OperandLayoutConstraint& constraint) { in InsertOperandLayoutConstraint()448 const OperandLayoutConstraint*1436 dynamic_cast<const OperandLayoutConstraint*>( in PropagateConstraints()1532 const OperandLayoutConstraint& operand_constraint, in PropagateOperandConstraint()1624 const OperandLayoutConstraint* constraint = in PropagateOperandConstraint()[all …]