mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

mnist tensorrt incomplete

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-01 18:58:59 +02:00
parent bed0b57fad
commit 714bd5f757
4 changed files with 135 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/utils.h
+2 -2
View File
@@ -63,8 +63,8 @@
}
void readBinaryFile(const char* fname, int size, value_type** data_h, value_type** data_d, int seek = 0);
int checkResult(int size, value_type *data_d, value_type *correct_d);
void printDeviceVector(int size, value_type* vec_d);
int checkResult(int size, value_type *data_d, value_type *correct_d, bool device = true);
void printDeviceVector(int size, value_type* vec_d, bool device = true);
void resize(int size, value_type **data);
void matrixTranspose(cublasHandle_t handle, value_type* srcData, value_type* dstData, int rows, int cols);