Lines Matching refs:padOp
44 LogicalResult matchAndRewrite(tensor::PadOp padOp, in matchAndRewrite()
47 Value padValue = padOp.getConstantPaddingValue(); in matchAndRewrite()
49 return rewriter.notifyMatchFailure(padOp, "non constant padding"); in matchAndRewrite()
53 Value source = padOp.getSource(); in matchAndRewrite()
57 padOp, "expected source to be linalg.generic op"); in matchAndRewrite()
62 padOp, "only supported for ops with all parallel iterator types"); in matchAndRewrite()
66 dyn_cast<ReifyRankedShapedTypeOpInterface>(padOp.getOperation()); in matchAndRewrite()
71 padOp, "failed to get shape of pad op result"); in matchAndRewrite()
74 Location loc = padOp.getLoc(); in matchAndRewrite()
77 RankedTensorType padResultType = padOp.getResultType(); in matchAndRewrite()
93 SmallVector<OpFoldResult> offsets = padOp.getMixedLowPad(); in matchAndRewrite()
117 padOp, clonedOp.getResult(resultNumber), fillTensor.getResult(0), in matchAndRewrite()