/* 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 the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with ctak. If not, see . */ #ifndef WEIGHTS_5_H #define WEIGHTS_5_H #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. */ #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_logit_weights[FIVE_DENSE_NUM] NN_ALIGNED; extern const float five_logit_bias; #endif