diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h b/llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h index 56d3ecf6bd7ee3fcf34533da57309a835652ce27..86754b82d7a02441299f301c8da73eba8239da55 100644 --- a/llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h +++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h @@ -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 dim3_size, int dim4_size){ @@ -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); FILE* file = fopen(labels_file, "rb");