ij<`dZddlZddlZddlmZddlmZddlmZddlm Z  d dZ dZ d Z y) z)PyTorch ExportedProgram export utilities.N)treeconvert_spec_to_tensor)get_input_signature)io_utilsc  ddl}t|}|jdstd|t |}|j d}| t |}|dnd tj fd|}t|}|t|||d<tj5tjd d  |jj||fi|} ddd|jj# |||nd } | rt%j&d|d|S#t$r} t!d | | d} ~ wwxYw#1swYhxYw)abExport the model as a PyTorch ExportedProgram (`.pt2`) artifact. Uses `torch.export.export` to capture the model's computation graph in an Ahead-of-Time (AOT) fashion. The resulting artifact contains only ATen-level operations and can be loaded via `torch.export.load`, run with `ExportedProgram.module()`, or compiled further with `torch.compile`. Args: model: The Keras model to export. filepath: `str` or `pathlib.Path` object. Path to save the exported artifact. Must end with `.pt2`. input_signature: Optional input signature. If `None`, inferred from the model's built input spec. verbose: `bool`. Whether to print a message after export. Defaults to `True`. **kwargs: Additional keyword arguments forwarded to `torch.export.export`, including `strict`, `dynamic_shapes`, `prefer_deferred_runtime_asserts_over_guards`, and `preserve_module_call_signature`. For a single-input model, `dynamic_shapes` can be a dict such as `{0: torch.export.Dim("batch", min=1, max=128)}`. Note: when `dynamic_shapes` is provided, the sample input used for tracing uses a batch size of 2 instead of 1. This avoids a PyTorch limitation where dimensions of size 1 are specialized to constants during export. Example: ```python model.export("path/to/model.pt2", format="torch") import torch loaded_program = torch.export.load("path/to/model.pt2") output = loaded_program.module()(torch.randn(1, 10)) ``` rNz.pt2zBThe PyTorch export requires the filepath to end with '.pt2'. Got: dynamic_shapesct|S)N)replace_none_numberr)xr s k/var/www/html/emotional.easysim.app/public_html/venv/lib/python3.12/site-packages/keras/src/export/torch.pyzexport_torch..Ns( #6 ignorez*.*not properly registered as a submodule.*)messagezFailed to export model to PyTorch format. Common causes: unsupported operations, data-dependent control flow, or shape constraints not satisfied. Original error: Tz"Saved PyTorch ExportedProgram at 'z'.)torchstrendswith ValueError_get_export_kwargsgetrr map_structuretuple_normalize_dynamic_shapeswarningscatch_warningsfilterwarningsexport Exception RuntimeErrorsaver print_msg) modelfilepathinput_signatureverbosekwargsr export_kwargsr sample_inputsexported_programeactual_verboser s @r export_torchr/ sX8}H   V $ $: '  'v.M"&&'78N-e4 .9!q&&   M -(M!*C M+ &'  " A  2u||22     & LL&1 ' 3WN?zLM O #$%#'    s*(ED'' E0D??EEEcddl}tj|jjjDchc]}|dvr| }}t t |j|}|r=djt |}dj|}td|d|d|jDcic] \}}||vr||}}}|jdd|Scc}wcc}}w) Nr>modargsr)z, z,Unsupported arguments for `format="torch"`: z. Supported arguments are: .strictF) rinspect signaturer parameterssortedset differencejoinritems setdefault) user_kwargsrnamesupported_arg_namesunknown_arg_namessupported_args unknown_argsvaluer*s rrrts %%ell&9&9:EE  0 0  s;/::;NOP6*=#>?yy!23  :n77Gq J  ',,. D% & & e MXu- /s C&>C+c&t|trd|vsd|vr|St|dk(r|fgS|St|ttfrLt|dk(rt|dttfr|St|t|k(r t |gS|S)Nr2r)r r) isinstancedictlenlistr)r r+s rrrs.$' ^ #x>'A! ! }  "#%& &.4-0 ~ ! # 1 e}) " ! ~ #m"4 4.)* * r)NN) __doc__r5r keras.srcrkeras.src.export.export_utilsrrkeras.src.utilsrr/rrrrrOs3/@=$   eP:r