From 845ab7579685b6702c92c1088ec11e71bde51f3c Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 05 Aug 2016 22:27:07 +0000
Subject: [PATCH] some more stuff

---
 src/blas.h |    2 ++
 1 files changed, 2 insertions(+), 0 deletions(-)

diff --git a/src/blas.h b/src/blas.h
index b4cfcf2..4fdaa41 100644
--- a/src/blas.h
+++ b/src/blas.h
@@ -1,5 +1,6 @@
 #ifndef BLAS_H
 #define BLAS_H
+void reorg(float *x, int size, int layers, int batch, int forward);
 void pm(int M, int N, float *A);
 float *random_matrix(int rows, int cols);
 void time_random_matrix(int TA, int TB, int m, int k, int n);
@@ -69,6 +70,7 @@
 void weighted_sum_gpu(float *a, float *b, float *s, int num, float *c);
 void mult_add_into_gpu(int num, float *a, float *b, float *c);
 
+void reorg_ongpu(float *x, int w, int h, int c, int batch, int stride, int forward, float *out);
 
 #endif
 #endif

--
Gitblit v1.10.0