Laurent Mazare 6fb665004c Enable im2col on the cpu side. (#805) ...

* Enable im2col on the cpu side.

* Hook im2col on the cpu backend.

* Use the kernel offset.

* Avoid an unnecessary copy.

* Handle non-contiguous kernels.

* Add a const to select the conv2d kernel.

2023-09-11 09:28:13 +01:00

..

cpu

Tweaks to softmax. (#745)

2023-09-05 15:22:27 +01:00

quantized

Fix some errors about BlockQ8_1 (#776)

2023-09-08 13:29:40 +01:00

accelerate.rs

Accelerate support for gelu. (#782)

2023-09-08 21:58:56 +01:00

backend.rs

Add the powf op. (#664)

2023-08-29 20:48:18 +01:00

backprop.rs

Add tanh. (#675)

2023-08-30 13:54:50 +01:00

conv.rs

Dilated convolutions (#657)

2023-08-29 16:12:11 +01:00

convert.rs

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

cpu_backend.rs

Enable im2col on the cpu side. (#805)

2023-09-11 09:28:13 +01:00

cuda_backend.rs

Add Im2Col support on the gpu side. (#808)

2023-09-11 08:52:33 +01:00

cudnn.rs

Cudnn fix. (#758)

2023-09-06 17:39:39 +01:00

device.rs

Handle arbitrary shapes in Tensor::new. (#718)

2023-09-02 19:59:21 +01:00

display.rs

Fix the debug implementation. (#648)

2023-08-28 22:51:39 +01:00

dtype.rs

Add some documentation. (#743)

2023-09-05 09:51:12 +01:00

dummy_cuda_backend.rs

Add the powf op. (#664)

2023-08-29 20:48:18 +01:00

error.rs

Segment Anything - process images (#766)

2023-09-07 19:22:45 +01:00

indexer.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

layout.rs

Retrieve more information from PyTorch checkpoints. (#515)

2023-08-19 15:05:34 +01:00

lib.rs

Remove set_training. (#784)

2023-09-09 08:27:37 +01:00

mkl.rs

Add tanh. (#675)

2023-08-30 13:54:50 +01:00

npy.rs

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

op.rs

Accelerate support for gelu. (#782)

2023-09-08 21:58:56 +01:00

pickle.rs

More pickle support. (#588)

2023-08-24 18:45:10 +01:00

safetensors.rs

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

scalar.rs

Get the comparison operation to work on scalar values. (#780)

2023-09-08 20:13:29 +01:00

shape.rs

Shape with holes (#770)

2023-09-08 08:38:13 +01:00

storage.rs

Add the powf op. (#664)

2023-08-29 20:48:18 +01:00

strided_index.rs

Broadcasting performance optimization (cpu) (#182)

2023-07-17 13:41:09 +01:00

tensor.rs

Get the comparison operation to work on scalar values. (#780)

2023-09-08 20:13:29 +01:00

test_utils.rs

Move the test-utils bits to a shared place. (#619)

2023-08-27 09:42:22 +01:00

utils.rs

Q6K quantization (#495)

2023-08-17 22:22:57 +01:00

variable.rs

Improve the mnist training example. (#276)

2023-07-29 16:28:22 +01:00