1. 08 12月, 2022 1 次提交
  2. 16 11月, 2022 1 次提交
  3. 30 9月, 2022 1 次提交
    • S
      support pure bfloat16 for more ops (#46364) · b7b231a6
      sneaxiy 提交于
      * support pure bfloat16
      
      * support bf16 linear
      
      * update PR to pass CI
      
      * tiny fix where_grad_kernel.cu
      
      * add bfloat16 to selu_grad to pass CI
      
      * fix selu grad compilation error
      b7b231a6
  4. 21 6月, 2022 1 次提交
  5. 05 6月, 2022 1 次提交
  6. 31 3月, 2022 1 次提交
  7. 23 3月, 2022 1 次提交
    • K
      Add complex type compatibility for stft api and stft op. (#40113) · 319f95d0
      KP 提交于
      * Add stft_op.
      
      * Add stft_grad_op.
      
      * Add stft_op unittest.
      
      * [DLTP-45176] Add complex compatibility in static mode for stft api.
      
      * [DLTP-45176] Add complex compatibility in static mode for stft api.
      
      * Add doc.
      
      * Update unitests of stft op.
      
      * Update spectral helper.
      
      * fix coding style.
      319f95d0
  8. 21 3月, 2022 1 次提交