From 1edcf73a73d2007afc61289245763f5cf0c29e10 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Thu, 04 Dec 2014 07:20:29 +0000 Subject: [PATCH] Detection good, split up col images --- src/convolutional_layer.h | 4 ++-- 1 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/convolutional_layer.h b/src/convolutional_layer.h index 970a9b1..1ceca16 100644 --- a/src/convolutional_layer.h +++ b/src/convolutional_layer.h @@ -47,7 +47,7 @@ #ifdef GPU void forward_convolutional_layer_gpu(convolutional_layer layer, cl_mem in); -void backward_convolutional_layer_gpu(convolutional_layer layer, cl_mem delta_cl); +void backward_convolutional_layer_gpu(convolutional_layer layer, cl_mem in, cl_mem delta_cl); void update_convolutional_layer_gpu(convolutional_layer layer); void push_convolutional_layer(convolutional_layer layer); #endif @@ -58,7 +58,7 @@ void update_convolutional_layer(convolutional_layer layer); image *visualize_convolutional_layer(convolutional_layer layer, char *window, image *prev_filters); -void backward_convolutional_layer(convolutional_layer layer, float *delta); +void backward_convolutional_layer(convolutional_layer layer, float *in, float *delta); image get_convolutional_image(convolutional_layer layer); image get_convolutional_delta(convolutional_layer layer); -- Gitblit v1.10.0