Skip to content

Commit

Permalink
Refine LSTMBlockCell to support dynamic rnn
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 committed Jun 30, 2020
1 parent 2e04393 commit 7460648
Show file tree
Hide file tree
Showing 3 changed files with 230 additions and 345 deletions.
Loading

0 comments on commit 7460648

Please sign in to comment.