Si/1SSKJr SSKrSSKrSSKJr SSKJr SSKr SSK r SSK r \(aSSK J r JrJrJrJr "SS5rS S SjjrS S jrSS jrg)) annotationsN)deque) TYPE_CHECKING) FunctionProto ModelProto NodeProto TensorProtoValueInfoProtoc\rSrSrSSjr\SSj5r\SSj5rSSjrSSjr SSjr SSjr SS jr SS jr SS jrS rg ) Extractorc&XlURRUlURURR5UlURURR 5UlUR RURURR55 UR RURURR55 URUR5Ul gN) modelgraph_build_name2obj_dict initializer initializers value_info value_infosupdateinputoutput_build_output_dictoutmap)selfrs I/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/utils.py__init__Extractor.__init__s ZZ%% 484M4M JJ " "5 7;6O6O JJ ! !7   9 9$**:J:J KL  9 9$**:K:K LM&*&=&=djj&I cFUVs0sHoRU_M sn$s snfr)name)objsobjs rrExtractor._build_name2obj_dict(s )-.## ...sc0n[UR5H+up#URHnUS:XaM XA;deX!U'M M- U$)N) enumeratenoder)routput_to_indexindexr) output_names rrExtractor._build_output_dict,sQ*,$UZZ0KE#{{ "$"999/4 , +1 r cUVs/sHo"UR;dMUPM nnU(a[SSRU535eUVs/sHo RUPM sn$s snfs snf)Nz3The following names were not found in value_infos: z, )r ValueErrorjoin)rio_names_to_extractr" missing_namess r_collect_new_ioExtractor._collect_new_io7s1 0T@P@P4PD0   EdiiP]F^E_` 4GG3F4  &3FGG HsA'A' A,cJU/nU(aUR5nXR;aMXPR;adURUnXc;aPURU5 UURRUR Vs/sH nUS:wdM UPM sn- nU(aMggs snf)aHelper function to find nodes which are connected to an output Arguments: node_output_name (str): The name of the output graph_input_names (set of string): The names of all inputs of the graph reachable (set of int): The set of indexes to reachable nodes in `nodes` r'N)popraddrr)r)rnode_output_namegraph_input_names reachablestackcurrent_output_namer+ input_names r_dfs_search_reachable_nodes%Extractor._dfs_search_reachable_nodesBs"""'))+ "7"kk1 $78)MM%(*.**//%*@*F*F*FJ%+#*FEes < B  B c[U5n[5nUHnURXSU5 M [U5Vs/sHo`RRUPM sn$s snfr)setr>sortedrr))r input_names output_names _input_namesr:r"r+s r_collect_reachable_nodes"Extractor._collect_reachable_nodesas[ ;' !e  D  , ,T K!5;94EF4E5 &4EFFFs"A#c0nURRHnX2URUR4'M /n[ U5nU(aUR 5nUR UR4U;aSURUR UR45nURU5 URUR5 U(aMU$r) r functionsr"domainrpopleftop_typer6appendextendr))rnodes function_mapfunctionreferred_local_functionsqueuer)s r!_collect_referred_local_functions+Extractor._collect_referred_local_functionsms>@  ,,H=E(--9 :-8: e ==?D dkk*l:'++T\\4;;,GH(//9 X]]+ e('r c>[5nUH9nURUR5 URUR5 M; URVs/sHoDU;dM URUPM nnUR Vs/sHoDU;dM UR UPM nn[ URR5nUS:wa[SUS35e[ URR5nUS:wa[SUS35eXV4$s snfs snf)Nrzlen_sparse_initializer is z, it must be 0.zlen_quantization_annotation is ) rArrrrrlenrsparse_initializerr/quantization_annotation) rrOall_tensors_namesr)trrlen_sparse_initializerlen_quantization_annotations r_collect_reachable_tensors$Extractor._collect_reachable_tensorss5'*eD  $ $TZZ 0  $ $T[[ 1+/*;*; *;QDU?U D  a *;  *.)9)9 )9ABS=S D  Q )9  "%TZZ%B%B!C !Q &,-C,DOT '*$***L*L&M# &! +12M1No^ &&!  s D%D DDc SURR-S-n[RR XX#XES9nUR R UR RSUS.n [RR"U40U D6$)NzExtracted from {})rrzonnx.utils.extract_model) ir_version opset_imports producer_namerI) rr"onnxhelper make_graphrrb opset_import make_model) rrOinputsoutputsrrlocal_functionsr"rmetas r _make_modelExtractor._make_models"DJJOO3c9 && k' **//!ZZ447(   {{%%e4t44r cURU5nURU5nURX5nURU5upgURU5nUR XSXFXx5$r)r3rFr^rTrn) rrCrDrjrkrOrrrls r extract_modelExtractor.extract_modelsq %%k2&&|4--kH"&"A"A%"H @@G 7  r )rrrrrN)rrreturnNone)rsdict)rszdict[str, int])r1 list[str]rslist[ValueInfoProto])r8strr9zset[str]r:zset[int]rsrt)rCrvrDrvrslist[NodeProto])rOryrslist[FunctionProto])rOryrsz.tuple[list[TensorProto], list[ValueInfoProto]])rOryrjrwrkrwrzlist[TensorProto]rrwrlrzrsr)rCrvrDrvrsr)__name__ __module__ __qualname____firstlineno__r staticmethodrrr3r>rFrTr^rnrq__static_attributes__r rr r s J// H$   > G G  G  G(( (.'' 8'455%5& 5 ' 5 ) 5-5 5*          r r cJ[RRU5(d[SU35eU(d [S5eU(d [S5eU(d [S5e[ U5[ [ U55:wa [S5e[ U5[ [ U55:wa [S5eU(a[ RRU5 U(aq[RRU5[ RR:a6[ RRX5 [ R"U5nOU(aj[ R"USS9n[ RRU5n[RRU5n[ R "Xg5 O[ R"U5n[#U5nUR%X#5n U R'5[ RR:a9[RR)U5S -n [ R*"XS U S 9 O[ R*"X5 U(a [ RRU5 g g ) aExtracts sub-model from an ONNX model. The sub-model is defined by the names of the input and output tensors *exactly*. Note: For control-flow operators, e.g. If and Loop, the _boundary of sub-model_, which is defined by the input and output tensors, should not _cut through_ the subgraph that is connected to the _main graph_ as attributes of these operators. Note: When the extracted model size is larger than 2GB, the extra data will be saved in "output_path.data". Arguments: input_path (str | os.PathLike): The path to original ONNX model. output_path (str | os.PathLike): The path to save the extracted ONNX model. input_names (list of string): The names of the input tensors that to be extracted. output_names (list of string): The names of the output tensors that to be extracted. check_model (bool): Whether to run model checker on the original model and the extracted model. infer_shapes (bool): Whether to infer the shapes of the original model. zInvalid input model path: z%Output model path shall not be empty!z&Input tensor names shall not be empty!z'Output tensor names shall not be empty!z0Duplicate names found in the input tensor names.z1Duplicate names found in the output tensor names.F)load_external_dataz.dataT)save_as_external_datalocationN)ospathexistsr/rWrArechecker check_modelgetsizeMAXIMUM_PROTOBUFshape_inferenceinfer_shapes_pathload infer_shapesdirnameload_external_data_for_modelr rqByteSizebasenamesave) input_path output_pathrCrDrrrbase_dire extractedrs rrqrqs4 77>>* % %5j\BCC @AA ABB BCC ;3s;/00KLL <CL 122LMM   , 3dll6S6SS ..zG +&  *?$$11%877??:. ))%: *%%A :Idll;;;77##K07: )xX ))   -r c/nUHn[RRXR5n[RR U5n[RR U5nUR U5(d[ SUS35eUR5(dUR5(a[ SUS35eURU5 M U$)zCheck that the content of ``tar`` will be extracted safely Args: tar: The tarball file base: The directory where the tarball will be extracted Returns: list of tarball members zThe tarball member z^ in downloading model contains directory traversal sequence which may contain harmful payload.zP in downloading model contains symbolic links which may contain harmful payload.) rrr0r"abspath startswith RuntimeErrorissymislnkrM)tarbaseresultmember member_pathabs_base abs_members r_tar_members_filterrsFggll45 77??4(WW__[1 $$X..%k]3RS  <<>>V\\^^%k]3DE   f Mr c[R"U5n[[S5(aURUSS9 OURU[ X!5S9 SSS5 g!,(df  g=f)a)Safely extracts a tar file to a specified directory. This function ensures that the extraction process mitigates against directory traversal vulnerabilities by validating or sanitizing paths within the tar file. It also provides compatibility for different versions of the tarfile module by checking for the availability of certain attributes or methods before invoking them. Args: model_tar_path: The path to the tar file to be extracted. local_model_with_data_dir_path: The directory path where the tar file contents will be extracted to. data_filterdata)rfilter)rmembersN)tarfileopenhasattr extractallr)model_tar_pathlocal_model_with_data_dir_pathmodel_with_data_zippeds r_extract_model_safer$sg n %)? 7M * * " - -3F .  # - -3+* .  & % %s AA  A.)TT)rstr | os.PathLikerrrCrvrDrvrboolrrrsrt)rztarfile.TarFilerrrszlist[tarfile.TarInfo])rrrrrsrt) __future__rrr collectionsrtypingr onnx.checkerre onnx.helperonnx.shape_inference onnx.onnx_pbrrrr r r rqrrrr rrs#  f f \ @.!@."@.@. @.  @.  @. @.F  1>%GX r