mlx_rs::transforms::compile

Function compile_with_state

Source
pub fn compile_with_state<F, U, A, O, E>(
    f: F,
    shapeless: impl Into<Option<bool>>,
) -> impl for<'a> FnMut(&mut U, F::Args<'a>) -> Result<O, Exception>
where F: CompileWithState<U, A, O, E> + Copy + 'static, U: Updatable,
Expand description

Similar to crate::transforms::compile but allows for functions that take a mutable reference to a state.