A2va 21055b5697 Add PRelu operation (#2904) ...

* Add PRelu operation

* Apply rustfmt.

---------

Co-authored-by: Laurent <laurent.mazare@gmail.com>

2025-04-19 07:24:10 +02:00

..

activation.rs

Add PRelu operation (#2904)

2025-04-19 07:24:10 +02:00

batch_norm.rs

Detach the tensors on batch-norm eval. (#1702)

2024-02-13 14:26:32 +01:00

conv.rs

Expose the cudnn algo in the conv ops. (#2892)

2025-04-14 08:25:32 +02:00

embedding.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

encoding.rs

Fix clippy lints for 1.76. (#1682)

2024-02-08 16:48:47 +01:00

func.rs

Lint fixes introduced with Rust 1.83 (#2646)

2024-11-28 23:00:21 +01:00

group_norm.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

init.rs

VarBuilder cleanup (#627)

2023-08-27 18:03:26 +01:00

kv_cache.rs

Rotating kv cache positions (#2901)

2025-04-15 23:09:26 +02:00

layer_norm.rs

ModernBERT model (#2713)

2025-01-13 08:39:27 +01:00

lib.rs

Gumbel-Softmax sampling. (#2894)

2025-04-14 15:42:42 +02:00

linear.rs

Avoid using batched-matmul in nn::Linear. (#2883)

2025-04-12 19:53:58 +02:00

loss.rs

Fix for clippy 1.86. (#2864)

2025-04-03 19:38:27 +02:00

ops.rs

Cuda cleanup. (#2880)

2025-04-11 21:43:35 +02:00

optim.rs

Expose AdamW parameters (#1449)

2023-12-16 18:41:56 -06:00

rnn.rs

Make the RNN configs accessible from the models. (#2541)

2024-10-04 14:22:23 +02:00

rotary_emb.rs

Cuda cleanup. (#2880)

2025-04-11 21:43:35 +02:00

sampling.rs

Gumbel-Softmax sampling. (#2894)

2025-04-14 15:42:42 +02:00

sequential.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00

var_builder.rs

Clippy fixes for 1.84. (#2710)

2025-01-10 10:15:15 +01:00

var_map.rs

Update docs (#2553)

2024-11-11 22:13:52 +01:00