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

Fixing bugs in utils

parent 7183547b
No related branches found
No related tags found
No related merge requests found
...@@ -218,7 +218,7 @@ void* readInputTensor(char* file_name, int data_type, int dim1_size, int dim2_si ...@@ -218,7 +218,7 @@ void* readInputTensor(char* file_name, int data_type, int dim1_size, int dim2_si
struct Tensor* readTrainedWeights(char* file_name, int data_type, struct Tensor* readTrainedWeights(const char* file_name, int data_type,
int dim1_size, int dim2_size, int dim1_size, int dim2_size,
int dim3_size, int dim4_size){ int dim3_size, int dim4_size){
...@@ -254,7 +254,7 @@ struct Tensor* readTrainedWeights(char* file_name, int data_type, ...@@ -254,7 +254,7 @@ struct Tensor* readTrainedWeights(char* file_name, int data_type,
} }
uint8_t* readLabels(char* labels_file, int num_labels){ uint8_t* readLabels(const char* labels_file, int num_labels){
uint8_t* labels = (uint8_t*) malloc(sizeof(uint8_t) * num_labels); uint8_t* labels = (uint8_t*) malloc(sizeof(uint8_t) * num_labels);
FILE* file = fopen(labels_file, "rb"); FILE* file = fopen(labels_file, "rb");
......
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