laurent a9c113248a Take references as input for Tensor::cat.

2023-06-25 13:03:05 +01:00

..

backprop.rs

Boilerplate code for the sum operator.

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

cpu_backend.rs

Fix the reduce_sum implementation and add some tests.

2023-06-25 10:55:04 +01:00

cuda_backend.rs

Boilerplate code for the sum operator.

2023-06-25 09:35:17 +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

Boilerplate code for the sum operator.

2023-06-25 09:35:17 +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

Boilerplate code for the sum operator.

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

strided_index.rs

Move the StridedIndex in its own module.

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

tensor.rs

Take references as input for Tensor::cat.

2023-06-25 13:03:05 +01:00