- 02 7月, 2021 1 次提交
-
-
由 XiangGao 提交于
-
- 15 6月, 2021 1 次提交
-
-
由 jiangcheng 提交于
* add reduce_sum_op by add self-kernel * set all ReduceKernel MPType for accuracy * add float16 test script which input is integer number * solve reduce sum float16 check_grad problem * solve conflict and change test script for CI * change kernel register for CI * remove all useless template
-
- 27 5月, 2021 1 次提交
-
-
由 chentianyu03 提交于
* modify kron OP to complex template types * modify reshape, slice, trace, transpose OPs to complex template types * modify to complex template types in eigen slice files * change to complex template types for pad.cc and pac.cu * format code style
-
- 21 4月, 2021 1 次提交
-
-
由 wuhuanzhou 提交于
* remove thrust includes, test=develop * fix compilation error, test=develop * fix compilation of truncated_gaussian_random_op, test=develop
-
- 08 3月, 2021 1 次提交
-
-
由 Qi Li 提交于
-
- 01 3月, 2021 1 次提交
-
-
由 Qi Li 提交于
-
- 04 12月, 2020 1 次提交
-
-
由 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
-
- 13 7月, 2020 1 次提交
-
-
由 yaoxuefeng 提交于
-
- 27 4月, 2020 1 次提交
-
-
由 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
-