Lines Matching defs:lhsIndex

390                                  vector::ContractionOp op, int64_t lhsIndex,
976     int64_t lhsIndex = batchDimMap[0].first;
978     auto newOp = lowerParallel(rewriter, op, lhsIndex, rhsIndex, mask);
996   for (int64_t lhsIndex = 0, e = lhsType.getRank(); lhsIndex < e; ++lhsIndex) {
997     if (lhsContractingDimSet.count(lhsIndex) == 0) {
998       auto newOp = lowerParallel(rewriter, op, lhsIndex, /*rhsIndex=*/-1, mask);
1009       auto newOp = lowerParallel(rewriter, op, /*lhsIndex=*/-1, rhsIndex, mask);
1032                                                       int64_t lhsIndex,
1042   if (lhsIndex >= 0) {
1043     iterIndex = iMap[0].getDimPosition(lhsIndex);
1046         diag << "expected lhsIndex=" << lhsIndex << " and rhsIndex=" << rhsIndex
1049     if (lhsType.getScalableDims()[lhsIndex])
1051         diag << "Unrolling scalable dimension (lhsIndex=" << lhsIndex
1054     dimSize = lhsType.getDimSize(lhsIndex);
1066       diag << "expected either lhsIndex=" << lhsIndex
1097     auto lhs = reshapeLoad(loc, op.getLhs(), lhsType, lhsIndex, d, rewriter);
1139   int64_t lhsIndex = *lookupLhs;
1141   int64_t dimSize = lhsType.getDimSize(lhsIndex);
1144       diag << "expect LHS dimension " << lhsIndex
1175     auto lhs = reshapeLoad(loc, op.getLhs(), lhsType, lhsIndex, d, rewriter);