burn/crates/burn-vision
Genna Wingert 163a2069c6
Optimize connected components (#2810)
* Add boolean and/or to bool tensors and fix version mismatch for rand

* Add tests

* Fix tch

* Fix for cubecl update

* Add missing test attribute

* Optimize NdArray into_data and other ways to make connected components faster

* Use existing into_data implementation

* Make lint `allow` because it's buggy

* `bool_into_data` should have a dtype of `BoolElem`

* Don't convert to bool in place

* Don't assert type equality for bool tests

* Fix nn/mask tests

* Fix and/or tests

* Add `to_bool` and fix `Bool::Elem`

* Fix display for bool

* Fix hardcoded `DType::Bool` uses
2025-02-17 09:51:38 -05:00
..
src Optimize connected components (#2810) 2025-02-17 09:51:38 -05:00
tests [Feat] Add new burn-vision crate with one initial op (#2753) 2025-02-11 14:24:26 -05:00
Cargo.toml Optimize connected components (#2810) 2025-02-17 09:51:38 -05:00