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
c36befaf2bac420c125f6354bb24f0c84dfb41b8
tkDNN/include/tkDNN
T
History
Davide Sapienza cf3fbeddbd Fix the sorting kernels used in the CenterNet pre and post-processing. 
This commit moves the kernels in the correct sub-directory.
It creates new header file for Thrust kernels. It splits the
kernels into two files: 'normalize.cu' contains CenterNet
pre-processing operations, 'postprocessing.cu' contains the
CenterNet post-processing operations.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-04-07 18:28:12 +02:00
..
models
include dir fix, cmake dir
2019-09-17 15:22:39 +02:00
pluginsRT
csresnext50-panet-spp works with TensorRT
2020-03-12 18:24:41 +01:00
CenternetDetection.h
Fix the sorting kernels used in the CenterNet pre and post-processing.
2020-04-07 18:28:12 +02:00
DetectionNN.h
Fix compile errors dut to Int8BatchStream
2020-04-06 19:11:39 +02:00
evaluation.h
Add again evalutation.h
2020-04-06 18:35:52 +02:00
Int8BatchStream.h
Fix compile errors dut to Int8BatchStream
2020-04-06 19:11:39 +02:00
Int8Calibrator.h
Add the Int8 calibrator and the tensorRT Int8 inference
2020-03-27 00:48:14 +01:00
kernels.h
Add csresnext50-panet-spp test. Works on CUDNN. Does not work with TensorRT
2020-03-11 18:37:26 +01:00
kernelsThrust.h
Fix the sorting kernels used in the CenterNet pre and post-processing.
2020-04-07 18:28:12 +02:00
Layer.h
Modify map demo, using abstract class. Move draw function in abstract class
2020-03-23 19:32:37 +01:00
MobilenetDetection.h
Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors.
2020-04-01 19:18:43 +02:00
Network.h
Move the calibration table saving to the tests/network folder.
2020-04-06 18:06:21 +02:00
NetworkRT.h
Fix the tensorRT code to support versions prior to 6.0
2020-04-06 19:54:40 +02:00
tkdnn.h
add DLA, plugin for shortcut and leaky. new verison 0.4
2020-01-15 21:48:18 +01:00
utils.h
Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors.
2020-04-01 19:18:43 +02:00
Yolo3Detection.h
Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors.
2020-04-01 19:18:43 +02:00