mlx_rs

Macro layer_norm

Source 
macro_rules! layer_norm {
    (x = $x:expr, eps = $eps:expr) => { ... };
    (x = $x:expr, eps = $eps:expr, stream = $stream:expr) => { ... };
    (x = $x:expr, weight = $weight:expr, eps = $eps:expr) => { ... };
    (x = $x:expr, weight = $weight:expr, eps = $eps:expr, stream = $stream:expr) => { ... };
    (x = $x:expr, bias = $bias:expr, eps = $eps:expr) => { ... };
    (x = $x:expr, bias = $bias:expr, eps = $eps:expr, stream = $stream:expr) => { ... };
    (x = $x:expr, weight = $weight:expr, bias = $bias:expr, eps = $eps:expr) => { ... };
    (
        x = $x:expr, weight = $weight:expr, bias = $bias:expr, eps =
        $eps:expr, stream = $stream:expr
    ) => { ... };
    (x = $x:expr, weight = $weight:expr, bias = $bias:expr, eps = $eps:expr) => { ... };
    (
        x = $x:expr, weight = $weight:expr, bias = $bias:expr, eps =
        $eps:expr, stream = $stream:expr
    ) => { ... };
}
Expand description

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