2021-05-11 17:34:40 +00:00
|
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
|
|
|
// Licensed under the MIT License.
|
|
|
|
|
|
|
|
|
|
#ifndef OnnxruntimeModule_h
|
|
|
|
|
#define OnnxruntimeModule_h
|
|
|
|
|
|
|
|
|
|
#import <React/RCTBridgeModule.h>
|
2023-06-16 09:37:02 +00:00
|
|
|
#import <React/RCTBlobManager.h>
|
2021-05-11 17:34:40 +00:00
|
|
|
|
2024-08-14 23:51:22 +00:00
|
|
|
@interface OnnxruntimeModule : NSObject <RCTBridgeModule>
|
2021-05-11 17:34:40 +00:00
|
|
|
|
2024-08-14 23:51:22 +00:00
|
|
|
- (void)setBlobManager:(RCTBlobManager*)manager;
|
2023-06-16 09:37:02 +00:00
|
|
|
|
2024-08-14 23:51:22 +00:00
|
|
|
- (NSDictionary*)loadModel:(NSString*)modelPath
|
|
|
|
|
options:(NSDictionary*)options;
|
2021-05-11 17:34:40 +00:00
|
|
|
|
2024-08-14 23:51:22 +00:00
|
|
|
- (NSDictionary*)loadModelFromBuffer:(NSData*)modelData
|
|
|
|
|
options:(NSDictionary*)options;
|
2023-04-29 00:34:26 +00:00
|
|
|
|
2024-08-14 23:51:22 +00:00
|
|
|
- (void)dispose:(NSString*)key;
|
2023-06-08 23:17:33 +00:00
|
|
|
|
2024-08-14 23:51:22 +00:00
|
|
|
- (NSDictionary*)run:(NSString*)url
|
|
|
|
|
input:(NSDictionary*)input
|
|
|
|
|
output:(NSArray*)output
|
|
|
|
|
options:(NSDictionary*)options;
|
2021-05-11 17:34:40 +00:00
|
|
|
|
|
|
|
|
@end
|
|
|
|
|
|
|
|
|
|
#endif /* OnnxruntimeModule_h */
|