Lines Matching refs:contractOp
313 LogicalResult matchAndRewrite(vector::ContractionOp contractOp, in matchAndRewrite()
315 auto targetShape = getTargetShape(options, contractOp); in matchAndRewrite()
318 auto dstVecType = contractOp.getResultType().cast<VectorType>(); in matchAndRewrite()
319 SmallVector<int64_t, 4> originalSize = *contractOp.getShapeForUnroll(); in matchAndRewrite()
321 Location loc = contractOp.getLoc(); in matchAndRewrite()
323 AffineMap dstAffineMap = contractOp.getIndexingMapsArray()[accIndex]; in matchAndRewrite()
330 contractOp.getIteratorTypes().size(), contractOp, options); in matchAndRewrite()
336 SmallVector<Value, 4> slicesOperands(contractOp.getNumOperands()); in matchAndRewrite()
350 AffineMap lhsPermutationMap = contractOp.getIndexingMapsArray()[0]; in matchAndRewrite()
353 extractOperand(0, contractOp.getLhs(), lhsPermutationMap, lhsOffets); in matchAndRewrite()
356 extractOperand(3, contractOp.getMasks()[0], lhsPermutationMap, in matchAndRewrite()
360 AffineMap rhsPermutationMap = contractOp.getIndexingMapsArray()[1]; in matchAndRewrite()
363 extractOperand(1, contractOp.getRhs(), rhsPermutationMap, rhsOffets); in matchAndRewrite()
366 extractOperand(4, contractOp.getMasks()[1], rhsPermutationMap, in matchAndRewrite()
369 AffineMap accPermutationMap = contractOp.getIndexingMapsArray()[2]; in matchAndRewrite()
378 extractOperand(2, contractOp.getAcc(), accPermutationMap, accOffets); in matchAndRewrite()
384 rewriter, loc, contractOp, slicesOperands, targetType); in matchAndRewrite()
400 rewriter.replaceOp(contractOp, result); in matchAndRewrite()