| | |
| | | #ifndef COST_LAYER_H |
| | | #define COST_LAYER_H |
| | | #include "opencl.h" |
| | | |
| | | typedef enum{ |
| | | SSE, DETECTION |
| | | SSE |
| | | } COST_TYPE; |
| | | |
| | | typedef struct { |
| | | int inputs; |
| | | int batch; |
| | | int coords; |
| | | int classes; |
| | | float *delta; |
| | | float *output; |
| | | COST_TYPE type; |
| | | #ifdef GPU |
| | | cl_mem delta_cl; |
| | | float * delta_gpu; |
| | | #endif |
| | | } cost_layer; |
| | | |
| | |
| | | void backward_cost_layer(const cost_layer layer, float *input, float *delta); |
| | | |
| | | #ifdef GPU |
| | | void forward_cost_layer_gpu(cost_layer layer, cl_mem input, cl_mem truth); |
| | | void backward_cost_layer_gpu(const cost_layer layer, cl_mem input, cl_mem delta); |
| | | void forward_cost_layer_gpu(cost_layer layer, float * input, float * truth); |
| | | void backward_cost_layer_gpu(const cost_layer layer, float * input, float * delta); |
| | | #endif |
| | | |
| | | #endif |