2014-10-13 11:29:01 +04:00
|
|
|
#ifndef COST_LAYER_H
|
|
|
|
|
#define COST_LAYER_H
|
|
|
|
|
#include "opencl.h"
|
|
|
|
|
|
2014-11-28 21:38:26 +03:00
|
|
|
typedef enum{
|
|
|
|
|
SSE, DETECTION
|
|
|
|
|
} COST_TYPE;
|
|
|
|
|
|
2014-10-13 11:29:01 +04:00
|
|
|
typedef struct {
|
|
|
|
|
int inputs;
|
|
|
|
|
int batch;
|
|
|
|
|
float *delta;
|
|
|
|
|
float *output;
|
2014-11-28 21:38:26 +03:00
|
|
|
COST_TYPE type;
|
2014-10-13 11:29:01 +04:00
|
|
|
#ifdef GPU
|
|
|
|
|
cl_mem delta_cl;
|
|
|
|
|
#endif
|
|
|
|
|
} cost_layer;
|
|
|
|
|
|
2014-11-28 21:38:26 +03:00
|
|
|
COST_TYPE get_cost_type(char *s);
|
|
|
|
|
char *get_cost_string(COST_TYPE a);
|
|
|
|
|
cost_layer *make_cost_layer(int batch, int inputs, COST_TYPE type);
|
2014-10-13 11:29:01 +04:00
|
|
|
void forward_cost_layer(const cost_layer layer, float *input, float *truth);
|
|
|
|
|
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);
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
#endif
|
