/*
	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 Takwrap.  If not, see <https://www.gnu.org/licenses/>.
*/

#define KERN_SIZE  3
#define KERN_CHAN  6
#define KERN_NUM   12
#define DENSE1_NUM 11
#define DENSE2_NUM 8

#define KERN_OSIZE (5-KERN_SIZE+1) // 5
#define CONV_NUM   (KERN_NUM * KERN_OSIZE * KERN_OSIZE) // 108

extern const float conv2d_weights[KERN_NUM][KERN_SIZE][KERN_SIZE][KERN_CHAN];
extern const float conv2d_biases[KERN_NUM];
extern const float dense1_weights[DENSE1_NUM][CONV_NUM+2];
extern const float dense1_biases[DENSE1_NUM];
extern const float dense2_weights[DENSE2_NUM][DENSE1_NUM];
extern const float dense2_biases[DENSE2_NUM];
extern const float output_weights[DENSE2_NUM];
extern const float output_bias;