mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

art, cudnn

Browse Source
This commit is contained in:
Joseph Redmon
2016-05-13 11:59:43 -07:00
parent 054e2b1954
commit 13209df7bb
11 changed files with 286 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/layer.h
View File

@ -2,6 +2,7 @@
#define BASE_LAYER_H
#include "activations.h"
#include "stddef.h"
struct layer;
typedef struct layer layer;
@ -157,6 +158,8 @@ struct layer{
struct layer *input_h_layer;
struct layer *state_h_layer;
size_t workspace_size;
#ifdef GPU
float *z_gpu;
float *r_gpu;
@ -207,6 +210,16 @@ struct layer{
float * rand_gpu;
float * squared_gpu;
float * norms_gpu;
#ifdef CUDNN
cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;
cudnnTensorDescriptor_t dsrcTensorDesc, ddstTensorDesc;
cudnnFilterDescriptor_t filterDesc;
cudnnFilterDescriptor_t dfilterDesc;
cudnnConvolutionDescriptor_t convDesc;
cudnnConvolutionFwdAlgo_t fw_algo;
cudnnConvolutionBwdDataAlgo_t bd_algo;
cudnnConvolutionBwdFilterAlgo_t bf_algo;
#endif
#endif
};