laurent a57b314780 Add a batch dimension on the bert example.

2023-07-04 06:10:52 +01:00

..

backprop.rs

Use broadcasted scalars for const tensors.

2023-06-29 11:56:40 +01:00

cpu_backend.rs

Add a batch dimension on the bert example.

2023-07-04 06:10:52 +01:00

cuda_backend.rs

Add a batch dimension on the bert example.

2023-07-04 06:10:52 +01:00

device.rs

Refactor the hierarchy.

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

display.rs

Add the kernels.

2023-06-30 10:26:56 +01:00

dtype.rs

Add dtype support.

2023-07-02 20:12:26 +01:00

dummy_cuda_backend.rs

Simplify the pattern matching logic in the cuda backend.

2023-06-29 09:21:11 +01:00

error.rs

Add a batch dimension on the bert example.

2023-07-04 06:10:52 +01:00

layout.rs

Get the cpu backend to compile.

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

lib.rs

Move some safetensors bits in the candle-core crate.

2023-07-03 08:37:46 +01:00

npy.rs

Add the kernels.

2023-06-30 10:26:56 +01:00

op.rs

Add the kernels.

2023-06-30 10:26:56 +01:00

safetensors.rs

Address comments.

2023-07-03 13:52:27 +02:00

shape.rs

Use broadcasted scalars for const tensors.

2023-06-29 11:56:40 +01:00

storage.rs

Simplify the pattern matching logic in the cuda backend.

2023-06-29 09:21:11 +01:00

strided_index.rs

Get the cpu tests to run.

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

tensor.rs

Improve how we check that the dims are in bounds.

2023-06-30 09:11:00 +01:00

utils.rs

PyTorch like display implementation.

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