Francesco Gatti
22 lines
1.1 KiB
C
22 lines
1.1 KiB
C
#ifndef KERNELS_H
|
|
#define KERNELS_H
|
|
|
|
#include "utils.h"
|
|
|
|
void activationELUForward(dnnType* srcData, dnnType* dstData, int size, cudaStream_t stream = cudaStream_t(0));
|
|
void activationLEAKYForward(dnnType* srcData, dnnType* dstData, int size, cudaStream_t stream = cudaStream_t(0));
|
|
void activationLOGISTICForward(dnnType* srcData, dnnType* dstData, int size, cudaStream_t stream = cudaStream_t(0));
|
|
|
|
void reorgForward( dnnType* srcData, dnnType* dstData,
|
|
int n, int c, int h, int w, int stride, cudaStream_t stream = cudaStream_t(0));
|
|
void softmaxForward(float *input, int n, int batch, int batch_offset,
|
|
int groups, int group_offset, int stride, float temp, float *output, cudaStream_t stream = cudaStream_t(0));
|
|
|
|
|
|
void shortcutForward(dnnType* srcData, dnnType* dstData, int n1, int c1, int h1, int w1, int s1,
|
|
int n2, int c2, int h2, int w2, int s2,
|
|
cudaStream_t stream = cudaStream_t(0));
|
|
|
|
void float2half(float* srcData, __half* dstData, int size, const cudaStream_t stream = cudaStream_t(0));
|
|
#endif //KERNELS_H
|