129 Commits

Author SHA1 Message Date
98ad4ac760 Migrate conv2d layer 2024-09-10 21:48:47 +02:00
757584544c Migrate conv2d layer 2024-09-10 21:19:50 +02:00
74f49d6a00 Migrate output layer 2024-09-10 19:20:00 +02:00
f7b525e494 Migrate input layer 2024-09-10 19:11:21 +02:00
fe7c16ac36 Migrate concat layer 2024-09-09 22:16:22 +02:00
a0665fb05c Migrate max pooling 2024-09-09 21:48:36 +02:00
8a5d556b7e Rename cpp version of avg pooling 2024-09-09 21:37:00 +02:00
76e5225001 Migrate avg pooling 2024-09-09 21:36:13 +02:00
75475790ac Migrate Dense layer 2024-09-08 13:36:53 +02:00
0dca8348bd Migrate Activation layer 2024-09-08 12:49:13 +02:00
591507ed21 Split tests to cpu and cuda 2024-09-07 21:24:34 +02:00
f8220f0ec1 Restructure cuda backend 2024-09-05 22:23:47 +02:00
65727dfee8 Move cuda source to separate directory 2024-09-05 17:39:09 +02:00
bc9bff10cd Load running mean and var from weight file 2024-08-25 19:33:33 +02:00
9704d0d53e Add running mean and running var to batchnorm 2024-08-25 19:05:10 +02:00
a4a2801ee4 Fix missing activation in batch norm 2024-06-02 14:41:14 +02:00
8168f02f58 Add adaptive avg pooling 2024-05-30 17:17:31 +02:00
479c1119e7 Fix module layer issues 2024-05-30 13:08:13 +02:00
8ac2da004c Fix adding module layers 2024-05-29 21:02:26 +02:00
046c1b0cc2 Improve model summary 2024-05-28 19:24:56 +02:00
098fd65074 Add model summary 2024-05-28 18:42:58 +02:00
df47a31f36 Rename dim2d to shape2d 2024-05-27 21:14:51 +02:00
94a16b4352 Add padding to max pooling 2024-05-26 19:03:10 +02:00
4a67b708f0 Add padding to avg pooling 2024-05-26 18:54:12 +02:00
cba177e417 Add getOutputDims to 2d layers 2024-05-26 14:28:43 +02:00
37c29d6734 Fix wrong conv2d output size calculation 2024-05-23 21:38:11 +02:00
6dca8ccd3c Unify 2d layer naming 2024-05-20 16:23:58 +02:00
74098b24e3 Add support for non square matrices 2024-05-20 15:20:43 +02:00
6f8b5f4081 Rename batchnorm 2024-05-20 13:05:48 +02:00
b5fb205df8 Update module add layer 2024-05-19 20:22:08 +02:00
33d4a43dca Update concat layer 2024-05-19 20:21:13 +02:00
e23ffe1ee1 Remove default constructor from Module 2024-05-19 16:16:57 +02:00
4a1c4a5f91 Add epsilon param to batch norm 2024-05-19 15:13:22 +02:00
c84f58b97c Implement modules 2024-05-15 18:52:31 +02:00
7c48ed86d2 Implement vector variance function 2024-05-14 21:58:23 +02:00
33a3c6afcb Fix batch norm layer 2024-05-14 21:42:01 +02:00
da8f3167cb Add utils vector mean function 2024-05-14 21:41:18 +02:00
5c8d3f7e25 Compute mean and variance 2024-04-29 20:55:11 +02:00
0ab623fa23 Implement vector mean calculation 2024-04-28 22:04:15 +02:00
96536a44fd Implement batch norm test 2024-04-28 21:46:42 +02:00
f60d62f6bd Implement batch norm layer 2024-04-28 19:58:00 +02:00
1f708f3ea3 Change vector scalar math kernels 2024-04-28 19:03:06 +02:00
69111f6cb1 Add version to model bin format 2024-04-23 19:53:48 +02:00
74262309ad Fix max pooling output size calc 2024-04-22 20:58:02 +02:00
c2acad151b Implement simple model validation 2024-04-22 20:57:40 +02:00
f17debc244 Implement getOutputSize and getInputSize for seq layers 2024-04-22 20:31:58 +02:00
a32c737785 Allocate activation on heap 2024-04-22 18:59:16 +02:00
26cea9b12c Fix more cuda memcheck errors 2024-04-21 22:56:55 +02:00
58af95eb25 Remove cudaDeviceReset calls from tests 2024-04-21 22:47:12 +02:00
bdbb3f978e Fix matmul and max reduce memcheck errors 2024-04-21 22:11:02 +02:00