mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
f4cd35f9b1
### Description <!-- Describe your changes. --> ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> Co-authored-by: Edward Chen <18449977+edgchen1@users.noreply.github.com>
39 lines
1.1 KiB
Objective-C
39 lines
1.1 KiB
Objective-C
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#import <Foundation/Foundation.h>
|
|
|
|
#import "ort_session.h"
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
/**
|
|
* Options for configuring the Xnnpack execution provider.
|
|
*/
|
|
@interface ORTXnnpackExecutionProviderOptions : NSObject
|
|
|
|
/**
|
|
* How many threads used for the Xnnpack execution provider.
|
|
*/
|
|
@property int intra_op_num_threads;
|
|
|
|
@end
|
|
|
|
@interface ORTSessionOptions (ORTSessionOptionsXnnpackEP)
|
|
|
|
/**
|
|
* Available since 1.14.
|
|
* Enables the Xnnpack execution provider in the session configuration options.
|
|
* It is appended to the execution provider list which is ordered by
|
|
* decreasing priority.
|
|
*
|
|
* @param options The Xnnpack execution provider configuration options.
|
|
* @param error Optional error information set if an error occurs.
|
|
* @return Whether the provider was enabled successfully.
|
|
*/
|
|
- (BOOL)appendXnnpackExecutionProviderWithOptions:(ORTXnnpackExecutionProviderOptions*)options
|
|
error:(NSError**)error;
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|