mlx_rs::transforms::compile

Trait CallMut

pub trait CallMut<A, O, E> {
    // Required method
    fn call_mut(&mut self, args: A) -> Result<O, Exception>;
}

Expand description

A trait for a compiled function that can be called.

Required Methods§

fn call_mut(&mut self, args: A) -> Result<O, Exception>

Calls the compiled function with the given arguments.

Implementors§

impl<'a, F, G> CallMut<&'a Array, Array, ()> for Compiled<F, G>
where F: FnMut(&Array) -> Array + 'a, G: FnMut(&[Array]) -> Vec<Array> + 'a,

impl<'a, F, G> CallMut<&'a Array, Array, Exception> for Compiled<F, G>
where F: FnMut(&Array) -> Result<Array, Exception> + 'a, G: FnMut(&[Array]) -> Result<Vec<Array>, Exception> + 'a,

impl<'a, F, G> CallMut<&'a [Array], Vec<Array>, ()> for Compiled<F, G>
where F: FnMut(&[Array]) -> Vec<Array> + 'a, G: FnMut(&[Array]) -> Vec<Array> + 'a,

impl<'a, F, G> CallMut<&'a [Array], Vec<Array>, Exception> for Compiled<F, G>
where F: FnMut(&[Array]) -> Result<Vec<Array>, Exception> + 'a, G: FnMut(&[Array]) -> Result<Vec<Array>, Exception> + 'a,

impl<'a, F, G> CallMut<(&'a Array, &'a Array), Array, ()> for Compiled<F, G>
where F: FnMut((&Array, &Array)) -> Array + 'a, G: FnMut(&[Array]) -> Vec<Array> + 'a,

impl<'a, F, G> CallMut<(&'a Array, &'a Array), Array, Exception> for Compiled<F, G>
where F: FnMut((&Array, &Array)) -> Result<Array, Exception> + 'a, G: FnMut(&[Array]) -> Result<Vec<Array>, Exception> + 'a,

impl<'a, F, G> CallMut<(&'a Array, &'a Array, &'a Array), Array, ()> for Compiled<F, G>
where F: FnMut((&Array, &Array, &Array)) -> Array + 'a, G: FnMut(&[Array]) -> Vec<Array> + 'a,

impl<'a, F, G> CallMut<(&'a Array, &'a Array, &'a Array), Array, Exception> for Compiled<F, G>
where F: FnMut((&Array, &Array, &Array)) -> Result<Array, Exception> + 'a, G: FnMut(&[Array]) -> Result<Vec<Array>, Exception> + 'a,