Laurent Mazare 11c7e7bd67 Some fixes for yolo-v3. (#529) ...

* Some fixes for yolo-v3.

* Use the running stats for inference in the batch-norm layer.

* Get some proper predictions for yolo.

* Avoid the quadratic insertion.

2023-08-20 23:19:15 +01:00

..

cpu

Quantized support for f16 and f32 (#457)

2023-08-15 21:09:37 +01:00

quantized

Neon support for quantization. (#519)

2023-08-19 22:07:29 +01:00

accelerate.rs

Add a yolo-v3 example. (#528)

2023-08-20 18:19:37 +01:00

backend.rs

add max_pool2d (#371)

2023-08-09 18:05:26 +01:00

backprop.rs

Add a yolo-v3 example. (#528)

2023-08-20 18:19:37 +01:00

conv.rs

Add a conv1d benchmark based on the whisper sizes. (#377)

2023-08-09 20:27:03 +01:00

convert.rs

Expose the tensor write-bytes function. (#412)

2023-08-11 17:13:42 +01:00

cpu_backend.rs

Simd support (#448)

2023-08-15 09:50:38 +01:00

cuda_backend.rs

Cudnn support (#445)

2023-08-14 21:30:41 +01:00

cudnn.rs

Cudnn support (#445)

2023-08-14 21:30:41 +01:00

device.rs

Transpose the weight matrixes for llama2.c. (#321)

2023-08-04 13:32:20 +01:00

display.rs

Remove some dead-code pragmas. (#137)

2023-07-11 09:33:59 +01:00

dtype.rs

Rename vec-dot to vec-ops. (#449)

2023-08-15 10:48:57 +01:00

dummy_cuda_backend.rs

add max_pool2d (#371)

2023-08-09 18:05:26 +01:00

error.rs

Get the ggml based llama to generate some text. (#464)

2023-08-16 12:41:07 +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

Preliminary support for importing PyTorch weights. (#511)

2023-08-19 11:26:32 +01:00

mkl.rs

Add a yolo-v3 example. (#528)

2023-08-20 18:19:37 +01:00

npy.rs

Retrieve tensor data from PyTorch files. (#516)

2023-08-19 15:57:18 +01:00

op.rs

Add a yolo-v3 example. (#528)

2023-08-20 18:19:37 +01:00

pickle.rs

Small tweaks to tensor-tools. (#517)

2023-08-19 16:50:26 +01:00

safetensors.rs

fix: can directly save the loaded weights (#421)

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

shape.rs

Add a broadcast variant to matmul. (#523)

2023-08-20 13:20:42 +01:00

storage.rs

Relax the requirements on CustomOp. (#486)

2023-08-17 11:12:05 +01:00

strided_index.rs

Broadcasting performance optimization (cpu) (#182)

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

tensor.rs

Some fixes for yolo-v3. (#529)

2023-08-20 23:19:15 +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