From bc810016a1f2eadf33a1ac800b64962a42f3d402 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 02 Aug 2017 22:36:22 +0000
Subject: [PATCH] cuDNN 6.0 supported. Also speed of console example improved.
---
src/route_layer.h | 26 ++++++++------------------
1 files changed, 8 insertions(+), 18 deletions(-)
diff --git a/src/route_layer.h b/src/route_layer.h
index 086ef87..45467d9 100644
--- a/src/route_layer.h
+++ b/src/route_layer.h
@@ -1,28 +1,18 @@
#ifndef ROUTE_LAYER_H
#define ROUTE_LAYER_H
#include "network.h"
+#include "layer.h"
-typedef struct {
- int batch;
- int outputs;
- int n;
- int * input_layers;
- int * input_sizes;
- float * delta;
- float * output;
- #ifdef GPU
- float * delta_gpu;
- float * output_gpu;
- #endif
-} route_layer;
+typedef layer route_layer;
-route_layer *make_route_layer(int batch, int n, int *input_layers, int *input_size);
-void forward_route_layer(const route_layer layer, network net);
-void backward_route_layer(const route_layer layer, network net);
+route_layer make_route_layer(int batch, int n, int *input_layers, int *input_size);
+void forward_route_layer(const route_layer l, network_state state);
+void backward_route_layer(const route_layer l, network_state state);
+void resize_route_layer(route_layer *l, network *net);
#ifdef GPU
-void forward_route_layer_gpu(const route_layer layer, network net);
-void backward_route_layer_gpu(const route_layer layer, network net);
+void forward_route_layer_gpu(const route_layer l, network_state state);
+void backward_route_layer_gpu(const route_layer l, network_state state);
#endif
#endif
--
Gitblit v1.10.0