From 5ba91c38708b5af6e227c0742426b805d39bfafe Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Sat, 15 Sep 2018 00:29:08 +0000
Subject: [PATCH] temp
---
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