mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
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.
Files
002355e7a7e96724efde33f0b45ec326b32b08a7
tkDNN/src
T
History
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
..
kernels
Fix memory leak
2020-02-06 23:09:12 +01:00
Activation.cpp
Add CenterNet based on Resnet101, TensorRT not implemented.
2019-12-18 18:59:54 +01:00
CenternetDetection.cpp
Change CenterNet input dimension.
2020-02-06 18:08:06 +01:00
Conv2d.cpp
Add grouped convolutions in CUDNN and tensorRT.
2020-02-05 14:32:58 +01:00
DeformConv2d.cpp
Fix memory leak
2020-02-06 23:09:12 +01:00
Dense.cpp
string input and flir test
2019-09-14 19:03:13 +02:00
Flatten.cpp
namespace change
2018-12-14 21:55:16 +01:00
Layer.cpp
CenterNet TensorRT works. TensorRT serialization not yet implemented
2019-12-20 11:05:03 +01:00
LayerWgs.cpp
Add grouped convolutions in CUDNN and tensorRT.
2020-02-05 14:32:58 +01:00
MulAdd.cpp
namespace change
2018-12-14 21:55:16 +01:00
Network.cpp
Merge with master, works with Jetpack 4.3
2020-01-15 19:24:19 +01:00
NetworkRT.cpp
Fix memory leak
2020-02-06 23:09:12 +01:00
Pooling.cpp
Centernet: fix pooling problem, add centrnet demo
2020-01-21 12:50:18 +01:00
Region.cpp
string input and flir test
2019-09-14 19:03:13 +02:00
Reorg.cpp
yolo3 ok
2018-12-21 15:28:47 +01:00
Route.cpp
ResNet working
2019-10-29 17:21:31 +01:00
Shortcut.cpp
yolo3 86 route error
2018-12-23 16:20:17 +01:00
Softmax.cpp
namespace change
2018-12-14 21:55:16 +01:00
sorting.cu
Centernet: fix pooling problem, add centrnet demo
2020-01-21 12:50:18 +01:00
Upsample.cpp
upsample ok
2018-12-20 17:08:31 +01:00
utils.cpp
Add CenterNet based on Resnet101, TensorRT not implemented.
2019-12-18 18:59:54 +01:00
Yolo3Detection.cpp
Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet
2020-01-20 16:41:19 +01:00
Yolo.cpp
Yolov3_tiny works on tensorRT
2019-12-23 16:11:41 +01:00