Use domain rank helper fn + rebase

Author: raikonenfnu

compiler/src/iree/compiler/Dialect/LinalgExt/Transforms/Padding.cpp

@@ -90,7 +90,7 @@ padAttention(IREE::LinalgExt::AttentionOp attnOp,
   Location loc = attnOp.getLoc();
   rewriter.setInsertionPoint(attnOp);
 
-  int64_t domainRank = maps[0].getNumDims();
+  int64_t domainRank = opInfo.getDomainRank();
   if (domainRank != 5) {
     return definiteFailureHelper(
         transformOp, attnOp,