sgE!rddlZddlZddlmZddZGddZdefdZdefd Zdefd Z d Z d Z d Z y)N)validate_map_locationc>t|ttjfr\tjj |st d|dtjj|rt d|dt|}t|ttjfr>tjjtj||}t|Stjj|j|}t|S)au Load a :class:`LiteScriptModule` saved with :func:`torch.jit._save_for_lite_interpreter`. Args: f: a file-like object (has to implement read, readline, tell, and seek), or a string containing a file name map_location: a string or torch.device used to dynamically remap storages to an alternative set of devices. Returns: A :class:`LiteScriptModule` object. Example: .. testcode:: import torch import io # Load LiteScriptModule from saved file path torch.jit._load_for_lite_interpreter('lite_script_module.pt') # Load LiteScriptModule from io.BytesIO object with open('lite_script_module.pt', 'rb') as f: buffer = io.BytesIO(f.read()) # Load all tensors to the original device torch.jit.mobile._load_for_lite_interpreter(buffer) The provided filename  does not exist is a directory) isinstancestrosPathLikepathexists ValueErrorisdirrtorch_C_load_for_lite_interpreterfspath&_load_for_lite_interpreter_from_bufferreadLiteScriptModule)f map_location cpp_modules L/var/www/html/venv/lib/python3.12/site-packages/torch/jit/mobile/__init__.pyrrs<!c2;;'(ww~~a 5aSHI I 77== 5aSHI I(6L!c2;;'(XX881|T J '' XXDD FFHl  J ''c6eZdZfdZdZdZdZdZxZS)rc0||_t| yN)_csuper__init__)selfr __class__s rr!zLiteScriptModule.__init__9s rc8|jj|Srrforwardr"inputs r__call__zLiteScriptModule.__call__=wwu%%rc8|jj|Sr)r find_method)r" method_names rr,zLiteScriptModule.find_method@sww"";//rc8|jj|Srr%r's rr&zLiteScriptModule.forwardCr*rc:|jj||Sr)r run_method)r"r-r(s rr0zLiteScriptModule.run_methodFsww!!+u55r) __name__ __module__ __qualname__r!r)r,r&r0 __classcell__)r#s@rrr8s&0&6rrmodulecTtjj|jS)zkReturn a set of root operator names (with overload name) that are used by any method in this mobile module.)rr_export_operator_listr)r5s rr7r7Js 88 ) )&)) 44rreturnct|ttjfr\tjj |st d|dtjj|rt d|dt|ttjfr2tjjtj|Stjj|jS)a%Take a file-like object to return an integer. Args: f_input: a file-like object (has to implement read, readline, tell, and seek), or a string containing a file name Returns: version: An integer. If the integer is -1, the version is invalid. A warning will show in the log. Example: .. testcode:: from torch.jit.mobile import _get_model_bytecode_version # Get bytecode version from a saved file path version = _get_model_bytecode_version("path/to/model.ptl") rrr)rr r r r r rrrr_get_model_bytecode_versionr'_get_model_bytecode_version_from_bufferrf_inputs rr:r:Os('C-.ww~~g&5gYoNO O 77== !5gYoNO O'C-.xx33BIIg4FGGxx?? OOrct|ttjfr\tjj |st d|dtjj|rt d|dt|ttjfr2tjjtj|Stjj|jS)a6Take a file-like object and return a set of string, like ("int", "Optional"). Args: f_input: a file-like object (has to implement read, readline, tell, and seek), or a string containing a file name Returns: type_list: A set of string, like ("int", "Optional"). These are types used in bytecode. Example: .. testcode:: from torch.jit.mobile import _get_mobile_model_contained_types # Get type list from a saved file path type_list = _get_mobile_model_contained_types("path/to/model.ptl") rrr)rr r r r r rrrr!_get_mobile_model_contained_typesr-_get_mobile_model_contained_types_from_bufferrr<s rr?r?os('C-.ww~~g&5gYoNO O 77== !5gYoNO O'C-.xx99"))G:LMMxxEEgllnUUrcxt|ttjfr\tjj |st d|dtjj|rt d|dt|ttjfrgt|ttjfrGtjjtj|tj||Stjj|jt||S)aTake a input string containing a file name (file-like object) and a new destination to return a boolean. Args: f_input: a file-like object (has to implement read, readline, tell, and seek), or a string containing a file name f_output: path to new model destination to_version: the expected output model bytecode version Returns: success: A boolean. If backport success, return true, otherwise false rrr)rr r r r r rrrr_backport_for_mobiler _backport_for_mobile_from_bufferr)r=f_output to_versions rrBrBs'C-.ww~~g&5gYoNO O 77== !5gYoNO O7S"++./8c2;;/0xx,, IIg  ( 3Z  xx88 LLNCM:  rct|ttjfr\tjj |st d|dtjj|rt d|dt|ttjfr3tjjtj||Stjj|j|S)zTake a string containing a file name (file-like object). Args: f_input: a file-like object (has to implement read, readline, tell, and seek), or a string containing a file name rrr)rr r r r r rrrr_backport_for_mobile_to_bufferr*_backport_for_mobile_from_buffer_to_bufferr)r=rEs rrGrGs'C-.ww~~g&5gYoNO O 77== !5gYoNO O'C-.xx66ryy7I:VVxxBB LLNJ  rct|ttjfr\tjj |st d|dtjj|rt d|dt|ttjfr2tjjtj|Stjj|jS)a{Retrieve the root (top level) operators of a model and their corresponding compatibility info. These root operators can call other operators within them (traced ops), and a root op can call many different traced ops depending on internal code paths in the root op. These traced ops are not returned by this function. Those operators are abstracted into the runtime as an implementation detail (and the traced ops themselves can also call other operators) making retrieving them difficult and their value from this api negligible since they will differ between which runtime version the model is run on. Because of this, there is a false positive this api can't prevent in a compatibility usecase. All the root ops of a model are present in a target runtime, but not all the traced ops are which prevents a model from being able to run. Args: f_input: a file-like object (has to implement read, readline, tell, and seek), or a string containing a file name Returns: Operators and info: A Dictionary mapping strings (the qualified names of the root operators) of the model to their OperatorInfo structs. Example: .. testcode:: from torch.jit.mobile import _get_model_ops_and_info # Get bytecode version from a saved file path ops_and_info = _get_model_ops_and_info("path/to/model.ptl") rrr) rr r r r r rrrr_get_model_ops_and_inforrr<s rrJrJs:'C-.ww~~g&5gYoNO O 77== !5gYoNO O'C-.xx// '0BCCxx// ??rr) r rtorch.jit._serializationrrrr7intr:r?rBrGrJrrrNsY :-(`66$5"25 PCP@V#V@ : ,&@r