Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions extension/apple/ExecuTorch/Exported/ExecuTorchModule.h
Original file line number Diff line number Diff line change
Expand Up @@ -126,27 +126,27 @@ NS_SWIFT_NAME(Module)
* Initializes a module with a file path, data path and a specified load mode.
*
* @param filePath A string representing the path to the ExecuTorch program file.
* @param dataFilePath A string representing the path to a .ptd file with
* @param dataFiles A list of strings representing paths to .ptd files with
* external tensors and external data.
* @param loadMode A value from ExecuTorchModuleLoadMode that determines the
* file loading behavior.
* @return An initialized ExecuTorchModule instance.
*/
- (instancetype)initWithFilePath:(NSString *)filePath
dataFilePath:(NSString *)dataFilePath
dataFiles:(NSArray<NSString *> *)dataFiles
loadMode:(ExecuTorchModuleLoadMode)loadMode
NS_DESIGNATED_INITIALIZER;

/**
* Initializes a module with a file path, data path and a specified load mode.
*
* @param filePath A string representing the path to the ExecuTorch program file.
* @param dataFilePath A string representing the path to a .ptd file with
* @param dataFiles A list of strings representing paths to .ptd files with
* external tensors and external data.
* @return An initialized ExecuTorchModule instance.
*/
- (instancetype)initWithFilePath:(NSString *)filePath
dataFilePath:(NSString *)dataFilePath;
dataFiles:(NSArray<NSString *> *)dataFiles;

/**
* Initializes a module with a file path and a specified load mode.
Expand Down
19 changes: 13 additions & 6 deletions extension/apple/ExecuTorch/Exported/ExecuTorchModule.mm
Original file line number Diff line number Diff line change
Expand Up @@ -250,13 +250,20 @@ @implementation ExecuTorchModule {
}

- (instancetype)initWithFilePath:(NSString *)filePath
dataFilePath:(NSString *)dataFilePath
dataFiles:(NSArray<NSString *> *)dataFiles
loadMode:(ExecuTorchModuleLoadMode)loadMode {
self = [super init];
if (self) {
// Convert NSArray<NSString *> to std::vector<std::string>
std::vector<std::string> dataFilesVector;
if (dataFiles != nil) {
for (NSString *dataFile in dataFiles) {
dataFilesVector.push_back(dataFile.UTF8String);
}
}
_module = std::make_unique<Module>(
filePath.UTF8String,
dataFilePath.UTF8String,
dataFilesVector,
static_cast<Module::LoadMode>(loadMode)
);
_inputs = [NSMutableDictionary new];
Expand All @@ -266,21 +273,21 @@ - (instancetype)initWithFilePath:(NSString *)filePath
}

- (instancetype)initWithFilePath:(NSString *)filePath
dataFilePath:(NSString *)dataFilePath {
dataFiles:(NSArray<NSString *> *)dataFiles {
return [self initWithFilePath:filePath
dataFilePath:dataFilePath
dataFiles:dataFiles
loadMode:ExecuTorchModuleLoadModeFile];
}

- (instancetype)initWithFilePath:(NSString *)filePath
loadMode:(ExecuTorchModuleLoadMode)loadMode {
return [self initWithFilePath:filePath
dataFilePath:@""
dataFiles:@[]
loadMode:loadMode];
}
- (instancetype)initWithFilePath:(NSString *)filePath {
return [self initWithFilePath:filePath
dataFilePath:@""
dataFiles:@[]
loadMode:ExecuTorchModuleLoadModeFile];
}

Expand Down
Loading