• J
    Added vanilla LSTM and LSTM with peepholes oneDNN fp32 kernel (#30661) · d834f4e6
    jakpiase 提交于
    * added external reorder to profiler
    
    * resolved conflict
    
    * added enable_static
    
    * initial version of lstm, not working yet
    
    * added lstm to operators.cmake
    
    * added vanilla lstm mkldnn op
    
    * added peephole weights integration
    
    * minor changes
    
    * added formatting
    
    * added fusion_lstm_mkldnn to static_whitelist
    
    * added formatting
    
    * removed comment
    
    * moved use_peepholes attribute inside is_cached block
    
    * reverted wrong changes
    
    * minor formatting change
    
    * minor changes
    d834f4e6
test_fusion_gru_mkldnn_op.py 2.0 KB