1. 27 1月, 2021 1 次提交
    • J
      REUPLOAD Added vanilla LSTM and LSTM with peepholes oneDNN fp32 kernel (#30719) · f8da5536
      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
      
      * changed stream handling
      
      * minor change
      
      * added datatype to GetExpectedKernelType()
      
      * added reading stream from TLS
      f8da5536
  2. 26 1月, 2021 2 次提交
  3. 25 1月, 2021 1 次提交
  4. 14 12月, 2020 1 次提交
  5. 19 11月, 2020 1 次提交
  6. 09 10月, 2020 1 次提交
  7. 01 10月, 2020 1 次提交
  8. 14 9月, 2020 1 次提交
  9. 12 9月, 2020 1 次提交
    • L
      Fix GRU mkldnn kernel fail on look_table_v2 (#27198) · 5c4eed66
      lidanqing 提交于
      * Fix the lookup_table_v2 failed on GRU mkldnn kernel issue
      test=develop
      
      * fix according to reviews, removed x_num_col_dims
      test=develop
      
      * update gru model. change according to reviews
      test=develop
      
      * change according to reviews
      test=develop
      5c4eed66
  10. 06 8月, 2020 1 次提交
    • A
      Add oneDNN fusion_gru kernel (#25594) · 68c6160e
      Adam 提交于
      * Add oneDNN fusion_gru kernel and fix fc+gru pass
      test=develop
      
      * Formatting changes
      test=develop
      
      * Lint fixes
      test=develop
      
      * Add memory::format_tag::any to GRU weights
      test=develop
      
      * Fix build with CUDA
      
      * Fix build with CUDA v2
      68c6160e