diff --git a/extension/apple/ExecuTorch/Exported/ExecuTorchModule.h b/extension/apple/ExecuTorch/Exported/ExecuTorchModule.h index 3e6cd55a165..51d99780da3 100644 --- a/extension/apple/ExecuTorch/Exported/ExecuTorchModule.h +++ b/extension/apple/ExecuTorch/Exported/ExecuTorchModule.h @@ -134,10 +134,21 @@ __attribute__((deprecated("This API is experimental."))) * @return An initialized ExecuTorchModule instance. */ - (instancetype)initWithFilePath:(NSString *)filePath - dataFilePath:(NSString *)dataPath + dataFilePath:(NSString *)dataFilePath 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 + * external tensors and external data. + * @return An initialized ExecuTorchModule instance. + */ +- (instancetype)initWithFilePath:(NSString *)filePath + dataFilePath:(NSString *)dataFilePath; + /** * Initializes a module with a file path and a specified load mode. * diff --git a/extension/apple/ExecuTorch/Exported/ExecuTorchModule.mm b/extension/apple/ExecuTorch/Exported/ExecuTorchModule.mm index e1dea859fb7..ce58f2fb21a 100644 --- a/extension/apple/ExecuTorch/Exported/ExecuTorchModule.mm +++ b/extension/apple/ExecuTorch/Exported/ExecuTorchModule.mm @@ -265,6 +265,13 @@ - (instancetype)initWithFilePath:(NSString *)filePath return self; } +- (instancetype)initWithFilePath:(NSString *)filePath + dataFilePath:(NSString *)dataFilePath { + return [self initWithFilePath:filePath + dataFilePath:dataFilePath + loadMode:ExecuTorchModuleLoadModeFile]; +} + - (instancetype)initWithFilePath:(NSString *)filePath loadMode:(ExecuTorchModuleLoadMode)loadMode { return [self initWithFilePath:filePath