Searched refs:lhsIndex (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 1960 int64_t lhsIndex = batchDimMap[0].first; in matchAndRewrite() local
1962 rewriter.replaceOp(op, lowerParallel(op, lhsIndex, rhsIndex, rewriter)); in matchAndRewrite()
1978 for (int64_t lhsIndex = 0, e = lhsType.getRank(); lhsIndex < e; ++lhsIndex) { in matchAndRewrite() local
1979 if (lhsContractingDimSet.count(lhsIndex) == 0) { in matchAndRewrite()
1981 op, lowerParallel(op, lhsIndex, /*rhsIndex=*/-1, rewriter)); in matchAndRewrite()
2008 int64_t lhsIndex, int64_t rhsIndex, in lowerParallel() argument
2017 if (lhsIndex >= 0) { in lowerParallel()
2018 iterIndex = iMap[0].getDimPosition(lhsIndex); in lowerParallel()
2021 dimSize = lhsType.getDimSize(lhsIndex); in lowerParallel()
2044 auto lhs = reshapeLoad(loc, op.lhs(), lhsType, lhsIndex, d, rewriter); in lowerParallel()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/Vector/ |
| D | VectorTransforms.h | 442 Value lowerParallel(vector::ContractionOp op, int64_t lhsIndex,
|