jwfromm opened a new pull request #7747: URL: https://github.com/apache/tvm/pull/7747
Although not officially supported by Onnx, Pytorch will sometimes produce graphs with Aten nodes when it cant figure out how to correctly export. One example of this is DLRM models, which use EmbeddingBag layers. This PR introduces a converter for handling ATen operators and adds the ability to import `Aten:EmbeddingBag` Nodes. This allows TVM to correctly import full DLRM graphs which is pretty neat. -- 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. For queries about this service, please contact Infrastructure at: [email protected]
