LordMathis 0dca8348bd Migrate Activation layer

2024-09-08 12:49:13 +02:00

..

activation.hpp

Migrate Activation layer

2024-09-08 12:49:13 +02:00

add.hpp

Restructure cuda backend

2024-09-05 22:23:47 +02:00

avg_pooling.cuh

Add adaptive avg pooling

2024-05-30 17:17:31 +02:00

batch_norm.cuh

Load running mean and var from weight file

2024-08-25 19:33:33 +02:00

concat.cuh

Update concat layer

2024-05-19 20:21:13 +02:00

conv2d.cuh

Rename dim2d to shape2d

2024-05-27 21:14:51 +02:00

dense.cuh

Add support for non square matrices

2024-05-20 15:20:43 +02:00

input.cuh

Implement getOutputSize and getInputSize for seq layers

2024-04-22 20:31:58 +02:00

layer.cuh

Rename dim2d to shape2d

2024-05-27 21:14:51 +02:00

max_pooling.cuh

Implement Inception block D

2024-05-27 21:30:42 +02:00

output.cuh

Implement getOutputSize and getInputSize for seq layers

2024-04-22 20:31:58 +02:00