Joseph Redmon
2014-10-14 7756cccb793bb4950c241f2804195ea859d1b407
src/connected_layer.h
@@ -2,6 +2,7 @@
#define CONNECTED_LAYER_H
#include "activations.h"
#include "opencl.h"
typedef struct{
    float learning_rate;
@@ -20,12 +21,22 @@
    float *weight_adapt;
    float *bias_adapt;
    float *weight_momentum;
    float *bias_momentum;
    float *output;
    float *delta;
    
    #ifdef GPU
    cl_mem weights_cl;
    cl_mem biases_cl;
    cl_mem weight_updates_cl;
    cl_mem bias_updates_cl;
    cl_mem weight_momentum_cl;
    cl_mem bias_momentum_cl;
    cl_mem output_cl;
    cl_mem delta_cl;
    #endif
    ACTIVATION activation;
} connected_layer;