Lines Matching refs:contractOp
288 LogicalResult matchAndRewrite(vector::ContractionOp contractOp, in matchAndRewrite()
290 VectorType oldAccType = contractOp.getAccType().dyn_cast<VectorType>(); in matchAndRewrite()
296 if (llvm::size(contractOp.getMasks()) != 0) in matchAndRewrite()
304 auto oldIndexingMaps = contractOp.getIndexingMapsArray(); in matchAndRewrite()
307 auto oldIteratorTypes = contractOp.getIteratorTypes(); in matchAndRewrite()
323 SmallVector<Value> operands = {contractOp.getLhs(), contractOp.getRhs(), in matchAndRewrite()
324 contractOp.getAcc()}; in matchAndRewrite()
361 contractOp.getContext()); in matchAndRewrite()
363 contractOp.getLoc(), operands[it.index()], perm); in matchAndRewrite()
383 contractOp.getContext())); in matchAndRewrite()
388 contractOp.getLoc(), operands[it.index()], in matchAndRewrite()
393 contractOp.getLoc(), newOperands[0], newOperands[1], newOperands[2], in matchAndRewrite()
395 rewriter.getArrayAttr(newIteratorTypes), contractOp.getKind()); in matchAndRewrite()
397 contractOp, contractOp->getResultTypes()[0], newContractOp); in matchAndRewrite()