1. 27 5月, 2021 1 次提交
  2. 21 4月, 2021 1 次提交
    • W
      remove thrust include files (#32395) · ab6f8745
      wuhuanzhou 提交于
      * remove thrust includes, test=develop
      
      * fix compilation error, test=develop
      
      * fix compilation of truncated_gaussian_random_op, test=develop
      ab6f8745
  3. 08 3月, 2021 1 次提交
  4. 01 3月, 2021 1 次提交
  5. 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
  6. 13 7月, 2020 1 次提交
  7. 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