onnxruntime

1
// Copyright (c) Microsoft Corporation. All rights reserved.
2
// Licensed under the MIT License.
3

4
#import "ort_xnnpack_execution_provider.h"
5

6
#import "cxx_api.h"
7
#import "error_utils.h"
8
#import "ort_session_internal.h"
9

10
NS_ASSUME_NONNULL_BEGIN
11

12
@implementation ORTXnnpackExecutionProviderOptions
13

14
@end
15

16
@implementation ORTSessionOptions (ORTSessionOptionsXnnpackEP)
17

18
- (BOOL)appendXnnpackExecutionProviderWithOptions:(ORTXnnpackExecutionProviderOptions*)options
19
                                            error:(NSError**)error {
20
  try {
21
    NSDictionary* provider_options = @{
22
      @"intra_op_num_threads" : [NSString stringWithFormat:@"%d", options.intra_op_num_threads]
23
    };
24
    return [self appendExecutionProvider:@"XNNPACK" providerOptions:provider_options error:error];
25
  }
26
  ORT_OBJC_API_IMPL_CATCH_RETURNING_BOOL(error);
27
}
28

29
@end
30

31
NS_ASSUME_NONNULL_END
32