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 执行提供程序是否仅在配备苹果神经网络引擎 (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 }