• Pruthvi's avatar
    Vanilla RNN Optimization (#4439) · 412f5f7b
    Pruthvi authored
    * WIP PM and callback for vanilla RNN
    
    * - Graph pass to fuse vanilla RNN Cell
    - Unit test case for vanilla rnn through TF serialized graph
    
    * - emit MKLDNN kernel for vanilla RNN ii) test case for Vanilla for cpu v/s inter
    
    * i) style check ii) serialized graph for tf vanilla rnn
    
    * fix warnings
    
    * i) fixed emitter code ii) test case passes
    
    * - added support for Vanilla RNN for MKLDNN > v1.0
    
    * fix compiler warnings
    
    * Fix build error
    
    * fix unit test case
    Co-authored-by: 's avatarRobert Kimball <robert.kimball@intel.com>
    Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
    412f5f7b
vanilla_rnn_3_time_step.json 16.3 KB