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。

    声明

    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。默认情况下，它将创建神经网络模型。需要 Core ML 5 或更高版本。

    声明

    Objective-C

    @property BOOL createMLProgram;

    Swift

    var createMLProgram: Bool { get set }