Skip to content

Commit

Permalink
Refine LSTMBlockCell to support dynamic rnn (#5963)
Browse files Browse the repository at this point in the history 
1. Refine conversion of `LSTMBlockCell`
       1) Make its output follows definition in TensorFlow
       2) Avoid introducing variables which doesn't match any placeholder nodes in TensorFlow graph

    2. About change in test_forward_ptb
       States nodes of LSTMBlockCell in this PB file  are actually Constant node.
       TF can feed data to those Constant nodes but relay can't do that, so current conversion of LSTMBockCell introduces extra variables to solve this issue.
       But this causes that relay IR doesn't match original TF graph. This PR solves this issue by convert those states node into placeholders.
  • Loading branch information
@lixiaoquan
lixiaoquan authored Jul 16, 2020
1 parent 30415e5 commit b85c239
Show file tree
Hide file tree
Showing 3 changed files with 231 additions and 345 deletions.
Loading

0 comments on commit b85c239

Please sign in to comment.