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

Minor change to include definition signatures

parent 1f6ab12d
No related branches found
No related tags found
No related merge requests found
...@@ -5,13 +5,8 @@ ...@@ -5,13 +5,8 @@
#include <sstream> #include <sstream>
//#include <tensorTypes.h>
#include <tensor_runtime.h> #include <tensor_runtime.h>
#include <tensor.h> #include <tensor.h>
//#include "../../../../../projects/hpvm-tensor-rt/tensor_runtime/include/tensor.h"
//#include "../../../projects/hpvm-tensor-rt/tensor_runtime/include/tensor.h"
//#include "types.h"
void printTensorInfo(void* tensor_ptr){ void printTensorInfo(void* tensor_ptr){
...@@ -183,7 +178,7 @@ void compareValues(void* tensor_ptr, float* data, size_t num_elems){ ...@@ -183,7 +178,7 @@ void compareValues(void* tensor_ptr, float* data, size_t num_elems){
} }
void* readInputTensor(char* file_name, int data_type, int dim1_size, int dim2_size, void* readInputTensor(const char* file_name, int data_type, int dim1_size, int dim2_size,
int dim3_size, int dim4_size){ int dim3_size, int dim4_size){
int type_size = 4; // NOTE: Assuming floating point tensors int type_size = 4; // NOTE: Assuming floating point tensors
...@@ -221,7 +216,7 @@ void* readInputTensor(char* file_name, int data_type, int dim1_size, int dim2_si ...@@ -221,7 +216,7 @@ void* readInputTensor(char* file_name, int data_type, int dim1_size, int dim2_si
struct Tensor* readTrainedWeights(char* file_name, int data_type, int dim1_size, int dim2_size, struct Tensor* readTrainedWeights(const char* file_name, int data_type, int dim1_size, int dim2_size,
int dim3_size, int dim4_size){ int dim3_size, int dim4_size){
// FIXIT: Don't assume floating point types // FIXIT: Don't assume floating point types
......
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