From 5ef74c2031a040f30a670dc7d60790fc6a9ec720 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Fri, 02 May 2014 22:20:34 +0000 Subject: [PATCH] Slowly refactoring and pushing to GPU --- src/activations.h | 7 +++++-- 1 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/activations.h b/src/activations.h index 889453f..68d2222 100644 --- a/src/activations.h +++ b/src/activations.h @@ -7,8 +7,11 @@ ACTIVATION get_activation(char *s); -double activate(double x, ACTIVATION a); -double gradient(double x, ACTIVATION a); +char *get_activation_string(ACTIVATION a); +float activate(float x, ACTIVATION a); +float gradient(float x, ACTIVATION a); +void gradient_array(const float *x, const int n, const ACTIVATION a, float *delta); +void activate_array(float *x, const int n, const ACTIVATION a); #endif -- Gitblit v1.10.0