mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
303 Commits 11 Branches 5 Tags
cb5b7a1d98f36dc6f5303479d4e0ea28be7023df
Commit Graph

9 Commits

Author SHA1 Message Date
Davide Sapienza 106cbb5c73 Remove the optional parameters from the Pooling layer. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-09 19:20:27 +02:00
Micaela Verucchi fabad7bace Add mnist download, change ERROR_TKDNN into ERROR_TENSORRT 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-09 19:02:25 +02:00
Micaela Verucchi 3502c5b676 Fix pooling, add return code in each test, add return code handling in test_all_tests script 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-09 15:25:05 +02:00
Davide Sapienza ce20868bad Add TKDNN_MODE variable to the name of the network. 
This commit permits to obtain different .rt files for different
precision optimizations of the same network.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-03-30 15:08:25 +02:00
Francesco Gatti 555b32b5dc Refactoring 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-14 17:22:58 +01:00
Francesco Gatti cf7fbadcd2 Add function to download weigths if do no exist, for each test. Add some controls in map demo. 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
 2020-03-14 17:02:07 +01:00
nvidia 854a4c316a Add ResizeLayerRT plugin 
Signed-off-by: nvidia <micaelaverucchi@gmail.com>
 2019-12-20 15:27:41 +01:00
Micaela Verucchi 44b2bce3ff Yolo3_tiny CUDNN works, TensorRT doesn't. Add n_masks to Yolo layer. 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2019-12-05 20:43:48 +00:00
Micaela Verucchi a2db98670a Add Yolov3 (COCO80) and Yolov3-tiny (COCO80), TensorRT for tiny not working 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2019-12-04 17:01:40 +00:00