1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#ifndef OnnxruntimeModule_h
#define OnnxruntimeModule_h
#import <React/RCTBridgeModule.h>
#import <React/RCTBlobManager.h>
@interface OnnxruntimeModule : NSObject <RCTBridgeModule>
- (void)setBlobManager:(RCTBlobManager*)manager;
- (NSDictionary*)loadModel:(NSString*)modelPath
options:(NSDictionary*)options;
- (NSDictionary*)loadModelFromBuffer:(NSData*)modelData
options:(NSDictionary*)options;
- (void)dispose:(NSString*)key;
- (NSDictionary*)run:(NSString*)url
input:(NSDictionary*)input
output:(NSArray*)output
options:(NSDictionary*)options;
@end
#endif /* OnnxruntimeModule_h */
|
