Code

[Onnx] Support Bidirectional RNNs#8337

Code

Merged

mbrookhart merged 27 commits intoapache:mainapache/tvm:mainfrom AndrewZhaoLuo:andrewluo-onnx-add-bidirectional-lstmAndrewZhaoLuo/tvm:andrewluo-onnx-add-bidirectional-lstmCopy head branch name to clipboardJun 28, 2021

+543-339

Commits

Commits on Jun 28, 2021

• modify lstm to be easily bidirectional

  AndrewZhaoLuo

  committed
  56fd322

  Copy full SHA for 56fd322

• make it obvious some matriciies are packed via prime notation

  AndrewZhaoLuo

  committed
  2447d35

  Copy full SHA for 2447d35

• fix var name

  AndrewZhaoLuo

  committed
  85fd46d

  Copy full SHA for 85fd46d

• more var names

  AndrewZhaoLuo

  committed
  bcfb4d0

  Copy full SHA for bcfb4d0

• add op split

  AndrewZhaoLuo

  committed
  d713392

  Copy full SHA for d713392

• keyword arg names

  AndrewZhaoLuo

  committed
  c8f61a4

  Copy full SHA for c8f61a4

• missing implicit cls arg

  AndrewZhaoLuo

  committed
  26d1cd7

  Copy full SHA for 26d1cd7

• deal with extra dimensions

  AndrewZhaoLuo

  committed
  537727b

  Copy full SHA for 537727b

• last of the fixes

  AndrewZhaoLuo

  committed
  ac3bb92

  Copy full SHA for ac3bb92

• refactor rnn tests to support directions

  AndrewZhaoLuo

  committed
  a7ba617

  Copy full SHA for a7ba617

• bidirectional tests

  AndrewZhaoLuo

  committed
  a3dd53d

  Copy full SHA for a3dd53d

• test forward results

  AndrewZhaoLuo

  committed
  91c6483

  Copy full SHA for 91c6483

• go backwards

  AndrewZhaoLuo

  committed
  1cdc2ae

  Copy full SHA for 1cdc2ae

• more fixes

  AndrewZhaoLuo

  committed
  3382508

  Copy full SHA for 3382508

• reverse tokens on reverse pass

  AndrewZhaoLuo

  committed
  ebc7078

  Copy full SHA for ebc7078

• parameterized directions

  AndrewZhaoLuo

  committed
  023da4f

  Copy full SHA for 023da4f

• double up activations in bidirect

  AndrewZhaoLuo

  committed
  c0d1b5f

  Copy full SHA for c0d1b5f

• slow attribute forgetting

  AndrewZhaoLuo

  committed
  3859700

  Copy full SHA for 3859700

• lstm interface is v. confus

  AndrewZhaoLuo

  committed
  5f3b4be

  Copy full SHA for 5f3b4be

• test forward complete

  AndrewZhaoLuo

  committed
  3b2e486

  Copy full SHA for 3b2e486

• add GRU outline

  AndrewZhaoLuo

  committed
  fba5ba9

  Copy full SHA for fba5ba9

• revisiion2

  AndrewZhaoLuo

  committed
  4adca4b

  Copy full SHA for 4adca4b

• why was tehre a not

  AndrewZhaoLuo

  committed
  bf02a27

  Copy full SHA for bf02a27

• gru tests

  AndrewZhaoLuo

  committed
  7954480

  Copy full SHA for 7954480

• missing bounds, copy pasta!

  AndrewZhaoLuo

  committed
  d29011a

  Copy full SHA for d29011a

• add comment

  AndrewZhaoLuo

  committed
  855b54d

  Copy full SHA for 855b54d

• ensure all args fp

  AndrewZhaoLuo

  committed
  804bc7b

  Copy full SHA for 804bc7b