Struct MlxCompiledMultiOutputGraph 

pub struct MlxCompiledMultiOutputGraph { /* private fields */ }

Owns one compiled MLX graph that produces an ordered vector of outputs.

Model-side compiled composites that need several outputs at once — the multi-token-prediction verification window returns successor states beside its logits — construct one of these with their own graph builder and apply it through MlxBindingsContext::apply_compiled_multi_output_graph. Output order is the builder’s write order and never changes after compilation.

Implementations

impl MlxCompiledMultiOutputGraph

pub fn new( graph_builder: MlxGraphBuilder, compile_operation: &'static str, shapeless: bool, ) -> Result<Self, MlxCError>

Compiles the given graph builder into a replayable multi-output graph. The shapeless flag carries the same shape-specialization tradeoff as the single-output compiled graph constructor.

Trait Implementations

impl Debug for MlxCompiledMultiOutputGraph

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.