Skip to content

Commit

Permalink
[Torch] Support Python list, more realistic recurrent networks (#5306)
Browse files Browse the repository at this point in the history
* use funcs from prelude, pass around convert_map

* get relay input type from user ishape

* handle tuple unpack

* experimenting with static tensor array

* use prelude concat instead of cons + rev

* minor clean up

* fix layer norm conversion bug, unwrap tensor array

* add infer shape on tensor array

* pass around prelude for now

* compile worked but runtime error

* fix tensor array wrapping

* begin list dynamic test

* is_list_dynamic first version

* finish dynamic list test

* a few fix

* use shape_of function if Any is found

* improve size conversion

* working on adding free vars to loop block

* fixed inlined inner loop issue

* clean up free var handling

* add support for tensor array concat

* adding ta concat on last axis

* fix concat, but got runtime error

* disable concat on axis -1 for now

* add lstm tests

* revert unrelated change

* fix stacked bidir test

* minor fix to test

* relax tol a bit, revert dnnl change to avoid conflict

* simplify infer type, use input tensor shape rather than concat shape

* more shape fix
  • Loading branch information
masahi authored Apr 13, 2020
1 parent cd0d52d commit 0145cd5
Show file tree
Hide file tree
Showing 3 changed files with 787 additions and 199 deletions.
Loading

0 comments on commit 0145cd5

Please sign in to comment.