Searched refs:tosaConv2DOp (Results 1 – 1 of 1) sorted by relevance
112 auto tosaConv2DOp = cast<tosa::Conv2DOp>(op); in matchAndRewrite() local115 tosaConv2DOp.input().getType().dyn_cast<mlir::RankedTensorType>(); in matchAndRewrite()122 tosaConv2DOp.weight().getType().dyn_cast<mlir::RankedTensorType>(); in matchAndRewrite()130 tosaConv2DOp.getResult().getType().dyn_cast<mlir::RankedTensorType>(); in matchAndRewrite()149 op->getLoc(), newTosaConv2DOpType, tosaConv2DOp.input(), in matchAndRewrite()150 tosaConv2DOp.weight(), tosaConv2DOp.bias(), tosaConv2DOp.pad(), in matchAndRewrite()151 tosaConv2DOp.stride(), tosaConv2DOp.dilation()); in matchAndRewrite()