Thomas Santerre 9563a5fee4 Add support for conv_transpose2d on Metal backend (#1903) ...

* add support for conv transpose 2d and add bench mark for float types

* update bench calculation

* enable testing all conv operations on metal

2024-03-21 18:08:45 +01:00

..

affine.rs

Metal: Activate bfloat affine and add benchmark (#1543)

2024-01-12 11:19:49 +01:00

conv_transpose2d.rs

Add support for conv_transpose2d on Metal backend (#1903)

2024-03-21 18:08:45 +01:00

matmul.rs

Seperate benchmarks by enabled features (#1538)

2024-01-11 15:35:38 +01:00

mod.rs

Add support for conv_transpose2d on Metal backend (#1903)

2024-03-21 18:08:45 +01:00

random.rs

Merge branch 'main' into ivarflakstad/metal-prng

2024-01-12 07:19:58 +01:00

where_cond.rs

Metal: f16 and bf16 where_cond + benchmark (#1545)

2024-01-12 11:18:11 +01:00