From f92b20580a21663c5db9eb8608f8cabd7adbeb10 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 13 Aug 2018 22:51:31 +0000
Subject: [PATCH] Some fixes for AVX support on CPU
---
src/gemm.h | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/gemm.h b/src/gemm.h
index 62dbe30..1472cca 100644
--- a/src/gemm.h
+++ b/src/gemm.h
@@ -5,7 +5,7 @@
#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);
+ float *weights, float *input, float *output, float *mean);
static inline void set_bit(unsigned char *const dst, size_t index) {
size_t dst_i = index / 8;
--
Gitblit v1.10.0