Home
last modified time | relevance | path

Searched refs:paddedOp (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTransforms.cpp257 ArrayRef<bool> packPaddings, LinalgOp &paddedOp) { in rewriteAsPaddedOp() argument
289 paddedOp = opToPad.clone(b, loc, resultTensorTypes, newOperands); in rewriteAsPaddedOp()
295 for (const auto &en : llvm::enumerate(paddedOp->getResults())) { in rewriteAsPaddedOp()
413 LinalgOp paddedOp; in returningMatchAndRewrite() local
416 options.paddingValues, options.packPaddings, paddedOp); in returningMatchAndRewrite()
422 if (static_cast<int64_t>(en.index()) >= paddedOp.getNumInputsAndOutputs()) in returningMatchAndRewrite()
424 OpOperand *opOperand = &paddedOp->getOpOperand(en.index()); in returningMatchAndRewrite()
430 if (llvm::any_of(paddedOp.getShape(opOperand), ShapedType::isDynamic)) in returningMatchAndRewrite()
453 filter.replaceLinalgTransformationFilter(rewriter, paddedOp); in returningMatchAndRewrite()
455 return paddedOp; in returningMatchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h1204 ArrayRef<bool> packPaddings, LinalgOp &paddedOp);