mlx_rs

Macro scaled_dot_product_attention

Source 
macro_rules! scaled_dot_product_attention {
    ($queries:expr, $keys:expr, $values:expr, $scale:expr) => { ... };
    ($queries:expr, $keys:expr, $values:expr, $scale:expr, stream = $stream:expr) => { ... };
    ($queries:expr, $keys:expr, $values:expr, $scale:expr, mask = $mask:expr) => { ... };
    (
        $queries:expr, $keys:expr, $values:expr, $scale:expr, mask =
        $mask:expr, stream = $stream:expr
    ) => { ... };
}
Expand description

Macro generated for the function crate::fast::scaled_dot_product_attention. See the function documentation for more details.