Lines Matching defs:conv2d
72 Value conv2d;
74 conv2d = rewriter.create<tosa::Conv2DOp>(
81 conv2d = rewriter.create<tosa::Conv2DOp>(
89 rewriter.replaceOp(op, conv2d);
231 Value conv2d;
233 conv2d = CreateOpAndInferShape<tosa::Conv2DOp>(
242 conv2d = CreateOpAndInferShape<tosa::Conv2DOp>(
253 ShapedType convTy = cast<ShapedType>(conv2d.getType());
262 conv2d = CreateOpAndInferShape<tosa::ReshapeOp>(
263 rewriter, loc, UnrankedTensorType::get(resultETy), conv2d,
271 conv2d = CreateOpAndInferShape<tosa::TransposeOp>(
272 rewriter, loc, UnrankedTensorType::get(convETy), conv2d,
278 conv2d = CreateOpAndInferShape<tosa::ReshapeOp>(
279 rewriter, loc, UnrankedTensorType::get(resultETy), conv2d,
304 rewriter, loc, UnrankedTensorType::get(resultETy), conv2d,