37 TosaOp createOpAndInfer(PatternRewriter &rewriter, Location loc, Type resultTy,  in createOpAndInfer()  argument
39   auto op = rewriter.create<TosaOp>(loc, resultTy, args...);  in createOpAndInfer()
87                                 PatternRewriter &rewriter) const final {  in matchAndRewrite()
123     auto reverse1 = rewriter.create<tosa::ReverseOp>(  in matchAndRewrite()
124         loc, weightTy, weight, rewriter.getI64IntegerAttr(1));  in matchAndRewrite()
125     auto reverse2 = rewriter.create<tosa::ReverseOp>(  in matchAndRewrite()
126         loc, weightTy, reverse1, rewriter.getI64IntegerAttr(2));  in matchAndRewrite()
130       conv2d = rewriter.create<tosa::Conv2DOp>(  in matchAndRewrite()
132           rewriter.getI64ArrayAttr(convPad), rewriter.getI64ArrayAttr(stride),  in matchAndRewrite()
133           rewriter.getI64ArrayAttr({1, 1}), *op.getQuantizationInfo());  in matchAndRewrite()
135       conv2d = rewriter.create<tosa::Conv2DOp>(  in matchAndRewrite()
137           rewriter.getI64ArrayAttr(convPad), rewriter.getI64ArrayAttr(stride),  in matchAndRewrite()
138           rewriter.getI64ArrayAttr({1, 1}));  in matchAndRewrite()
141     rewriter.replaceOp(op, conv2d);  in matchAndRewrite()
151                                 PatternRewriter &rewriter) const final {  in matchAndRewrite()
199         RankedTensorType::get({4, 2}, rewriter.getI32Type()), weightPadding);  in matchAndRewrite()
201         rewriter, loc, weightPaddingAttr.getType(), weightPaddingAttr);  in matchAndRewrite()
206           rewriter, loc, UnrankedTensorType::get(weightETy), weight,  in matchAndRewrite()
208           rewriter.getAttr<PadOpQuantizationAttr>(quantInfo.getWeightZp()));  in matchAndRewrite()
211       weight = createOpAndInfer<tosa::PadOp>(rewriter, loc,  in matchAndRewrite()
226         rewriter, loc, UnrankedTensorType::get(weightETy), weight,  in matchAndRewrite()
227         rewriter.getI64ArrayAttr(weightReshapeDims0));  in matchAndRewrite()
230     Value transposeWeightVal = rewriter.create<tosa::ConstOp>(  in matchAndRewrite()
231         loc, RankedTensorType::get({6}, rewriter.getI32Type()),  in matchAndRewrite()
232         rewriter.getI32TensorAttr({2, 4, 0, 1, 3, 5}));  in matchAndRewrite()
235         rewriter, loc, UnrankedTensorType::get(weightETy), weight,  in matchAndRewrite()
243         rewriter, loc, UnrankedTensorType::get(weightETy), weight,  in matchAndRewrite()
244         rewriter.getI64ArrayAttr(weightReshapeDims1));  in matchAndRewrite()
248         rewriter, loc, UnrankedTensorType::get(weightETy), weight,  in matchAndRewrite()
249         rewriter.getI64IntegerAttr(1));  in matchAndRewrite()
251         rewriter, loc, UnrankedTensorType::get(weightETy), weight,  in matchAndRewrite()
252         rewriter.getI64IntegerAttr(2));  in matchAndRewrite()
262         RankedTensorType::get({4, 2}, rewriter.getI32Type()), inputPadding);  in matchAndRewrite()
265         rewriter, loc, inputPaddingAttr.getType(), inputPaddingAttr);  in matchAndRewrite()
270           rewriter, loc, UnrankedTensorType::get(inputETy), input,  in matchAndRewrite()
272           rewriter.getAttr<PadOpQuantizationAttr>(quantInfo.getInputZp()));  in matchAndRewrite()
274       input = createOpAndInfer<tosa::PadOp>(rewriter, loc,  in matchAndRewrite()
280     auto zeroBias = rewriter.create<tosa::ConstOp>(  in matchAndRewrite()
287             rewriter.getZeroAttr(biasETy)));  in matchAndRewrite()
293                    rewriter, loc, UnrankedTensorType::get(resultETy), input,  in matchAndRewrite()
295                    /*pad=*/rewriter.getI64ArrayAttr({0, 0, 0, 0}),  in matchAndRewrite()
296                    /*stride=*/rewriter.getI64ArrayAttr({1, 1}),  in matchAndRewrite()
297                    /*dilation=*/rewriter.getI64ArrayAttr({1, 1}),  in matchAndRewrite()
302                    rewriter, loc, UnrankedTensorType::get(resultETy), input,  in matchAndRewrite()
304                    /*pad=*/rewriter.getI64ArrayAttr({0, 0, 0, 0}),  in matchAndRewrite()
305                    /*stride=*/rewriter.getI64ArrayAttr({1, 1}),  in matchAndRewrite()
306                    /*dilation=*/rewriter.getI64ArrayAttr({1, 1}))  in matchAndRewrite()
321         rewriter, loc, UnrankedTensorType::get(resultETy), conv2d,  in matchAndRewrite()
322         rewriter.getI64ArrayAttr(convReshapeDims0));  in matchAndRewrite()
325     Value transposeConvVal = rewriter.create<tosa::ConstOp>(  in matchAndRewrite()
326         loc, RankedTensorType::get({6}, rewriter.getI32Type()),  in matchAndRewrite()
327         rewriter.getI32TensorAttr({0, 1, 3, 2, 4, 5}));  in matchAndRewrite()
330         rewriter, loc, UnrankedTensorType::get(convETy), conv2d,  in matchAndRewrite()
337         rewriter, loc, UnrankedTensorType::get(resultETy), conv2d,  in matchAndRewrite()
338         rewriter.getI64ArrayAttr(convReshapeDims1));  in matchAndRewrite()
348                      rewriter, loc, UnrankedTensorType::get(resultETy), conv2d,  in matchAndRewrite()
349                      rewriter.getI64ArrayAttr(sliceBegin),  in matchAndRewrite()
350                      rewriter.getI64ArrayAttr(resultTy.getShape()))  in matchAndRewrite()
354         createOpAndInfer<tosa::AddOp>(rewriter, loc, op.getType(), slice, bias);  in matchAndRewrite()
356     rewriter.replaceOp(op, addBias.getResult());  in matchAndRewrite()