This repository has been archived on 2026-02-22. You can view files and clone it. You cannot open issues or pull requests or push a commit.

Davide Sapienza 5abf1613f2 Fix memory leak ...

This commit moves cublasCreate out from dcn_v2_cuda_forward
to save some milliseconds and it adds cublasDestroy (cause
of memory leak).

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>

2020-02-06 23:09:12 +01:00

..

activation_elu.cu

stream in TRT plugin

2017-08-10 19:21:15 +02:00

activation_leaky.cu

stream in TRT plugin

2017-08-10 19:21:15 +02:00

activation_logistic.cu

stream in TRT plugin

2017-08-10 19:21:15 +02:00

activation_sigmoid.cu

Centernet: fix pooling problem, add centrnet demo

2020-01-21 12:50:18 +01:00

convert.cu

fp16 implementation, TODO deallocate in LayerWgs

2017-08-30 14:37:25 +00:00

deformable_conv.cu

Fix memory leak

2020-02-06 23:09:12 +01:00

fill.cu

upsample ok

2018-12-20 17:08:31 +01:00

reorg.cu

stream in TRT plugin

2017-08-10 19:21:15 +02:00

resize.cu

Yolov3_tiny works on tensorRT

2019-12-23 16:11:41 +01:00

shortcut.cu

shortcut cu

2018-12-20 09:53:17 +01:00

softmax.cu

stream in TRT plugin

2017-08-10 19:21:15 +02:00

upsample.cu

upsample ok

2018-12-20 17:08:31 +01:00