From dda993f3dd3c753dfd580d485b39c1001830fee4 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 22 Feb 2018 19:54:40 +0000 Subject: [PATCH] Use half_float16 instead of float32 if defined both CUDNN and CUDNN_HALF. Use Tensor Cores. --- src/route_layer.h | 1 + 1 files changed, 1 insertions(+), 0 deletions(-) diff --git a/src/route_layer.h b/src/route_layer.h index 77245a6..45467d9 100644 --- a/src/route_layer.h +++ b/src/route_layer.h @@ -8,6 +8,7 @@ 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 l, network_state state); -- Gitblit v1.10.0