laurent d839d5d9fd Basic support for broadcasting backprop.

2023-06-23 16:31:44 +01:00

..

cpu_backend.rs

Bugfix for the strided copy + add some assertions.

2023-06-23 16:28:18 +01:00

cuda_backend.rs

Bugfix for the strided copy + add some assertions.

2023-06-23 16:28:18 +01:00

device.rs

Transfer tensors between devices.

2023-06-23 08:35:22 +01:00

dtype.rs

Adding embedding op (not generic gather, no select).

2023-06-23 13:13:26 +02:00

dummy_cuda_backend.rs

Adding embedding op (not generic gather, no select).

2023-06-23 13:13:26 +02:00

error.rs

Adding embedding op (not generic gather, no select).

2023-06-23 13:13:26 +02:00

lib.rs

Add a dummy module.

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

op.rs

More u32 support.

2023-06-23 14:54:03 +01:00

shape.rs

Abstract the implementation of Shape.

2023-06-22 12:39:15 +01:00

storage.rs

Dummy broadcast placeholder functions.

2023-06-23 14:07:05 +01:00

strided_index.rs

Move the StridedIndex in its own module.

2023-06-21 07:44:36 +01:00

tensor.rs

Basic support for broadcasting backprop.

2023-06-23 16:31:44 +01:00