Fix conv2d
parent
aa41257ea6
commit
5d96164352
|
@ -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
|
||||
|
|
|
@ -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++)
|
||||
|
|
Loading…
Reference in New Issue