diff options
Diffstat (limited to 'include/weights_5.h')
| -rw-r--r-- | include/weights_5.h | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/include/weights_5.h b/include/weights_5.h index db2a69f..16a9465 100644 --- a/include/weights_5.h +++ b/include/weights_5.h @@ -21,16 +21,14 @@ #define FIVE_INP_NUM 3 + 5 * 5 #define FIVE_DENSE_NUM 64 -/* - * 16-byte alignment so the evaluation dot products can be - * auto-vectorised (SSE2 / NEON); the generated weights file defines - * the arrays with the same attribute. - */ +// this does some kind of magic vectorisation #define NN_ALIGNED __attribute__((aligned(16))) +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_biases[FIVE_DENSE_NUM]; extern const float five_logit_weights[FIVE_DENSE_NUM] NN_ALIGNED; extern const float five_logit_bias; + #endif |