Lines Matching defs:rewriter

51 FailureOr<Operation *> transposeConv2DHelper(RewriterBase &rewriter,
76     input = rewriter.create<tensor::EmptyOp>(loc, newFilterShape, elementTy)
79     input = rewriter
87       rewriter.create<linalg::TransposeOp>(loc, filter, input, filterPerm);
107       rewriter.create<HWCFConvOp>(loc, resultTy, newInputs, op.getOutputs(),
109   rewriter.replaceOp(op, newConv);
118                                 PatternRewriter &rewriter) const final {
119     if (failed(transposeConv2DHelper<FHWCConvOp, HWCFConvOp>(rewriter, op))) {
127 FailureOr<Operation *> transposeConv2D(RewriterBase &rewriter,
131                                linalg::Conv2DNhwcHwcfOp>(rewriter, op);
134 FailureOr<Operation *> transposeConv2D(RewriterBase &rewriter,
138                                linalg::Conv2DNhwcHwcfQOp>(rewriter, op);