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

Removing ugly print from tensor utils

parent ed00fc37
No related branches found
No related tags found
No related merge requests found
......@@ -43,13 +43,13 @@ void dumpWeightsToFile(char *file_name, void *weights_ptr) {
abort();
}
// printf("size_in_bytes = %lu \n", weights->size_in_bytes);
size_t bytes_written =
fwrite(weights->host_data, 1, weights->size_in_bytes, fp);
// printf("bytes_written = %lu \n", bytes_written);
fclose(fp);
}
void fillTensorWithOnes(void *tensor_ptr) {
struct Tensor *tensor = (struct Tensor *)tensor_ptr;
......@@ -201,7 +201,7 @@ struct Tensor *readTrainedWeights(const char *file_name, int data_type,
long int size_in_bytes =
type_size * dim1_size * dim2_size * dim3_size * dim4_size;
float *tensor_data = (float *)malloc(sizeof(float) * num_elems);
printf("size_in_bytes = %lu \n", size_in_bytes);
//printf("size_in_bytes = %lu \n", size_in_bytes);
int file_header_size = 0;
......@@ -214,9 +214,6 @@ struct Tensor *readTrainedWeights(const char *file_name, int data_type,
fseek(file, file_header_size, SEEK_CUR); // Skipping the file header
size_t bytes_read = fread(tensor_data, 1, size_in_bytes, file);
// printf("size in bytes = %lu, bytes read = %lu \n", size_in_bytes,
// bytes_read);
fclose(file);
struct Tensor *weights = (struct Tensor *)create4DTensor(
......@@ -254,7 +251,6 @@ struct Tensor *readInputBatch(const char *file_name, long data_type,
fclose(file);
// printf ("FIXED input BATCH read \n");
struct Tensor *weights = (struct Tensor *)create4DTensor(
data_type, nchw, dim1_size, dim2_size, dim3_size, dim4_size);
......
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