Joseph Redmon
2014-12-16 d6cdea49eb509af852e6279922863cf63adb4a56
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