1. 05 6月, 2022 1 次提交
  2. 11 3月, 2022 1 次提交
  3. 06 3月, 2022 1 次提交
  4. 20 2月, 2022 1 次提交
  5. 17 2月, 2022 1 次提交
  6. 09 2月, 2022 1 次提交
    • H
      Move trace op to pten (#39227) · d7dddf94
      hong 提交于
      * add trace op
      
      * bug fix
      
      * bug fix; test=develop
      
      * thrust bug fix; test=develop
      
      * remove useless register; test=develop
      
      * fix bug; test=develop
      
      * update trace kernel; test=develop
      
      * move kernel args to trace_sig; test=develop
      d7dddf94
  7. 27 5月, 2021 1 次提交
  8. 06 1月, 2021 1 次提交
  9. 05 1月, 2021 1 次提交
  10. 04 12月, 2020 1 次提交
    • C
      Make transpose, trace, kron, reshape, sum op support complex type (#29321) · 879e913b
      chentianyu03 提交于
      * add complex64 and complex128 type; add +-*/@ and slice opreator for complex types
      
      * add test cases for complex elementwise, matmul and getitem unittest
      
      * add test cases for complex types
      
      * add test cases for complex matmul unittest
      
      * kron, reshape, transpose support complex types
      
      * sum and trace op support complex types
      
      * add test case of sum and trace op
      
      * fix the bug of imag part of complex not initialized
      
      * format file
      
      * format code style
      
      * kron support type promotion; modify test cases
      879e913b
  11. 13 7月, 2020 1 次提交
  12. 27 5月, 2020 1 次提交
  13. 27 4月, 2020 1 次提交
    • L
      Add trace op (#23873) · 077e5a0f
      Li Fuchen 提交于
      * add trace op, test=develop
      
      * Optimized the implementation of trace op, test=develop
      
      * fix a bug of include in trace_op.h, test=develop
      
      * move trace API from creation to math, test=develop
      
      * modified en doc. test=develop
      
      * add complex trace api
      
      * add complex sum api, test=develop
      
      * modified en doc of complex sum and trace, test=develop
      
      * modified doc and trace API, test=develop
      
      * modified en doc of trace and sum, test=develop
      
      * modified comment in complex kron API, test=develop
      
      * OP Should Not Have Unused Input, test=develop
      
      * add GetExpectedKernelType, test=develop
      077e5a0f