- 15 9月, 2020 11 次提交
-
-
-
由 Shang Zhizhou 提交于
* optimize slice TRT plugin This patch removes unnecessary barrier for data transfer of needed offset, so data transfer can be overlap with GPU kernel execution. This patch also fixes incorrect name of slice plugin. That is, replaces "layernorm" with "slice" test=develop * add serialize/deserialize to slice plugin * add static shape slice trt plugin * fix slice trt op convertor dynamic shape bug * fix format by clang-format * fix pylint format error * fix problems commented by peiyang Co-authored-by: NRyan Jeng <rjeng@nvidia.com>
-
由 chalsliu 提交于
-
由 Wilber 提交于
-
由 石晓伟 提交于
-
由 LoveAn 提交于
Add *.bat file for building compile environment on windows
-
由 Chen Weihang 提交于
* polish framework error msg part 6 * polish lossed item * fix failed unittest * polish by reviewer comments
-
由 Shang Zhizhou 提交于
* optimize errror report * add test case for pad op converter * fix some spelling mistake commented by peiyang
-
由 GaoWei8 提交于
* replace sequence length attr to input
-
由 cc 提交于
* Remove the cache in post_traning_quantization, test=develop
-
由 Pei Yang 提交于
* fix trt_dynamic_shape_ernie_deserialize_test * support when opt cache dir does not exist
-
- 14 9月, 2020 17 次提交
-
-
由 joanna.wozna.intel 提交于
-
由 YUNSHEN XIE 提交于
-
由 zhupengyang 提交于
-
由 lilong12 提交于
* improve err report, test=develop
-
由 lilong12 提交于
* bug fix, test=develop
-
由 LielinJiang 提交于
Fix conv deepwise bug when in_channels=1.
-
由 Zhong Hui 提交于
Enhance the error messages for files in operators/math
-
由 xiaoting 提交于
-
由 MRXLT 提交于
* add check for sparse parameters with weight_decay * move sparse check to adam.py
-
由 Chen Weihang 提交于
-
由 Chen Weihang 提交于
* move worker loop to top level * move reader process loop to top level * fix failed unittests
-
由 Pei Yang 提交于
-
由 Zhen Wang 提交于
Update amp_check_finite_and_scale_op and add an updating_loss_scaling op for static graph amp training. (#26240) * update amp_check_finite_and_scale_op for static_amp. * use amp_check_finite_and_scale in static graph amp. * update grads to zero when grads own infinite values(as for amp_checkout_finite_and_scale op). * add update_loss_scaling op in cpp. * add update_loss_scaling_op unit test. * update the doc of the check_finite_and_unscale op * Update the process of gradients updating skipping if the gradients have infinite values. * update the way to zero grads. * update test_update_loss_scaling_op.py * add log info when find infinite grads. * add the unit test for UpdateLossScaling Layer.
-
由 ShenLiang 提交于
* rm auto from localsgd
-
由 Adam 提交于
* Add int8 GRU kernel with UTs * Lint fixes * More lint fixes
-
由 石晓伟 提交于
-
由 Jack Zhou 提交于
Error description optimize for math dir
-
- 13 9月, 2020 1 次提交
-
-
由 Zhang Ting 提交于
-
- 12 9月, 2020 2 次提交
-
-
由 lidanqing 提交于
* Fix the lookup_table_v2 failed on GRU mkldnn kernel issue test=develop * fix according to reviews, removed x_num_col_dims test=develop * update gru model. change according to reviews test=develop * change according to reviews test=develop
-
由 LoveAn 提交于
* Add details to the summary and test it, test=document_fix * Add set +e before example, test=document_fix * Remove test code, test=document_fix * Optimize summary information and test it, test=document_fix * Remove test code, test=document_fix
-
- 11 9月, 2020 9 次提交
-
-
由 Leo Chen 提交于
* temporally disable zero_copy * add test * follow comments
-
由 Leo Chen 提交于
-
由 Aurelius84 提交于
* fix calcu_gradients * fix code place * fix embedding interface usage
-
由 Chen Weihang 提交于
-
由 Wilber 提交于
-
由 liym27 提交于
-
由 Chen Weihang 提交于
-
由 Aurelius84 提交于
* support to_static(model) * add warning and unittest
-
由 Wilber 提交于
-