LordMathis 8168f02f58 Add adaptive avg pooling

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

..

activation.cuh

Cleanup and refactor

2024-04-11 22:52:41 +02:00

add.cuh

Refactor layers

2024-03-19 21:35:05 +01:00

avg_pooling.cuh

Add adaptive avg pooling

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

batch_norm.cuh

Rename dim2d to shape2d

2024-05-27 21:14:51 +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