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

TensorRT version

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-01 17:51:49 +02:00
parent 1cfe70365f
commit ed5e5d58b5
6 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/kernels.h
+2 -2
View File
@@ -1,10 +1,10 @@
#include "utils.h"
#include "Layer.h"
void activationELUForward(value_type* srcData, value_type* dstData, int size);
void activationLEAKYForward(value_type* srcData, value_type* dstData, int size);
void activationLOGISTICForward(value_type* srcData, value_type* dstData, int size);
void reorgForward(value_type* srcData, value_type* dstData, tkDNN::dataDim_t dim, int stride);
void reorgForward( value_type* srcData, value_type* dstData,
int n, int c, int h, int w, int stride);
void softmaxForward(float *input, int n, int batch, int batch_offset,
int groups, int group_offset, int stride, float temp, float *output);