opencv viz

tests/mnist/test_mnist.cpp

@@ -25,11 +25,11 @@ int main() {
     tkDNN::NetworkRT netRT(&net);
 
     // Load input
-    value_type *data;
-    value_type *input_h;
+    dnnType *data;
+    dnnType *input_h;
     readBinaryFile(input_bin, dim.tot(), &input_h, &data);
 
-    value_type *out_data, *out_data2;
+    dnnType *out_data, *out_data2;
 
     std::cout<<"CUDNN inference:\n"; {
         dim.print(); //print initial dimension  
@@ -63,8 +63,8 @@ int main() {
  /*
     // Print real test
     std::cout<<"\n==== CHECK RESULT ====\n";
-    value_type *out;
-    value_type *out_h;
+    dnnType *out;
+    dnnType *out_h;
     readBinaryFile(output_bin, dim.tot(), &out_h, &out);
     printDeviceVector(dim.tot(), out);
 */