Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTransforms.cpp169 OpBuilder &b, linalg::LinalgOp opToPad, OpOperand *opOperand, in padOperandToSmallestStaticBoundingBox() argument
172 AffineMap indexingMap = opToPad.getTiedIndexingMap(opOperand); in padOperandToSmallestStaticBoundingBox()
173 ArrayRef<int64_t> shape = opToPad.getShape(opOperand); in padOperandToSmallestStaticBoundingBox()
198 opToPad.getLoc(), paddingAttr.getType(), paddingAttr); in padOperandToSmallestStaticBoundingBox()
254 linalg::rewriteAsPaddedOp(OpBuilder &b, LinalgOp opToPad, in rewriteAsPaddedOp() argument
258 Location loc = opToPad->getLoc(); in rewriteAsPaddedOp()
261 assert(opToPad.hasTensorSemantics() && in rewriteAsPaddedOp()
266 b.setInsertionPointAfter(opToPad); in rewriteAsPaddedOp()
269 newOperands.reserve(opToPad.getNumInputsAndOutputs()); in rewriteAsPaddedOp()
283 assert(reifiedResultShapes.size() == opToPad->getNumResults() && in rewriteAsPaddedOp()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h1201 rewriteAsPaddedOp(OpBuilder &b, LinalgOp opToPad,