From f856f6e7f1fd0a11244527b266c1fd47334fb5f3 Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Sun, 16 Sep 2018 01:54:09 +0000
Subject: [PATCH] 10-class model for card detection
---
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