Lines Matching refs:conv2d
128 Value conv2d; in matchAndRewrite() local
130 conv2d = rewriter.create<tosa::Conv2DOp>( in matchAndRewrite()
135 conv2d = rewriter.create<tosa::Conv2DOp>( in matchAndRewrite()
141 rewriter.replaceOp(op, conv2d); in matchAndRewrite()
290 Value conv2d; in matchAndRewrite() local
292 conv2d = createOpAndInfer<tosa::Conv2DOp>( in matchAndRewrite()
301 conv2d = createOpAndInfer<tosa::Conv2DOp>( in matchAndRewrite()
311 ShapedType convTy = conv2d.getType().cast<ShapedType>(); in matchAndRewrite()
320 conv2d = createOpAndInfer<tosa::ReshapeOp>( in matchAndRewrite()
321 rewriter, loc, UnrankedTensorType::get(resultETy), conv2d, in matchAndRewrite()
329 conv2d = createOpAndInfer<tosa::TransposeOp>( in matchAndRewrite()
330 rewriter, loc, UnrankedTensorType::get(convETy), conv2d, in matchAndRewrite()
336 conv2d = createOpAndInfer<tosa::ReshapeOp>( in matchAndRewrite()
337 rewriter, loc, UnrankedTensorType::get(resultETy), conv2d, in matchAndRewrite()
348 rewriter, loc, UnrankedTensorType::get(resultETy), conv2d, in matchAndRewrite()