pseudomo opened a new issue #9843: URL: https://github.com/apache/tvm/issues/9843
I am trying to optimize this model https://github.com/yucornetto/MGMatting ... model.load_state_dict(...) model = model.eval() model = torch.jit.trace(model, (image, mask), strict=False).eval() mod, params = relay.frontend.from_pytorch(model, [('image', image.shape), ('guidance', mask.shape)]) ... But _relay.frontend.from_pytorch_ fails with the following error: _The following operators are not implemented:_ [**'aten::dot', 'prim::DictConstruct', 'aten::mv'**] I could get rid of prim::DictConstruct operation but aten::dot and aten:mv are needed ### Environment Operating System: Windows 10 Torch version: 1.10.1+cpu TVM version: git commit 115919b120648fd0cb59efd0f429de1e335cb394 (18 Dec 2021) -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
