- 17 8月, 2020 1 次提交
-
-
由 Jack Zhou 提交于
* add nll loss API, update demo code of the comment
-
- 16 8月, 2020 1 次提交
-
-
由 lilong12 提交于
* adaptive expand op to 2.0 (align to torch.expand) , test=develop
-
- 15 8月, 2020 1 次提交
-
-
由 lilong12 提交于
* add tile_op, test=develop
-
- 14 8月, 2020 5 次提交
-
-
由 Yang Zhang 提交于
test=document_fix remove activation wording
-
由 Yang Zhang 提交于
test=develop,test=document_fix remove activation wording
-
由 Yang Zhang 提交于
test=develop,test=document_fix remove activation wording
-
由 Yang Zhang 提交于
test=develop,test=document_fix remove activation wording tanh -> tan
-
由 Yang Zhang 提交于
test=develop,test=document_fix explain input/out put range and out of boundary behavior
-
- 13 8月, 2020 3 次提交
-
-
由 wawltor 提交于
Update the code for the min,max, test=develop
-
由 Leo Chen 提交于
* add unchaged infershape function * add broadcast infershape function * fix bug * rename infershape functions * add UnaryOpUnchangedInferShapeCheckAxis * add error message * add test for common infer shape functions * dont update existed ops * dont update op_desc.h * add more test * add error check, refine error message
-
由 Leo Chen 提交于
* add auto_cast, test=develop * add loss scaler, test=develop * add comments, test=develop * refine code, test=develop * refine code, test=develop * do not set flags automatically, test=develop * fix custom op bug, test=develop * add more test, test=develop * refine enable logic, test=develop * enable amp test with GPU, test=develop * add unittest * add test for found_inf * follow comments * follow comments * remove global variable, use singleton * add some notes * update comments * update comments * update comments * add use_dynamic_loss_scaling argument * refine found_inf * refine found_inf
-
- 12 8月, 2020 2 次提交
- 10 8月, 2020 6 次提交
-
-
由 lilong12 提交于
* add the support for multi-node training
-
由 LutaoChu 提交于
update cumsum api and fix up the cumsum op
-
由 Adam 提交于
* Add oneDNN relu6 op * Lint fixes
-
由 Zhaolong Xing 提交于
test=develop
-
由 Chen Weihang 提交于
* add pin memory control * fix buffered reader init problem * fix unittest error * add unittest for coverage
-
由 Chen Weihang 提交于
* add cuda pinned place branch * add unittest * add skip when not gpu
-
- 08 8月, 2020 3 次提交
-
-
由 Feiyu Chan 提交于
1. update document template for unary elementwise layers(a.k.a. activation layer); 2. remove generate_op_noattr and use generate_activation instead; remove redundant function copies; 3. minor update for docstring to fix rst format errors. 4. fix doc for Rsqrt OP 5. add sample code for each activation separately; 6. remove the unused deprecated decorator.
-
由 joanna.wozna.intel 提交于
* Change use_quantizer attribute name and data type * Fix problem with setting attribute * Add changes due to review * Small change in function * Restore use_quantizer attr for compatibility
-
由 Leo Chen 提交于
* refine unsqueeze, test=develop * update unsqueeze, test=develop * refine unsqueeze, test=develop * refine unsqueeze, test=develop * update * remove None, test=develop * follow comments * support bool * update doc * follow comments * merge develop
-
- 06 8月, 2020 2 次提交
-
-
由 yaoxuefeng 提交于
-
由 Adam 提交于
* Add oneDNN fusion_gru kernel and fix fc+gru pass test=develop * Formatting changes test=develop * Lint fixes test=develop * Add memory::format_tag::any to GRU weights test=develop * Fix build with CUDA * Fix build with CUDA v2
-
- 05 8月, 2020 5 次提交
-
-
由 Zhong Hui 提交于
fix invalid read of pnorm gradient function and delete the unused code
-
由 Zhaolong Xing 提交于
* cunn8 support test=develop * fix ci error test=develop
-
由 Zhaolong Xing 提交于
test=develop
-
由 Pei Yang 提交于
* develop dynamic shape serilization * add test param for gelu * fix bugs * delete redundant comments * debug * fix conflict. test=develop * fix bug. test=develop * add trt dynamic shape serialized support * fix ernie serialized bug test=develop * fix codestyle test=develop * fix bug test=develop * fix bug.test=develop * modify cmakelist test=develop * fix bug test=develop * fix error message. test=develop * fix trt register plugin based on pr#25003 * add trt dynload * fix deserialization bug of not finding plugin registration * refine code style * recover engine key in tensorrt_subgraph_pass * for ci coverage * add unittest for deserialization Co-authored-by: Nhaozech <chenhaoze94@gmail.com>
-
由 wawltor 提交于
Update the code for the compare ops for the broadcast function
-
- 04 8月, 2020 1 次提交
-
-
由 wangchaochaohu 提交于
-
- 03 8月, 2020 3 次提交
-
-
由 tangwei12 提交于
* disable heart beat UT
-
由 xujiaqi01 提交于
* fix dump, fix cvm check test=develop * fix test=develop * fix test=develop * fix test=develop
-
由 Zhang Ting 提交于
-
- 31 7月, 2020 3 次提交
-
-
由 Zhang Ting 提交于
-
由 ShenLiang 提交于
* fix inverse bug, test=develop * fix the untest, test=develop * add singular checking, test=develop * fix the utest, test=develop * use memory::copy, test=develop * fix bost_get, test=develop * fix position, test=develop
-
由 Aurelius84 提交于
* modify GetExpectedKernelType * use input place * add ENFORCE check
-
- 30 7月, 2020 4 次提交
-
-
由 wawltor 提交于
Update the code for the compare_ops, update the api and doc
-
由 wangchaochaohu 提交于
-
由 lilong12 提交于
* bug fix, test=develop
-
由 tangwei12 提交于
-