From 6e1d5b45de988bb795c4c505f22f2170a78b7746 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 20 Jan 2015 06:06:18 +0000
Subject: [PATCH] fast sort of working
---
src/dropout_layer.h | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/src/dropout_layer.h b/src/dropout_layer.h
index 0a6f034..12a819e 100644
--- a/src/dropout_layer.h
+++ b/src/dropout_layer.h
@@ -8,8 +8,10 @@
float probability;
float scale;
float *rand;
+ float *output;
#ifdef GPU
cl_mem rand_cl;
+ cl_mem output_cl;
#endif
} dropout_layer;
--
Gitblit v1.10.0