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

good chance I didn't break anything

Browse Source
This commit is contained in:
Joseph Redmon
2016-09-12 13:55:20 -07:00
parent 8ec889f103
commit 5c067dc447
19 changed files with 558 additions and 298 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/layer.h
View File

@ -105,9 +105,7 @@ struct layer{
int *indexes;
float *rand;
float *cost;
float *filters;
char *cfilters;
float *filter_updates;
char *cweights;
float *state;
float *prev_state;
float *forgot_state;
@ -117,7 +115,7 @@ struct layer{
float *concat;
float *concat_delta;
float *binary_filters;
float *binary_weights;
float *biases;
float *bias_updates;
@ -194,11 +192,9 @@ struct layer{
float * save_delta_gpu;
float * concat_gpu;
float * concat_delta_gpu;
float * filters_gpu;
float * filter_updates_gpu;
float *binary_input_gpu;
float *binary_filters_gpu;
float *binary_weights_gpu;
float * mean_gpu;
float * variance_gpu;
@ -230,8 +226,8 @@ struct layer{
#ifdef CUDNN
cudnnTensorDescriptor_t srcTensorDesc, dstTensorDesc;
cudnnTensorDescriptor_t dsrcTensorDesc, ddstTensorDesc;
cudnnFilterDescriptor_t filterDesc;
cudnnFilterDescriptor_t dfilterDesc;
cudnnFilterDescriptor_t weightDesc;
cudnnFilterDescriptor_t dweightDesc;
cudnnConvolutionDescriptor_t convDesc;
cudnnConvolutionFwdAlgo_t fw_algo;
cudnnConvolutionBwdDataAlgo_t bd_algo;