From 08c7cf9c88befd845f00c00d85e40a9eead4b1b3 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sun, 19 Jun 2016 21:28:15 +0000
Subject: [PATCH] no mean on input binarization
---
src/layer.h | 10 ++++++++++
1 files changed, 10 insertions(+), 0 deletions(-)
diff --git a/src/layer.h b/src/layer.h
index d53fe38..0c457f7 100644
--- a/src/layer.h
+++ b/src/layer.h
@@ -28,6 +28,7 @@
CRNN,
BATCHNORM,
NETWORK,
+ XNOR,
BLANK
} LAYER_TYPE;
@@ -102,6 +103,9 @@
char *cfilters;
float *filter_updates;
float *state;
+ float *prev_state;
+ float *forgot_state;
+ float *forgot_delta;
float *state_delta;
float *concat;
@@ -159,6 +163,12 @@
struct layer *input_h_layer;
struct layer *state_h_layer;
+ float *z_cpu;
+ float *r_cpu;
+ float *h_cpu;
+
+ float *binary_input;
+
size_t workspace_size;
#ifdef GPU
--
Gitblit v1.10.0