You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2022/07/25 07:35:49 UTC

[GitHub] [tvm] zgplvyou commented on issue #7986: [PyTorch] ImplementationError: The following operators are not implemented: ['aten::embedding_bag']

zgplvyou commented on issue #7986:
URL: https://github.com/apache/tvm/issues/7986#issuecomment-1193689834

   > embedding_bag is supported in relay/frontend/onnx.py, you can refer to this file.
   
   Hi, I`m also relay from onnx dlrm model, in this file(onnx.py), the implementation of embedding_bag use a reshape op(        indices = _op.reshape(indices, indices_shape)), however, if the indices_shape is [3, -1],  the size of indices is [13, ], the indices can not be reshape to the size[3, ?],how to do it!


-- 
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: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org