opencv viz

tests/mnist/test_mnistRT.cpp

@@ -39,8 +39,8 @@ int main() {
     tkDNN::Softmax    l7(&net);
  
     // Load input
-    value_type *data;
-    value_type *input_h;
+    dnnType *data;
+    dnnType *input_h;
     readBinaryFile(input_bin, dim.tot(), &input_h, &data);
 
     dim.print(); //print initial dimension
@@ -55,8 +55,8 @@ int main() {
 
     // Print real test
     std::cout<<"\n==== CHECK CUDNN RESULT ====\n";
-    value_type *out;
-    value_type *out_h;
+    dnnType *out;
+    dnnType *out_h;
     readBinaryFile(output_bin, dim.tot(), &out_h, &out);
     std::cout<<"Diff: "<<checkResult(dim.tot(), out, data)<<"\n";