diff options
Diffstat (limited to 'include/weights_5.h')
| -rw-r--r-- | include/weights_5.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/include/weights_5.h b/include/weights_5.h index d48a05a..74a4366 100644 --- a/include/weights_5.h +++ b/include/weights_5.h @@ -22,10 +22,10 @@ #define FIVE_DENSE_NUM 64 // this does some kind of magic vectorisation -#define NN_ALIGNED __attribute__((aligned(16))) +#define NN_ALIGNED __attribute__((aligned(32))) extern const float FIVE_embedding[7]; -extern const float FIVE_dense_weights[FIVE_DENSE_NUM][FIVE_INP_NUM] NN_ALIGNED; +extern const float FIVE_dense_weights[FIVE_INP_NUM][FIVE_DENSE_NUM] NN_ALIGNED; extern const float FIVE_dense_biases[FIVE_DENSE_NUM]; extern const float FIVE_logit_weights[FIVE_DENSE_NUM] NN_ALIGNED; extern const float FIVE_logit_bias; |