mlx_rs

Macro topk

Source 
macro_rules! topk {
    ($a:expr, $k:expr) => { ... };
    ($a:expr, $k:expr, stream = $stream:expr) => { ... };
    ($a:expr, $k:expr, axis = $axis:expr) => { ... };
    ($a:expr, $k:expr, axis = $axis:expr, stream = $stream:expr) => { ... };
}
Expand description

Macro generated for the function crate::ops::indexing::topk. See the function documentation for more details.