mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

all test ok

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
tests/backbones/dla34/dla34.cpp
+4 -4
View File
@@ -312,9 +312,9 @@ int main()
printCenteredTitle(" CUDNN inference ", '=', 30);
{
dim1.print();
TIMER_START
TKDNN_TSTART
net.infer(dim1, data);
TIMER_STOP
TKDNN_TSTOP
dim1.print();
}
cudnn_out = net.layers[net.num_layers-1]->dstData;
@@ -326,9 +326,9 @@ int main()
printCenteredTitle(" TENSORRT inference ", '=', 30);
{
dim2.print();
TIMER_START
TKDNN_TSTART
netRT.infer(dim2, data);
TIMER_STOP
TKDNN_TSTOP
dim2.print();
}
rt_out = (dnnType *)netRT.buffersRT[1];
tests/backbones/resnet101/resnet101.cpp
+4 -4
View File
@@ -301,9 +301,9 @@ int main()
printCenteredTitle(" CUDNN inference ", '=', 30);
{
dim1.print();
TIMER_START
TKDNN_TSTART
net.infer(dim1, data);
TIMER_STOP
TKDNN_TSTOP
dim1.print();
}
cudnn_out = net.layers[net.num_layers-1]->dstData;
@@ -314,9 +314,9 @@ int main()
printCenteredTitle(" TENSORRT inference ", '=', 30);
{
dim2.print();
TIMER_START
TKDNN_TSTART
netRT.infer(dim2, data);
TIMER_STOP
TKDNN_TSTOP
dim2.print();
}
rt_out = (dnnType *)netRT.buffersRT[1];