Joseph Redmon
2014-12-18 9e98f1b68e2ebf618d7f65a0985f5efd11e413f9
src/softmax_layer.h
@@ -20,6 +20,7 @@
void backward_softmax_layer(const softmax_layer layer, float *delta);
#ifdef GPU
void pull_softmax_layer_output(const softmax_layer layer);
void forward_softmax_layer_gpu(const softmax_layer layer, cl_mem input);
void backward_softmax_layer_gpu(const softmax_layer layer, cl_mem delta);
#endif