From a723e1c62a27aeb39aaf7fcdeb3beb4e89fba32d Mon Sep 17 00:00:00 2001 From: Alexey <AlexeyAB@users.noreply.github.com> Date: Wed, 15 Aug 2018 20:52:09 +0000 Subject: [PATCH] Merge pull request #766 from HotChick91/AlexeyAB-mask --- src/route_layer.h | 9 +++++---- 1 files changed, 5 insertions(+), 4 deletions(-) diff --git a/src/route_layer.h b/src/route_layer.h index 1f0d6e3..45467d9 100644 --- a/src/route_layer.h +++ b/src/route_layer.h @@ -6,12 +6,13 @@ 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 l, network net); -void backward_route_layer(const route_layer l, network net); +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 l, network net); -void backward_route_layer_gpu(const route_layer l, 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