diff options
Diffstat (limited to 'include/weights_6.h')
| -rw-r--r-- | include/weights_6.h | 18 |
1 files changed, 13 insertions, 5 deletions
diff --git a/include/weights_6.h b/include/weights_6.h index 8da555d..b572ffa 100644 --- a/include/weights_6.h +++ b/include/weights_6.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,21 @@ 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_6_H +#define WEIGHTS_6_H + #define SIX_INP_NUM (3 + 6 * 6) #define SIX_DENSE_NUM 64 -extern const float six_dense_weights[SIX_DENSE_NUM][SIX_INP_NUM]; +/* See weights_5.h for the alignment rationale. */ +#define NN_ALIGNED __attribute__((aligned(16))) + +extern const float six_dense_weights[SIX_DENSE_NUM][SIX_INP_NUM] NN_ALIGNED; extern const float six_dense_biases[SIX_DENSE_NUM]; -extern const float six_output_weights[2][SIX_DENSE_NUM]; -extern const float six_output_bias[2]; +extern const float six_logit_weights[SIX_DENSE_NUM] NN_ALIGNED; +extern const float six_logit_bias; + +#endif |