33   matchAndRewrite(AnyOp op, OpAdaptor adaptor,
39 AnyOpConversion::matchAndRewrite(AnyOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
43   rewriter.replaceOp(op, {adaptor.getInputs().front()});  in matchAndRewrite()
54   matchAndRewrite(SrcOpTy op, typename SrcOpTy::Adaptor adaptor,  in matchAndRewrite()  argument
60     rewriter.replaceOpWithNewOp<DstOpTy>(op, adaptor.getLhs(),  in matchAndRewrite()
61                                          adaptor.getRhs());  in matchAndRewrite()
72   matchAndRewrite(BroadcastOp op, OpAdaptor adaptor,
121     BroadcastOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
138   llvm::append_range(ranks, llvm::map_range(adaptor.getShapes(), [&](Value v) {  in matchAndRewrite()
159             getBroadcastedDim(ImplicitLocOpBuilder(loc, b), adaptor.getShapes(),  in matchAndRewrite()
176   matchAndRewrite(ConstShapeOp op, OpAdaptor adaptor,
182     ConstShapeOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
210   matchAndRewrite(ConstSizeOp op, OpAdaptor adaptor,
216     ConstSizeOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
229   matchAndRewrite(IsBroadcastableOp op, OpAdaptor adaptor,
235     IsBroadcastableOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
253   llvm::append_range(ranks, llvm::map_range(adaptor.getShapes(), [&](Value v) {  in matchAndRewrite()
281             ImplicitLocOpBuilder(loc, b), adaptor.getShapes(), rankDiffs, iv);  in matchAndRewrite()
284         for (auto tup : llvm::zip(adaptor.getShapes(), rankDiffs)) {  in matchAndRewrite()
330   matchAndRewrite(GetExtentOp op, OpAdaptor adaptor,
336     GetExtentOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
347                                                  adaptor.getDim());  in matchAndRewrite()
353                                                  adaptor.getShape(),  in matchAndRewrite()
354                                                  ValueRange{adaptor.getDim()});  in matchAndRewrite()
364   matchAndRewrite(shape::RankOp op, OpAdaptor adaptor,
370 RankOpConverter::matchAndRewrite(shape::RankOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
376   rewriter.replaceOpWithNewOp<tensor::DimOp>(op, adaptor.getShape(), 0);  in matchAndRewrite()
387   matchAndRewrite(shape::ReduceOp op, OpAdaptor adaptor,
393 ReduceOpConverter::matchAndRewrite(shape::ReduceOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
405       rewriter.create<tensor::DimOp>(loc, indexTy, adaptor.getShape(), zero);  in matchAndRewrite()
410         Value extent = b.create<tensor::ExtractOp>(loc, adaptor.getShape(), iv);  in matchAndRewrite()
466   matchAndRewrite(ShapeEqOp op, OpAdaptor adaptor,
472 ShapeEqOpConverter::matchAndRewrite(ShapeEqOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
488   Value firstShape = adaptor.getShapes().front();  in matchAndRewrite()
493   for (Value shape : adaptor.getShapes().drop_front(1)) {  in matchAndRewrite()
536   matchAndRewrite(ShapeOfOp op, OpAdaptor adaptor,
542     ShapeOfOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
551   Value tensor = adaptor.getArg();  in matchAndRewrite()
599   matchAndRewrite(SplitAtOp op, OpAdaptor adaptor,
605     SplitAtOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
615   Value rank = b.create<tensor::DimOp>(adaptor.getOperand(), zero);  in matchAndRewrite()
618   Value originalIndex = adaptor.getIndex();  in matchAndRewrite()
626       b.create<tensor::ExtractSliceOp>(adaptor.getOperand(), zero, index, one);  in matchAndRewrite()
628   Value tail = b.create<tensor::ExtractSliceOp>(adaptor.getOperand(), index,  in matchAndRewrite()
641   matchAndRewrite(ToExtentTensorOp op, OpAdaptor adaptor,  in matchAndRewrite()  argument
643     if (!adaptor.getInput().getType().isa<RankedTensorType>())  in matchAndRewrite()
647                                                 adaptor.getInput());  in matchAndRewrite()