ORTCoreMLExecutionProviderOptions

Objective-C

@interface ORTCoreMLExecutionProviderOptions : NSObject

Swift

class ORTCoreMLExecutionProviderOptions : NSObject

用于配置 CoreML 执行提供程序的选项。

  • CoreML 执行提供程序是否仅在 CPU 上运行。

    声明

    Objective-C

    @property BOOL useCPUOnly;

    Swift

    var useCPUOnly: Bool { get set }

  • 在 CoreML 中排除 ANE（Apple 神经网络引擎）。

    声明

    Objective-C

    @property BOOL useCPUAndGPU;

    Swift

    var useCPUAndGPU: Bool { get set }

  • 是否在子图上启用 CoreML 执行提供程序。

    声明

    Objective-C

    @property BOOL enableOnSubgraphs;

    Swift

    var enableOnSubgraphs: Bool { get set }

  • CoreML 执行提供程序是否仅在具备 Apple 神经网络引擎 (ANE) 的设备上启用。

    声明

    Objective-C

    @property BOOL onlyEnableForDevicesWithANE;

    Swift

    var onlyEnableForDevicesWithANE: Bool { get set }

  • 仅允许 CoreML EP 处理输入形状为静态的节点。默认情况下，它也允许输入形状为动态的情况。但是，如果输入具有动态形状，性能可能会受到负面影响。

    声明

    Objective-C

    @property BOOL onlyAllowStaticInputShapes;

    Swift

    var onlyAllowStaticInputShapes: Bool { get set }

  • 创建 MLProgram。默认情况下，它将创建 NeuralNetwork 模型。需要 Core ML 5 或更高版本。

    声明

    Objective-C

    @property BOOL createMLProgram;

    Swift

    var createMLProgram: Bool { get set }