burn/crates
Adrian Müller 6b51b73a5f
Fix ONNX where op for scalar inputs (#2218)
* Fix ONNX where op dim_inference for scalar inputs

* Rewrite ONNX Where codegen to support scalars

* ONNX Where: Add tests for all_scalar inputs

---------

Co-authored-by: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>
2024-09-03 11:17:18 -04:00
..
burn Bump burn version to 0.15.0 2024-08-27 15:13:40 -04:00
burn-autodiff Bump burn version to 0.15.0 2024-08-27 15:13:40 -04:00
burn-candle Bump burn version to 0.15.0 2024-08-27 15:13:40 -04:00
burn-common Make compatible with thumbv6m-none-eabi + add raspberry pi pico example (#2096) 2024-08-23 07:39:39 -04:00
burn-core Remove copy restriction for const generic modules (#2222) 2024-09-03 09:39:12 -04:00
burn-cuda Bump burn version to 0.15.0 2024-08-27 15:13:40 -04:00
burn-dataset Fix compile for dataset crate with vision feature (#2228) 2024-09-01 17:03:37 -04:00
burn-derive Fix module derive with generics (#2127) 2024-08-08 16:24:51 -04:00
burn-fusion Bump burn version to 0.15.0 2024-08-27 15:13:40 -04:00
burn-import Fix ONNX where op for scalar inputs (#2218) 2024-09-03 11:17:18 -04:00
burn-jit Avoid 0 denominator in interpolate frac (#2224) 2024-09-01 16:37:32 -04:00
burn-ndarray Avoid 0 denominator in interpolate frac (#2224) 2024-09-01 16:37:32 -04:00
burn-no-std-tests Refactor xtask to use tracel-xtask and refactor CI workflow (#2063) 2024-08-28 15:57:13 -04:00
burn-tch Nonzero should return an empty vec for zero tensors (#2212) 2024-09-03 09:00:58 -04:00
burn-tensor Nonzero should return an empty vec for zero tensors (#2212) 2024-09-03 09:00:58 -04:00
burn-tensor-testgen Fix Cargo.toml repository links (#1749) 2024-05-09 15:40:05 -04:00
burn-train Don't panic when the progress is > 1.0 (#2229) 2024-09-01 16:33:25 -04:00
burn-wgpu Bump burn version to 0.15.0 2024-08-27 15:13:40 -04:00
onnx-ir Fix ONNX where op for scalar inputs (#2218) 2024-09-03 11:17:18 -04:00