Lines Matching refs:opToPad
169 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()
249 return makeComposedPadHighOp(b, opToPad->getLoc(), paddedTensorType, 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()
270 for (OpOperand *opOperand : opToPad.getInputAndOutputOperands()) { in rewriteAsPaddedOp()
272 b, opToPad, opOperand, paddingDimensions, paddingValues, packPaddings); in rewriteAsPaddedOp()
280 if (failed(cast<ReifyRankedShapedTypeOpInterface>(opToPad.getOperation()) in rewriteAsPaddedOp()
283 assert(reifiedResultShapes.size() == opToPad->getNumResults() && in rewriteAsPaddedOp()
288 ValueRange(newOperands).take_back(opToPad.getNumOutputs()).getTypes(); in rewriteAsPaddedOp()
289 paddedOp = opToPad.clone(b, loc, resultTensorTypes, newOperands); in rewriteAsPaddedOp()
294 paddedSubviewResults.reserve(opToPad->getNumResults()); in rewriteAsPaddedOp()