00001 #ifndef COST_LAYER_H 00002 #define COST_LAYER_H 00003 #include "layer.h" 00004 #include "network.h" 00005 00006 typedef layer cost_layer; 00007 00008 COST_TYPE get_cost_type(char *s); 00009 char *get_cost_string(COST_TYPE a); 00010 cost_layer make_cost_layer(int batch, int inputs, COST_TYPE type, float scale); 00011 void forward_cost_layer(const cost_layer l, network_state state); 00012 void backward_cost_layer(const cost_layer l, network_state state); 00013 void resize_cost_layer(cost_layer *l, int inputs); 00014 00015 #ifdef GPU 00016 void forward_cost_layer_gpu(cost_layer l, network_state state); 00017 void backward_cost_layer_gpu(const cost_layer l, network_state state); 00018 #endif 00019 00020 #endif