Searched refs:extractOperand (Results 1 – 2 of 2) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorUnrollDistribute.cpp | 339 auto extractOperand = [&](unsigned index, Value operand, in matchAndRewrite() local
353 extractOperand(0, contractOp.getLhs(), lhsPermutationMap, lhsOffets); in matchAndRewrite()
356 extractOperand(3, contractOp.getMasks()[0], lhsPermutationMap, in matchAndRewrite()
363 extractOperand(1, contractOp.getRhs(), rhsPermutationMap, rhsOffets); in matchAndRewrite()
366 extractOperand(4, contractOp.getMasks()[1], rhsPermutationMap, in matchAndRewrite()
378 extractOperand(2, contractOp.getAcc(), accPermutationMap, accOffets); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 249 auto extractOperand = in matchAndRewrite() local
252 if (!extractOperand || !canFoldIntoProducerOp(tensorCast) || in matchAndRewrite()
259 SmallVector<OpFoldResult, 4> sizes = extractOperand.getMixedSizes(); in matchAndRewrite()
261 extractFromI64ArrayAttr(extractOperand.getStaticSizes()), in matchAndRewrite()
262 extractOperand.getType().getShape()); in matchAndRewrite()
275 extractOperand.getSource(), extractOperand.getMixedOffsets(), sizes, in matchAndRewrite()
276 extractOperand.getMixedStrides()); in matchAndRewrite()
|