• 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
2rnn_layer_1timestep.json 21.9 KB