Code

[Frontend] [Torch] [ONNX] GRU layer#8781

Code

Merged

masahi merged 9 commits intoapache:mainapache/tvm:mainfrom Deelvin:vc/torch_GRUDeelvin/tvm:vc/torch_GRUCopy head branch name to clipboardAug 25, 2021

+774-441

Commits

Commits on Aug 24, 2021

• GRU cell was implemented in common.py. GRU was supported on pytorch frontend side

  

  Valery Chernov

  committed
  b6a6475

  Copy full SHA for b6a6475

• update GRU in common.py and onnx frontend

  

  Valery Chernov

  committed
  549ff68

  Copy full SHA for 549ff68

• fix issue related to GRU accuracy in pytorch and ONNX frontend

  

  Valery Chernov

  committed
  59eefa1

  Copy full SHA for 59eefa1

• small fixes and remove excess

  

  Valery Chernov

  committed
  d2f4914

  Copy full SHA for d2f4914

• common GRU was additionaly updated. tuned pytorch GRU was strongly accelerated

  

  Valery Chernov

  committed
  439d076

  Copy full SHA for 439d076

• GRU cell in ONNX frontend was used from common.py. previous implementation was removed

  

  Valery Chernov

  committed
  677eafb

  Copy full SHA for 677eafb

• small fixes in comments

  

  Valery Chernov

  committed
  1f889a3

  Copy full SHA for 1f889a3

• fixes after review. GRU test was implemented for pytorch frontend

  

  Valery Chernov

  committed
  0e19a6f

  Copy full SHA for 0e19a6f

• tests for RNN layers was unified for pytorch frontend

  

  Valery Chernov

  committed
  c722e27

  Copy full SHA for c722e27