laurent 817e4b5005 Rework the embeddings so that it works on non-contiguous weights + factor out some code.

2023-06-25 17:37:47 +01:00

..

backprop.rs

Boilerplate code for the sum operator.

2023-06-25 09:35:17 +01:00

cpu_backend.rs

Rework the embeddings so that it works on non-contiguous weights + factor out some code.

2023-06-25 17:37:47 +01:00

cuda_backend.rs

Rework the embeddings so that it works on non-contiguous weights + factor out some code.

2023-06-25 17:37:47 +01:00

device.rs

Helper function to build 3d arrays.

2023-06-24 06:29:06 +01:00

dtype.rs

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

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

dummy_cuda_backend.rs

Rework the embeddings so that it works on non-contiguous weights + factor out some code.

2023-06-25 17:37:47 +01:00

error.rs

More general broadcast setup.

2023-06-25 08:55:09 +01:00

lib.rs

Move the backprop bits to a separate file.

2023-06-24 20:57:49 +01:00

npy.rs

Read and write npy files.

2023-06-24 18:12:10 +01:00

op.rs

Boilerplate code for the sum operator.

2023-06-25 09:35:17 +01:00

shape.rs

Support wider shapes for llama.

2023-06-24 20:08:18 +01:00

storage.rs

Rework the embeddings so that it works on non-contiguous weights + factor out some code.

2023-06-25 17:37:47 +01:00

strided_index.rs

Move the StridedIndex in its own module.

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

tensor.rs

Rework the embeddings so that it works on non-contiguous weights + factor out some code.

2023-06-25 17:37:47 +01:00