• Home
  • Raw
  • Download

Lines Matching refs:linalgOp

1146   if (auto linalgOp = dyn_cast<LinalgOp>(parentOp))  in verify()  local
1650 auto linalgOp = dyn_cast<LinalgOp>(op); in matchAndRewrite() local
1651 if (!linalgOp) in matchAndRewrite()
1653 for (Value v : linalgOp.getInputsAndOutputBuffers()) { in matchAndRewrite()
1661 rewriter.eraseOp(linalgOp); in matchAndRewrite()
1675 auto linalgOp = dyn_cast<LinalgOp>(op); in matchAndRewrite() local
1676 if (!linalgOp) in matchAndRewrite()
1681 llvm::any_of(linalgOp.getShapedOperands(), [&](Value v) { in matchAndRewrite()
1695 for (Value v : linalgOp.getInputs()) { in matchAndRewrite()
1701 newOperands.append(linalgOp.getOutputBuffers().begin(), in matchAndRewrite()
1702 linalgOp.getOutputBuffers().end()); in matchAndRewrite()
1704 for (Value v : linalgOp.getInitTensors()) { in matchAndRewrite()
1710 auto extraOperands = linalgOp.getAssumedNonShapedOperands(); in matchAndRewrite()
1714 linalgOp.clone(rewriter, op->getLoc(), newResultTypes, newOperands); in matchAndRewrite()
1735 auto linalgOp = cast<LinalgOp>(op); in matchAndRewrite() local
1749 for (int i = 0, e = linalgOp.getNumInputs(); i != e; i++) { in matchAndRewrite()
1750 Value input = linalgOp.getInput(i); in matchAndRewrite()
1751 AffineMap indexingMap = linalgOp.getInputIndexingMap(i); in matchAndRewrite()
1760 if (canonicalInput.size() == linalgOp.getNumInputs()) in matchAndRewrite()
1765 for (auto v : llvm::enumerate(linalgOp.getInputs())) in matchAndRewrite()
1768 llvm::append_range(newOperands, linalgOp.getOutputBuffers()); in matchAndRewrite()
1769 llvm::append_range(newOperands, linalgOp.getInitTensors()); in matchAndRewrite()
1770 llvm::append_range(newOperands, linalgOp.getAssumedNonShapedOperands()); in matchAndRewrite()
1773 Operation *newOp = linalgOp.clone(rewriter, op->getLoc(), in matchAndRewrite()
1802 for (int i = 0, e = linalgOp.getNumInputs(); i < e; i++) { in matchAndRewrite()