Lines Matching refs:opOperand
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()
184 bool nofold = opOperand->getOperandNumber() < packPaddings.size() in padOperandToSmallestStaticBoundingBox()
185 ? packPaddings[opOperand->getOperandNumber()] in padOperandToSmallestStaticBoundingBox()
191 return opOperand->get(); in padOperandToSmallestStaticBoundingBox()
194 if (opOperand->getOperandNumber() >= paddingValues.size()) in padOperandToSmallestStaticBoundingBox()
196 Attribute paddingAttr = paddingValues[opOperand->getOperandNumber()]; in padOperandToSmallestStaticBoundingBox()
201 OpOperand *currOpOperand = opOperand; in padOperandToSmallestStaticBoundingBox()
248 paddedShape, getElementTypeOrSelf(opOperand->get())); in padOperandToSmallestStaticBoundingBox()
250 opOperand->get(), paddingValue, nofold); in padOperandToSmallestStaticBoundingBox()
270 for (OpOperand *opOperand : opToPad.getInputAndOutputOperands()) { in rewriteAsPaddedOp()
272 b, opToPad, opOperand, paddingDimensions, paddingValues, packPaddings); in rewriteAsPaddedOp()
424 OpOperand *opOperand = &paddedOp->getOpOperand(en.index()); in returningMatchAndRewrite() local
425 auto padOp = opOperand->get().getDefiningOp<tensor::PadOp>(); in returningMatchAndRewrite()
430 if (llvm::any_of(paddedOp.getShape(opOperand), ShapedType::isDynamic)) in returningMatchAndRewrite()