Code

[Frontend, pytorch] Vc/pytorch lstm#8447

Code

Merged

masahi merged 11 commits intoapache:mainapache/tvm:mainfrom Deelvin:vc/pytorch_lstmDeelvin/tvm:vc/pytorch_lstmCopy head branch name to clipboardJul 20, 2021

+657

Commits

Commits on Jul 13, 2021

• lstm layer conversion to relay from pytorch model (TorchScript) was supported

  

  Valery Chernov

  committed
  e119ef3

  Copy full SHA for e119ef3

• bidirectional LSTM layer was supported for pytorch API

  

  Valery Chernov

  committed
  306be52

  Copy full SHA for 306be52

• lstm tests were implemented. fixes in pytorch lstm

  

  Valery Chernov

  committed
  f598bce

  Copy full SHA for f598bce

• fix pytorch bidirectional lstm. update test comment

  

  Valery Chernov

  committed
  367c5fe

  Copy full SHA for 367c5fe

• black format and some small fixes

  

  Valery Chernov

  committed
  57f7473

  Copy full SHA for 57f7473

• LSTM with projection was supported for pytorch frontend. test was updated by new combination of LSTM types

  

  Valery Chernov

  committed
  c73b80d

  Copy full SHA for c73b80d

• lint fixes

  

  Valery Chernov

  committed
  c976a8e

  Copy full SHA for c976a8e

Commits on Jul 14, 2021

• add bias switcher for LSTM types test. fix LSTM implementation in pytorch frontend for case without biases. exception in the test for conversion LSTM with projection from pytorch to ONNX

  

  Valery Chernov

  committed
  ee14d93

  Copy full SHA for ee14d93

Commits on Jul 16, 2021

• transfer test_lstms to pytest format

  

  Valery Chernov

  committed
  67d380e

  Copy full SHA for 67d380e

Commits on Jul 19, 2021

• onnx model saving was implemented through io.BytesIO. creating/removing tmp dir was removed. remove unneccessary comments

  

  Valery Chernov

  committed
  16964e5

  Copy full SHA for 16964e5

• gpu target was added to the test

  

  Valery Chernov

  committed
  a5396a4

  Copy full SHA for a5396a4