From aa5996d58e68edfbefe51061856aecd549dd09c4 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 13 Jan 2015 01:27:08 +0000
Subject: [PATCH] Faster
---
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