(bencher: divan::Bencher, size: usize)
| 91 | |
| 92 | #[divan::bench(args = [256, 1024, 4096])] |
| 93 | fn sigmoid(bencher: divan::Bencher, size: usize) { |
| 94 | let b = backend(); |
| 95 | let x = Tensor::randn(0f32, 1.0, (1, size), &Device::Cpu).unwrap(); |
| 96 | bencher.bench_local(|| { |
| 97 | let _ = b.sigmoid(&x).unwrap(); |
| 98 | }); |
| 99 | } |
| 100 | |
| 101 | // ── embedding ────────────────────────────────────────────────────────── |
| 102 |