From 31b6b0bad3e9111aac4792ebfe37218835ce8efc Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 22 Aug 2018 23:44:21 +0000
Subject: [PATCH] XNOR-net 4x acceleration on CPU for yolov2-tiny - 22 FPS (CPU Core i7 6700K)
---
src/gemm.h | 4 ++++
1 files changed, 4 insertions(+), 0 deletions(-)
diff --git a/src/gemm.h b/src/gemm.h
index d393b0d..90b8ce9 100644
--- a/src/gemm.h
+++ b/src/gemm.h
@@ -39,6 +39,10 @@
int channels, int height, int width,
int ksize, int stride, int pad, float* data_col);
+void im2col_cpu_custom_align(float* data_im,
+ int channels, int height, int width,
+ int ksize, int stride, int pad, float* data_col, int bit_align);
+
void im2col_cpu_custom_bin(float* data_im,
int channels, int height, int width,
int ksize, int stride, int pad, float* data_col, int bit_align);
--
Gitblit v1.10.0