From 8436251a05ec0cfa906089ee022db17498e6f65f Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Tue, 21 Aug 2018 17:44:18 +0000
Subject: [PATCH] maxpool_layer on CPU uses AVX2 and OpenMP
---
src/gemm.h | 5 +++++
1 files changed, 5 insertions(+), 0 deletions(-)
diff --git a/src/gemm.h b/src/gemm.h
index 1472cca..3cdca99 100644
--- a/src/gemm.h
+++ b/src/gemm.h
@@ -46,6 +46,11 @@
float *B, int ldb,
float *C, int ldc);
+
+void forward_maxpool_layer_avx(float *src, float *dst, int *indexes, int size, int w, int h, int out_w, int out_h, int c,
+ int pad, int stride, int batch);
+
+
void gemm(int TA, int TB, int M, int N, int K, float ALPHA,
float *A, int lda,
float *B, int ldb,
--
Gitblit v1.10.0