Skip to content

Commit b660c2e

Browse files
authored
Add initializer with file and data file paths (#13830)
1 parent 1f18326 commit b660c2e

File tree

2 files changed

+19
-1
lines changed

2 files changed

+19
-1
lines changed

extension/apple/ExecuTorch/Exported/ExecuTorchModule.h

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,10 +134,21 @@ __attribute__((deprecated("This API is experimental.")))
134134
* @return An initialized ExecuTorchModule instance.
135135
*/
136136
- (instancetype)initWithFilePath:(NSString *)filePath
137-
dataFilePath:(NSString *)dataPath
137+
dataFilePath:(NSString *)dataFilePath
138138
loadMode:(ExecuTorchModuleLoadMode)loadMode
139139
NS_DESIGNATED_INITIALIZER;
140140

141+
/**
142+
* Initializes a module with a file path, data path and a specified load mode.
143+
*
144+
* @param filePath A string representing the path to the ExecuTorch program file.
145+
* @param dataFilePath A string representing the path to a .ptd file with
146+
* external tensors and external data.
147+
* @return An initialized ExecuTorchModule instance.
148+
*/
149+
- (instancetype)initWithFilePath:(NSString *)filePath
150+
dataFilePath:(NSString *)dataFilePath;
151+
141152
/**
142153
* Initializes a module with a file path and a specified load mode.
143154
*

extension/apple/ExecuTorch/Exported/ExecuTorchModule.mm

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -265,6 +265,13 @@ - (instancetype)initWithFilePath:(NSString *)filePath
265265
return self;
266266
}
267267

268+
- (instancetype)initWithFilePath:(NSString *)filePath
269+
dataFilePath:(NSString *)dataFilePath {
270+
return [self initWithFilePath:filePath
271+
dataFilePath:dataFilePath
272+
loadMode:ExecuTorchModuleLoadModeFile];
273+
}
274+
268275
- (instancetype)initWithFilePath:(NSString *)filePath
269276
loadMode:(ExecuTorchModuleLoadMode)loadMode {
270277
return [self initWithFilePath:filePath

0 commit comments

Comments
 (0)