1 files changed, 4 insertions, 3 deletions
diff --git a/src/nn.h b/src/nn.h
index 88b4462..05ddd28 100644
--- a/src/nn.h
+++ b/src/nn.h
@@ -15,12 +15,13 @@ typedef struct Layer {
     size_t neurons, input_nodes;
 } Layer;
 
-void nn_layer_init_weights(Layer *layer, size_t nmemb, size_t input_cols);
+void nn_layer_init_weights(Layer *layers, size_t nmemb, size_t input_cols);
 void nn_layer_free_weights(Layer *layer, size_t nmemb);
 
-double * nn_layer_forward(Layer layer, double *input, size_t input_shape[2]); //TODO
-double * nn_layer_backward(Layer layer, double *output, size_t out_shape[2]); //TODO
+void nn_layer_forward(Layer layer, double *out, size_t out_shape[2], double *input, size_t input_shape[2]); //TODO
+void nn_layer_backward(Layer *layer, double *out, size_t out_shape[2]); //TODO
 
 double sigmoid(double x);
 double relu(double x);
+double identity(double x);
 #endif