From a9fef1bd66e6b2c40c344c1bdcd33bb1d209461c Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 10 Aug 2018 21:26:53 +0000
Subject: [PATCH] Bug fixes. Tested im2col_cpu_custom_transpose - bad way.
---
src/gemm.h | 7 +++++++
1 files changed, 7 insertions(+), 0 deletions(-)
diff --git a/src/gemm.h b/src/gemm.h
index c71cd24..62dbe30 100644
--- a/src/gemm.h
+++ b/src/gemm.h
@@ -4,6 +4,9 @@
#include <stdint.h>
#include <stddef.h>
+void convolution_2d(int w, int h, int ksize, int n, int c, int pad, int stride,
+ float *weights, float *input, float *output);
+
static inline void set_bit(unsigned char *const dst, size_t index) {
size_t dst_i = index / 8;
int dst_shift = index % 8;
@@ -31,6 +34,10 @@
int channels, int height, int width,
int ksize, int stride, int pad, float* data_col);
+void im2col_cpu_custom_transpose(float* data_im,
+ int channels, int height, int width,
+ int ksize, int stride, int pad, float* data_col, int ldb_align);
+
void activate_array_cpu_custom(float *x, const int n, const ACTIVATION a);
--
Gitblit v1.10.0