- 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
-
- 28 5月, 2021 1 次提交
-
-
由 chentianyu03 提交于
-
- 23 3月, 2021 1 次提交
-
-
由 Qi Li 提交于
-
- 08 3月, 2021 1 次提交
-
-
由 Qi Li 提交于
-
- 01 3月, 2021 1 次提交
-
-
由 Qi Li 提交于
-
- 17 4月, 2020 1 次提交
-
-
由 Zhong Hui 提交于
* fix error message of l2_normalize, matmul, mean, etc. * add the test case for those ops
-
- 05 4月, 2020 1 次提交
-
-
由 wawltor 提交于
* Add the sum op to API 2.0, test=develop * Fix the import meesage in common_ops_import
-
- 12 11月, 2019 1 次提交
-
-
由 lilong12 提交于
* set the default value of alpha for prelu to 0.25, test=develop * add the call to __syncthreads(), test=develop * fix the implementation of cpu prelu, test=develop * repair the implementation of element mode prelu, test=develop * modify test_prelu_op.py, test=develop
-
- 17 10月, 2019 1 次提交
-
-
由 Zeng Jinle 提交于
* refine reduce code to save compiling time and binary sizes, test=develop * add reduce rank check to avoid bug, test=develop
-
- 05 9月, 2019 1 次提交
-
-
由 Tao Luo 提交于
test=develop
-
- 16 11月, 2018 1 次提交
-
-
由 Wu Yi 提交于
* wip simplify operator framework * wip * wip * done test=develop * clean test=develop * fix test=develop * fix deps test=develop * fix cpu build test=develop * fix tensorrt build test=develop * fix tests test=develop * fix test=develop * fix cpu build test=develop
-
- 09 10月, 2018 1 次提交
-
-
由 qingqing01 提交于
* Fix bug in reduce_op caused by PR #13534 * Fix output shape and enhance unit test. test=develop
-
- 29 9月, 2018 1 次提交
-
-
由 Dun 提交于
* refine reduce by cub * optimize KernelDepthwiseConvFilterGrad * optimize depthwise conv and reduce mean and reduce sum * fix bug: dilation * cuda arch and cuda 8 compatible
-