Struct CoreMLModel

pub struct CoreMLModel { /* private fields */ }

CoreML model wrapper that provides Candle tensor integration

Implementations

impl CoreMLModel

pub fn load<P: AsRef<Path>>(path: P) -> Result<Self, CandleError>

Load a CoreML model from a .mlmodelc directory with default configuration

pub fn load_from_file<P: AsRef<Path>>( path: P, config: &Config, ) -> Result<Self, CandleError>

Load a CoreML model from a .mlmodelc directory following standard Candle patterns

Note: Unlike other Candle models, CoreML models are pre-compiled and don’t use VarBuilder. This method provides a Candle-compatible interface while loading from CoreML files.

pub fn forward_single(&self, input: &Tensor) -> Result<Tensor, CandleError>

Run forward pass through the model with multiple inputs

Accepts tensors from CPU or Metal devices, rejects CUDA tensors. Returns output tensor on the same device as the input tensors.

Arguments

• inputs - Slice of tensors corresponding to the input_names in config order

Convenience method for single-input models (backward compatibility)

pub fn forward(&self, inputs: &[&Tensor]) -> Result<Tensor, CandleError>

pub fn config(&self) -> &Config

Get the model configuration

Trait Implementations

impl Debug for CoreMLModel

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

Formats the value using the given formatter.