LordMathis 4161caf3e1 Update BatchNorm2d to return sizes for running mean and var

2025-11-23 20:48:41 +01:00

..

backend

Fix some compilation errors

2025-11-23 18:50:57 +01:00

kernels

Fix some compilation errors

2025-11-23 18:50:57 +01:00

layers

Update BatchNorm2d to return sizes for running mean and var

2025-11-23 20:48:41 +01:00

utils

Fix some compilation errors

2025-11-23 18:50:57 +01:00

backend.hpp

Fix some compilation errors

2025-11-23 18:50:57 +01:00

cudanet.cuh

Migrate conv2d layer

2024-09-10 21:48:47 +02:00

layer.hpp

Refactor Layer interface to return size of weights and biases instead of Tensor references

2025-11-23 20:44:25 +01:00

model.hpp

Migrate model class to Tensor

2025-11-22 22:40:38 +01:00

module.hpp

Migrate model class to Tensor

2025-11-22 22:40:38 +01:00

shape.hpp

Implement custom Shape struct with __device__ support

2025-11-23 19:21:06 +01:00

tensor.hpp

Migrate batch norm layer

2025-11-21 23:24:14 +01:00