Lines Matching defs:rhsIndex

391                                  int64_t rhsIndex, Value mask) const;
977     int64_t rhsIndex = batchDimMap[0].second;
978     auto newOp = lowerParallel(rewriter, op, lhsIndex, rhsIndex, mask);
998       auto newOp = lowerParallel(rewriter, op, lhsIndex, /*rhsIndex=*/-1, mask);
1007   for (int64_t rhsIndex = 0, e = rhsType.getRank(); rhsIndex < e; ++rhsIndex) {
1008     if (rhsContractingDimSet.count(rhsIndex) == 0) {
1009       auto newOp = lowerParallel(rewriter, op, /*lhsIndex=*/-1, rhsIndex, mask);
1033                                                       int64_t rhsIndex,
1044     if (rhsIndex >= 0 && iterIndex != iMap[1].getDimPosition(rhsIndex))
1046         diag << "expected lhsIndex=" << lhsIndex << " and rhsIndex=" << rhsIndex
1055   } else if (rhsIndex >= 0) {
1056     iterIndex = iMap[1].getDimPosition(rhsIndex);
1057     if (rhsType.getScalableDims()[rhsIndex])
1059         diag << "Unrolling scalable dimension (rhsIndex=" << rhsIndex
1062     dimSize = rhsType.getDimSize(rhsIndex);
1067            << " or rhsIndex=" << rhsIndex << " to be nonnegative";
1098     auto rhs = reshapeLoad(loc, op.getRhs(), rhsType, rhsIndex, d, rewriter);
1140   int64_t rhsIndex = *lookupRhs;
1142   if (dimSize != rhsType.getDimSize(rhsIndex))
1145            << " to have the same size as RHS dimension " << rhsIndex;
1176     auto rhs = reshapeLoad(loc, op.getRhs(), rhsType, rhsIndex, d, rewriter);