laurent 6c9e6b5a99 Get the cuda tests to pass.

2023-06-28 15:53:23 +01:00

..

backprop.rs

Add the grad for narrow.

2023-06-28 10:46:00 +01:00

cpu_backend.rs

Adapt the cuda bits.

2023-06-28 15:43:03 +01:00

cuda_backend.rs

Get the cuda tests to pass.

2023-06-28 15:53:23 +01:00

device.rs

Refactor the hierarchy.

2023-06-27 11:57:27 +02:00

display.rs

Add some display tests + bugfixes.

2023-06-27 21:37:28 +01:00

dtype.rs

Get the cpu tests to run.

2023-06-28 14:32:02 +01:00

dummy_cuda_backend.rs

Adapt the cuda bits.

2023-06-28 15:43:03 +01:00

error.rs

Refactor the hierarchy.

2023-06-27 11:57:27 +02:00

layout.rs

Get the cpu backend to compile.

2023-06-28 14:12:38 +01:00

lib.rs

Start refactoring the stride.

2023-06-28 12:57:30 +01:00

npy.rs

Refactor the hierarchy.

2023-06-27 11:57:27 +02:00

op.rs

Add the relu op.

2023-06-28 09:38:54 +01:00

shape.rs

Refactor the hierarchy.

2023-06-27 11:57:27 +02:00

storage.rs

Adapt the cuda bits.

2023-06-28 15:43:03 +01:00

strided_index.rs

Get the cpu tests to run.

2023-06-28 14:32:02 +01:00

tensor.rs

Adapt the cuda bits.

2023-06-28 15:43:03 +01:00

utils.rs

PyTorch like display implementation.

2023-06-27 21:16:35 +01:00