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 2020/09/15 07:44:24 UTC
[GitHub] [incubator-tvm] masahi opened a new issue #6474: [Torch] Support aten::lstm
masahi opened a new issue #6474:
URL: https://github.com/apache/incubator-tvm/issues/6474
It seems the commonly used [`torch.nn.LSTM`](https://pytorch.org/docs/stable/generated/torch.nn.LSTM.html) module generates `aten::lstm` op, which we currently don't support. It is reported in https://discuss.tvm.apache.org/t/import-rnn-t-pytorch-model-into-tvm/7874 and it is likely we'll get more requests for this op in the future.
The corresponding PyTorch implementation is here https://github.com/pytorch/pytorch/blob/70c8daf43946b53af6493d058899ef952d27d339/aten/src/ATen/native/RNN.cpp#L1396
----------------------------------------------------------------
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:
users@infra.apache.org
[GitHub] [incubator-tvm] zhangzhen507 commented on issue #6474: [Torch] Support aten::lstm
Posted by GitBox <gi...@apache.org>.
zhangzhen507 commented on issue #6474:
URL: https://github.com/apache/incubator-tvm/issues/6474#issuecomment-697153906
> yes, `torch.nn.LSTM` is translated to `aten::lstm` in Torchscript.
Hi, do you have a plan when you will support the aten::lstm in tvm/relay/frontend/pytorch.py ?
----------------------------------------------------------------
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:
users@infra.apache.org
[GitHub] [incubator-tvm] masahi commented on issue #6474: [Torch] Support aten::lstm
Posted by GitBox <gi...@apache.org>.
masahi commented on issue #6474:
URL: https://github.com/apache/incubator-tvm/issues/6474#issuecomment-697252375
No from my end. I'm not expert on LSTM and don't have time for this.
----------------------------------------------------------------
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:
users@infra.apache.org
[GitHub] [incubator-tvm] jroesch commented on issue #6474: [Torch] Support aten::lstm
Posted by GitBox <gi...@apache.org>.
jroesch commented on issue #6474:
URL: https://github.com/apache/incubator-tvm/issues/6474#issuecomment-693177498
I believe the MXNet frontend lowers LSTM to Relay directly.
----------------------------------------------------------------
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:
users@infra.apache.org
[GitHub] [incubator-tvm] masahi commented on issue #6474: [Torch] Support aten::lstm
Posted by GitBox <gi...@apache.org>.
masahi commented on issue #6474:
URL: https://github.com/apache/incubator-tvm/issues/6474#issuecomment-693771090
yes, `torch.nn.LSTM` is translated to `aten::lstm` in Torchscript.
----------------------------------------------------------------
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:
users@infra.apache.org
[GitHub] [incubator-tvm] zhangzhen507 commented on issue #6474: [Torch] Support aten::lstm
Posted by GitBox <gi...@apache.org>.
zhangzhen507 commented on issue #6474:
URL: https://github.com/apache/incubator-tvm/issues/6474#issuecomment-693766187
But, the pre-trained rnn-t model https://zenodo.org/record/3662521 is a pytorch model. Can you support the torch.nn.LSTM by aten::lstm in TVM ?
----------------------------------------------------------------
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:
users@infra.apache.org