- 27 11月, 2020 4 次提交
-
-
由 lilong12 提交于
update expand as op to use the shape of the target tensor instead of the target tensor itself. (#29020) * update, test=develop
-
由 Jack Zhou 提交于
Add eigen gru and fix the dropout bug in the rnn
-
由 arlesniak 提交于
-
由 Shang Zhizhou 提交于
* remove -DSUPPORTS_CUDA_FP16 in cuda.cmake * comile with cuda9 * add some unittest * notest;test=coverage * add unittest for trt plugin swish && split * update ernie unittest * fix some error message * remove repeated judgement of CUDA version in mbEltwiseLayerNormOpConverter * fix comile errror when CUDA_ARCH_NAME < Pascal" * fix comile error * update unittest timeout * compile with cuda9 * update error msg * fix code style * add some comments * add define IF_CUDA_ARCH_SUPPORT_FP16 * rename IF_CUDA_ARCH_SUPPORT_FP16 to CUDA_ARCH_FP16_SUPPORTED
-
- 26 11月, 2020 2 次提交
-
-
由 Noel 提交于
Fix ops doc for some ops
-
由 joanna.wozna.intel 提交于
* Add bf16 pool2d and unify bf16 unit tests * Add change default ops test
-
- 25 11月, 2020 4 次提交
-
-
由 joejiong 提交于
add uint8 for reshape operator
-
由 taixiurong 提交于
-
由 joejiong 提交于
Simple code clean up
-
由 wawltor 提交于
remove eigen threadpool for the speed up
-
- 24 11月, 2020 2 次提交
- 23 11月, 2020 3 次提交
-
-
由 furnace 提交于
* refactor momentum op to combine weight_decay (scale op and sum op)
-
由 Jacek Czaja 提交于
-
由 yaoxuefeng 提交于
-
- 20 11月, 2020 10 次提交
-
-
由 gongweibao 提交于
-
由 Chen Weihang 提交于
-
由 joejiong 提交于
Adding uint8 support for squeeze operator.
-
由 wangchaochaohu 提交于
-
由 joanna.wozna.intel 提交于
* Add bf16 matmul, fc, elementwise add and mul * Correct unit test
-
由 yaoxuefeng 提交于
-
由 taixiurong 提交于
* 1.add xpu slice op 2. add xpu top_k op 3.modify xpu cast to new api * 1.add xpu slice op 2. add xpu top_k op 3.modify xpu cast to new api
-
由 Jack Zhou 提交于
* add lstm, simple rnn op kernel * fix the test_lstm for the rnn op * change func name * fix forward postprocess bug * add gru forward, backward code * remove unittest.skipIf; use a big rnn op instead of combination op * fix input doesn't have gradient bug * add eigen lstm forward, backward Co-authored-by: Nwawltor <fangzeyang0904@hotmail.com>
-
由 QingshuChen 提交于
* adjust kunlun header file *test=kunlun * update kunlun unittest *test=kunlun * update xpu unitest * test = kunlun * update xpu unittest * test=kunlun * update xpu unitest * test=kunlun
-
由 Zhang Ting 提交于
-
- 19 11月, 2020 3 次提交
-
-
由 yaoxuefeng 提交于
-
由 Wojciech Uss 提交于
* Add multi_gru op and tests * removed redundant disable_dygraph()
-
由 joejiong 提交于
Add new operator log10
-
- 18 11月, 2020 1 次提交
-
-
由 Guo Sheng 提交于
test=develop
-
- 17 11月, 2020 4 次提交
-
-
由 Jacek Czaja 提交于
-
由 Zhou Wei 提交于
* add unittest and check unittest for windows * fix lstm OP compile error on windows
-
由 石晓伟 提交于
Co-authored-by: Nyaoxuefeng6 <yaoxuefeng@baidu.com>
-
由 cc 提交于
-
- 16 11月, 2020 7 次提交
-
-
由 lilong12 提交于
-
由 wawltor 提交于
fix the gradient bug for the topk v2
-
由 wangchaochaohu 提交于
-
由 joanna.wozna.intel 提交于
* Add checkpoint to quantize * Change bfloat16 option
-
由 pangyoki 提交于
-
由 Guo Sheng 提交于
* Fix gradients with ignore_idx in softmax_with_cross_entropy. test=develop * Fix gradients with ignore_idx in softmax_with_cross_entropy on cpu. Remove softmax_with_cross_entropy from op_threshold_white_list. test=develop * Fix test_softmax_cross_entropy_op.py. test=develop
-
由 Leo Chen 提交于
-