Skip to main content

mlx_ops/
lib.rs

1//! Op registry, shape inference, broadcasting rules, and dtype promotion.
2
3pub mod broadcast;
4pub mod dtype_promotion;
5pub mod shape_inference;
6
7pub use broadcast::broadcast_shapes;
8pub use dtype_promotion::promote;
9pub use shape_inference::infer_shape;