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/";
__attribute__((noinline)) struct Tensor *readTrainedWeights(const char *file_name, int data_type,
long int dim1_size, long int dim2_size,
long int dim3_size, long int dim4_size) {
__attribute__((noinline)) void *readTrainedWeights(const char *file_name, int data_type,
long int dim1_size, long int dim2_size,
long int dim3_size, long int dim4_size) {
int type_size = 4; // NOTE: Assuming floating point tensors
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
}
__attribute__((noinline)) struct Tensor *readInputBatch(const char *file_name, long data_type,
long start, long end,
long dim2_size, long dim3_size, long dim4_size) {
__attribute__((noinline)) void * readInputBatch(const char *file_name, long data_type,
long start, long end,
long dim2_size, long dim3_size, long dim4_size) {
long int dim1_size = end - start;
// FIXIT: Don't assume floating point types
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