all test ok

This commit is contained in:
Francesco Gatti
2020-06-01 16:15:29 +02:00
parent d4e0d07e09
commit c8ed6d782a
16 changed files with 63 additions and 63 deletions
+4 -4
View File
@@ -35,9 +35,9 @@ int main() {
std::cout<<"CUDNN inference:\n"; {
dim.print(); //print initial dimension
TIMER_START
TKDNN_TSTART
out_data = net.infer(dim, data);
TIMER_STOP
TKDNN_TSTOP
dim.print();
}
@@ -49,9 +49,9 @@ int main() {
std::cout<<"TENSORRT inference:\n"; {
dim2.print();
TIMER_START
TKDNN_TSTART
out_data2 = netRT.infer(dim2, data);
TIMER_STOP
TKDNN_TSTOP
dim2.print();
}