LordMathis 1102aef293 Implement custom Shape struct with __device__ support

2025-11-23 19:21:06 +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

Fix some compilation errors

2025-11-23 18:50:57 +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

WIP Migrate AvgPool2d

2025-11-19 23:21:18 +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