From 9a01e6ccb7a74ff77e99060cf18acd6cfdb74b8e Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 11 Nov 2016 16:48:40 +0000
Subject: [PATCH] :fire: crush. crush. admit. :fire:
---
src/route_layer.h | 25 +++++++------------------
1 files changed, 7 insertions(+), 18 deletions(-)
diff --git a/src/route_layer.h b/src/route_layer.h
index 086ef87..77245a6 100644
--- a/src/route_layer.h
+++ b/src/route_layer.h
@@ -1,28 +1,17 @@
#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);
#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