Skip to main content

graph_output_array

Function graph_output_array 

Source
pub fn graph_output_array(
    build_graph: impl FnOnce(*mut mlx_array) -> c_int,
) -> Result<MlxArray, c_int>
Expand description

Runs one raw MLX operation inside a graph builder and owns its lazy output.