1. 14 4月, 2023 1 次提交
    • Y
      [Zero-Dim] support 0-D tensor for... · 6f41e177
      YangQun 提交于
      [Zero-Dim] support 0-D tensor for reduce/reshape/stack/prelu/expand_v2/gaussion onednn kernels (#52185)
      
      * support 0-D tensor for reduce/reshape/stack/prelu/expand_v2/gaussion ops
      
      * fix gaussian random mkldnn op ut
      6f41e177
  2. 31 3月, 2023 2 次提交
  3. 29 3月, 2023 1 次提交
  4. 28 3月, 2023 1 次提交
  5. 25 3月, 2023 1 次提交
  6. 09 3月, 2023 1 次提交
  7. 12 12月, 2022 1 次提交
    • J
      reshape onednn test reimplemented (#48850) · 41d27818
      Jacek Czaja 提交于
      * - UT reshape onednn
      
      - Fix
      
      test
      
      test2
      
      - test4
      
      - test5
      
      - test6
      
      test7
      
      - test8
      
      - Ut reinvented
      
      - cosmetic
      
      * - fix
      
      * - fix
      
      * - fix
      
      * - fix
      
      * - Fix
      
      * - fix
      
      * - fix
      
      * - fix
      
      * - Fix
      
      * lint
      41d27818
  8. 25 11月, 2022 1 次提交
  9. 23 10月, 2022 1 次提交
  10. 27 9月, 2022 1 次提交
  11. 05 6月, 2022 1 次提交
    • S
      【code format check upgrade】 step2:yapf (#42944) · a072fca8
      Sing_chan 提交于
      * use yapf to format all python file
      
      * yapf exclude two unittests file for they rely on writing and reading file, and format will break them
      
      * disable diff_py_file because too many diff files cause command following failed
      a072fca8
  12. 07 9月, 2021 1 次提交
  13. 16 8月, 2021 1 次提交
  14. 30 7月, 2021 1 次提交
    • J
      Added reshape, reshape2, squeeze and squeeze2 BF16/FP32 FWD/BWD kernels (#34219) · 22c4c189
      jakpiase 提交于
      * test version of matmul_v2
      
      * added matmul_v2 grad kernel
      
      * minor changes
      
      * minor changes
      
      * minor change for CI approval
      
      * CI fix
      
      * CI fix
      
      * added squeeze and squeeze2 kernels
      
      * CI fix
      
      * CI fix
      
      * CI fix
      
      * disabled tests when compiled with cuda
      
      * added setting format_tag by strides
      
      * added sigmoid BF16 FWD/BWD and gelu BF16 BWD
      
      * changes after review
      
      * Revert "added sigmoid BF16 FWD/BWD and gelu BF16 BWD"
      
      This reverts commit 6e3f76720b545abfcff9f6052b46b73a1e745cae.
      
      * Revert "Merge branch 'matmul_v2_grad' into squeeze2_op"
      
      This reverts commit 06fcf67843a4a7884eccdf67a02a03575e1d4cb8, reversing
      changes made to 6e3f76720b545abfcff9f6052b46b73a1e745cae.
      
      * minor change
      
      * added reshape1/2 kernels
      
      * moved some functions into private block
      
      * CI fix
      
      * CI fix
      
      * CI fix
      22c4c189