h SSKJr SSKrSSKrSSKrSSKrSSKJrJrJ r J r J r "SS5r SSSjjr g) ) annotationsN) FunctionProto ModelProto NodeProto TensorProtoValueInfoProtoc\rSrSrSSjr\S5rSrSSjrSSjr SSjr SSjr S r SS jr SS jrSS jrS rg) Extractorc*[RRU5UlURRUlUR URR 5UlUR URR5Ul gN) onnxshape_inference infer_shapesmodelgraph_build_name2obj_dict initializerwmap value_infovimap)selfrs D/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/utils.py__init__Extractor.__init__sc))66u= ZZ%% --djj.D.DE ..tzz/D/DE cFUVs0sHoRU_M sn$s snfr )name)objsobjs rrExtractor._build_name2obj_dicts )-.## ...sc@URU5n[U5n[U5nXT-nXT- n/nUHn URX95 M UH!n URURU 5 M# URU5n UV s/sHoU PM sn $s sn fr )rsetappendr) r original_ioio_names_to_extractoriginal_io_maporiginal_io_namess_io_names_to_extractio_names_to_keepnew_io_names_to_addnew_io_tensorsrnew_io_tensors_maps r_collect_new_io_coreExtractor._collect_new_io_cores33K@0 #$7 80D3G$D  ! !/"7 8%'D  ! !$**T"2 3( "66~F5HI5HT4(5HIIIs BcNURURRU5$r )r.rinputrnamess r_collect_new_inputsExtractor._collect_new_inputs+s(()9)95AArcNURURRU5$r )r.routputr2s r_collect_new_outputsExtractor._collect_new_outputs.s(():):EBBrcX;agURRHQnXR;aMXC;aMURU5 URHnUR XRU5 M MS gr )rnoder7r$r1_dfs_search_reachable_nodes)rnode_output_namegraph_input_namesreachable_nodesr;rs rr<%Extractor._dfs_search_reachable_nodes1sd  0 JJOOD{{2&  " "4 ( 00_#$rc/nUHnURXAU5 M URRVs/sH oUU;dM UPM nnU$s snfr )r<rr;)r input_names output_namesr?rnnodess r_collect_reachable_nodes"Extractor._collect_reachable_nodesEsQ  D  , ,T P!!JJOODOqO/COD Es A A cX^U4Sjn/nU"X5nU(aU"XC5nU(aMU$)Nc>^/nUHim[U4SjTRR5S5nU(dM6X1;dM=URU5 UR UR 5 Mk U$)Nc3># UH?nURTR:XdMURTR:XdM;Uv MA g7fr )rop_typedomain).0fr;s r [Extractor._collect_referred_local_functions..find_referred_funcs..^s?!5A66T\\167hh$++6M!5sA A  A )nextr functionsr$extendr;)rEreferred_local_functions new_nodesmatch_functionr;rs @rfind_referred_funcsHExtractor._collect_referred_local_functions..find_referred_funcsYsnI!%!%!5!5  "">n&T,33NC$$^%8%89 r)rrErWrTrUs` r!_collect_referred_local_functions+Extractor._collect_referred_local_functionsQs6 $$& 'H +IPIi('rc>[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 ) r#updater1r7rrlenrsparse_initializer ValueErrorquantization_annotation) rrEall_tensors_namesr;trrlen_sparse_initializerlen_quantization_annotations r_collect_reachable_tensors$Extractor._collect_reachable_tensorsrs'*eD  $ $TZZ 0  $ $T[[ 1.2YYQY?P:P|tyy|Y Q-1ZZRZ@Q;QmdjjmZ R!$TZZ%B%B!C !Q &,-C,DOT '*$***L*L&M# &! +12M1No^ &&RRs 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_namerR) rrrhelper make_graphrrj opset_import make_model) rrEinputsoutputsrrlocal_functionsrrmetas r _make_modelExtractor._make_models"DJJOO3c9 && k'  **//!ZZ447(   {{%%e4t44rcURU5nURU5nURX5nURU5upgUR U5nUR XSXFXx5n U $r )r4r8rFrfrZru) rrBrCrqrrrErrrsrs r extract_modelExtractor.extract_modelsu ))+6++L9--kH"&"A"A%"H @@G  7  r)rrrrN)rrreturnNone)r3 list[str]rzlist[ValueInfoProto])r=strr>r|r?list[NodeProto]rzr{)rBr|rCr|rzr)rErrzz.tuple[list[TensorProto], list[ValueInfoProto]])rErrqr}rrr}rzlist[TensorProto]rr}rszlist[FunctionProto]rzr)rBr|rCr|rzr)__name__ __module__ __qualname____firstlineno__r staticmethodrr.r4r8r<rFrZrfrurx__static_attributes__rYrrr r sF //J$BC%)   (      (B'' 8'055%5& 5 ' 5 ) 5-5 5,   rr c[RRU5(d[SU35eU(d [S5eU(d [S5e[R R U5 [R"U5n[U5nURX#5n[R"Xq5 U(a [R R U5 gg)a Extracts 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. 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 extracted model. zInvalid input model path: z%Output model path shall not be empty!z'Output tensor names shall not be empty!N) ospathexistsr`rchecker check_modelloadr rxsave) input_path output_pathrBrCrre extracteds rrxrxs, 77>>* % %5j\BCC @AA BCCLLZ( IIj !E%A :IIIi%   -r)T) rstr | os.PathLikerrrBr|rCr|rboolrzr{) __future__rr onnx.checkerr onnx.helperonnx.shape_inferencerrrrrr rxrYrrrsk# RR``P %.!%."%.%. %.  %.  %.r