- 24 3月, 2021 1 次提交
-
-
由 Shang Zhizhou 提交于
* fix conflict * fix compile error * cherry-pick #31316 * Refine cudnn softmax (#25757) * refine cudnn softmax * Trt elementwise plugin serialize (#31587) * add serialize unittest * fix element_op trt plugin serialize bug * remove PassVersionChecker.IsCompatible * fix unittest Co-authored-by: NPei Yang <peiyang@baidu.com> Co-authored-by: NGaoWei8 <53294385+GaoWei8@users.noreply.github.com>
-
- 12 4月, 2020 1 次提交
-
-
由 Zhaolong Xing 提交于
* add elementwise pool2d, prelu, shuffle channel test=develop * add scale and refine concat eltwise conveter test=develop * refine elementwise converter test=develop * refine ut test and enforce error. test=develop * modify const cast test=develop
-
- 08 3月, 2019 1 次提交
-
-
由 nhzlx 提交于
1). add static trt load model 2). fix bug: when device_id is not 0, the trt will have a bug test=develop
-
- 22 2月, 2019 1 次提交
-
-
由 nhzlx 提交于
1). add static trt load model 2). fix bug: when device_id is not 0, the trt will have a bug test=develop
-
- 20 11月, 2018 1 次提交
-
-
由 Yiqun Liu 提交于
* Initialize the elementwise plugin. * Implement the basic CUDA kernel of elementwise plugin. test=develop
-