LordMathis 10e73638b6 Add non square pooling and batch norm tests

2024-05-20 22:16:00 +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

Unify 2d layer naming

2024-05-20 16:23:58 +02:00

batch_norm.cuh

Unify 2d layer naming

2024-05-20 16:23:58 +02:00

concat.cuh

Update concat layer

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

conv2d.cuh

Add support for non square matrices

2024-05-20 15:20:43 +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

Add support for non square matrices

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

max_pooling.cuh

Add non square pooling and batch norm tests

2024-05-20 22:16:00 +02:00

output.cuh

Implement getOutputSize and getInputSize for seq layers

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