Skip to content

N
ngraph
Switch branch/tag
History Find file
  • Pruthvi's avatar
    RNN fusion across layers (#1085) · f75b8006
    Pruthvi authored 
    * - Added graph pass for fusing RNN op across layer
- Added test case for inter v/s cpu for verifying layer fused RNN
- more sanity checks in the RNN fusion graph pass
- added support to replace the recurrent cell state correctly in the fused RNN op

* Fixed multi layer rnn fusion unit test failure

* Addressed PR comments
    f75b8006
Name
Last commit
 Last update
..
10_bucket_LSTM.json Loading commit data...
1_lstm_cell_forward.json Loading commit data...
1rnn_layer_3lstm_cell.json Loading commit data...
2rnn_layer_1timestep.json Loading commit data...
2rnn_layer_3lstm_cell.json Loading commit data...
3_lstm_cell_forward.json Loading commit data...
Graph_fprop_sigmoid.json Loading commit data...
LSTM_backward.json Loading commit data...
LSTM_forward.json Loading commit data...
Seq2Seq_backward.json Loading commit data...
Seq2Seq_forward.json Loading commit data...
Sockeye_Seq2Seq_backward.json Loading commit data...
Sockeye_Seq2Seq_forward.json Loading commit data...
batch_dot_3.json Loading commit data...
bn_bprop.json Loading commit data...
bn_fprop.json Loading commit data...
bn_fprop_b2c3h2w2.json Loading commit data...
mnist_mlp_forward.json Loading commit data...
rnn-10-step-fusion-test.json Loading commit data...
tranpose.json Loading commit data...