Skip to content
Snippets Groups Projects
Commit a3dbbf8d authored by Hashim Sharif's avatar Hashim Sharif
Browse files

fixing tensorutils definitions to return void*

parent ec7df083
No related branches found
No related tags found
No related merge requests found
...@@ -15,9 +15,9 @@ std::string model_params_path = "../../test/dnn_benchmarks/model_params/"; ...@@ -15,9 +15,9 @@ std::string model_params_path = "../../test/dnn_benchmarks/model_params/";
__attribute__((noinline)) struct Tensor *readTrainedWeights(const char *file_name, int data_type, __attribute__((noinline)) void *readTrainedWeights(const char *file_name, int data_type,
long int dim1_size, long int dim2_size, long int dim1_size, long int dim2_size,
long int dim3_size, long int dim4_size) { long int dim3_size, long int dim4_size) {
int type_size = 4; // NOTE: Assuming floating point tensors int type_size = 4; // NOTE: Assuming floating point tensors
long int num_elems = dim1_size * dim2_size * dim3_size * dim4_size; long int num_elems = dim1_size * dim2_size * dim3_size * dim4_size;
...@@ -45,9 +45,9 @@ __attribute__((noinline)) struct Tensor *readTrainedWeights(const char *file_nam ...@@ -45,9 +45,9 @@ __attribute__((noinline)) struct Tensor *readTrainedWeights(const char *file_nam
} }
__attribute__((noinline)) struct Tensor *readInputBatch(const char *file_name, long data_type, __attribute__((noinline)) void * readInputBatch(const char *file_name, long data_type,
long start, long end, long start, long end,
long dim2_size, long dim3_size, long dim4_size) { long dim2_size, long dim3_size, long dim4_size) {
long int dim1_size = end - start; long int dim1_size = end - start;
// FIXIT: Don't assume floating point types // FIXIT: Don't assume floating point types
long int type_size = 4; // NOTE: Assuming floating point tensors long int type_size = 4; // NOTE: Assuming floating point tensors
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment