mnist tensorrt incomplete

include/Layer.h

@@ -69,7 +69,6 @@ public:
              const char* fname_weights, bool batchnorm = false); 
     virtual ~LayerWgs();
 
-protected:
     int inputs, outputs;
     std::string weights_path;
 
@@ -137,9 +136,9 @@ public:
 
     virtual value_type* infer(dataDim_t &dim, value_type* srcData);
 
-protected:
     int kernelH, kernelW, strideH, strideW;
 
+protected:
     cudnnFilterDescriptor_t filterDesc;
     cudnnConvolutionDescriptor_t convDesc;
     cudnnConvolutionFwdAlgo_t algo;

include/utils.h

@@ -63,8 +63,8 @@
 }
 
 void readBinaryFile(const char* fname, int size, value_type** data_h, value_type** data_d, int seek = 0);
-int  checkResult(int size, value_type *data_d, value_type *correct_d);
-void printDeviceVector(int size, value_type* vec_d);
+int  checkResult(int size, value_type *data_d, value_type *correct_d, bool device = true);
+void printDeviceVector(int size, value_type* vec_d, bool device = true);
 void resize(int size, value_type **data);
 
 void matrixTranspose(cublasHandle_t handle, value_type* srcData, value_type* dstData, int rows, int cols);