diff options
Diffstat (limited to 'include/weights_5.h')
| -rw-r--r-- | include/weights_5.h | 22 |
1 files changed, 17 insertions, 5 deletions
diff --git a/include/weights_5.h b/include/weights_5.h index 6f376ee..db2a69f 100644 --- a/include/weights_5.h +++ b/include/weights_5.h @@ -1,5 +1,5 @@ /* - This file is part of ct. + This file is part of ctak. This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -12,13 +12,25 @@ General Public License for more details. You should have received a copy of the GNU General Public License - along with ct. If not, see <https://www.gnu.org/licenses/>. + along with ctak. If not, see <https://www.gnu.org/licenses/>. */ +#ifndef WEIGHTS_5_H +#define WEIGHTS_5_H + #define FIVE_INP_NUM 3 + 5 * 5 #define FIVE_DENSE_NUM 64 -extern const float five_dense_weights[FIVE_DENSE_NUM][FIVE_INP_NUM]; +/* + * 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. + */ +#define NN_ALIGNED __attribute__((aligned(16))) + +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_output_weights[2][FIVE_DENSE_NUM]; -extern const float five_output_bias[2]; +extern const float five_logit_weights[FIVE_DENSE_NUM] NN_ALIGNED; +extern const float five_logit_bias; + +#endif |