Fix conv2d

pull/107/head
xiangbingj 2019-11-07 16:19:40 +08:00
parent aa41257ea6
commit 5d96164352
2 changed files with 3 additions and 3 deletions

View File

@ -33,7 +33,7 @@ extern "C" {
#define KENDRYTE_MIN(a, b) ((a) > (b) ? (b) : (a))
#define KENDRYTE_MAX(a, b) ((a) > (b) ? (a) : (b))
#define FIX_CACHE 1
#define FIX_CACHE 0
#ifdef __ASSEMBLY__
#define KENDRYTE_CAST(type, ptr) ptr

View File

@ -82,8 +82,8 @@ namespace kernels
memcpy(cache_weights, weights, (size_t)out_channels * in_shape[1] / groups * filter_h * filter_w*sizeof(float));
memcpy(cache_bias, bias, out_channels*sizeof(float));
#else
float *cache_weights = weights;
float *cache_bias = bias;
const float *cache_weights = weights;
const float *cache_bias = bias;
#endif
for (int32_t batch = 0; batch < in_shape[0]; batch++)