Lines Matching refs:convOp
850 linalg::Conv2DNhwcHwcfOp convOp, PatternRewriter &rewriter) const { in returningMatchAndRewrite() argument
851 if (failed(filter.checkAndNotify(rewriter, convOp))) in returningMatchAndRewrite()
853 if (convOp.hasBufferSemantics()) in returningMatchAndRewrite()
856 Value input = convOp.inputs().front(); in returningMatchAndRewrite()
857 Value kernel = convOp.inputs().back(); in returningMatchAndRewrite()
858 Value output = convOp.outputs().front(); in returningMatchAndRewrite()
887 Location loc = convOp.getLoc(); in returningMatchAndRewrite()
897 auto strides = llvm::to_vector<4>(convOp.strides().getValues<int64_t>()); in returningMatchAndRewrite()
901 auto dilations = llvm::to_vector<4>(convOp.dilations().getValues<int64_t>()); in returningMatchAndRewrite()
912 rewriter.replaceOp(convOp, inserted); in returningMatchAndRewrite()
920 DepthwiseConv2DNhwcHwcOp convOp, PatternRewriter &rewriter) const { in returningMatchAndRewrite() argument
921 if (failed(filter.checkAndNotify(rewriter, convOp))) in returningMatchAndRewrite()
923 if (convOp.hasBufferSemantics()) in returningMatchAndRewrite()
926 Value input = convOp.inputs().front(); in returningMatchAndRewrite()
927 Value kernel = convOp.inputs().back(); in returningMatchAndRewrite()
928 Value output = convOp.outputs().front(); in returningMatchAndRewrite()
957 Location loc = convOp.getLoc(); in returningMatchAndRewrite()
967 auto strides = llvm::to_vector<4>(convOp.strides().getValues<int64_t>()); in returningMatchAndRewrite()
971 auto dilations = llvm::to_vector<4>(convOp.dilations().getValues<int64_t>()); in returningMatchAndRewrite()
982 rewriter.replaceOp(convOp, inserted); in returningMatchAndRewrite()