From 9bae70b22549b68f5cdeece8b6c3b3de00c22714 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Mon, 16 Apr 2018 23:51:11 +0000 Subject: [PATCH] Accelerated by another 5% using FP16/32 Batch-norm for Tensor Cores. --- src/crop_layer.h | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/src/crop_layer.h b/src/crop_layer.h index b409351..3aa2d3d 100644 --- a/src/crop_layer.h +++ b/src/crop_layer.h @@ -2,7 +2,6 @@ #define CROP_LAYER_H #include "image.h" -#include "params.h" #include "layer.h" #include "network.h" @@ -11,6 +10,7 @@ image get_crop_image(crop_layer l); crop_layer make_crop_layer(int batch, int h, int w, int c, int crop_height, int crop_width, int flip, float angle, float saturation, float exposure); void forward_crop_layer(const crop_layer l, network_state state); +void resize_crop_layer(layer *l, int w, int h); #ifdef GPU void forward_crop_layer_gpu(crop_layer l, network_state state); -- Gitblit v1.10.0