h-JSSKrSSKrSSKrSSKrSSKJr SSKJrJrJ r SSK J r J r J r Jr "SS5rS\S\S S4S jrS \ S\S S4S jrS-S\S \S\ \S\ \S\ \S\ \S S4SjjrS.S \ S\S \ \S\S\S S4 SjjrS \ S S4SjrS\S\S S4SjrS\ S \\4SjrS\ S\\ /\\4S \\4SjrS\ S \\4S jrS\ S \\4S!jrS\ S \\4S"jrS\ S \\4S#jrS$\S \4S%jrS&\S \4S'jr S\S \4S(jr!S\S)\S S4S*jr"S \ S+\S \ 4S,jr#g)/N)chain)CallableIterableOptional)AttributeProto GraphProto ModelProto TensorProtoc&\rSrSrS\SS4SjrSrg)ExternalDataInfotensorreturnNc^SUlSUlSUlSUlSUlUR H#n[ XRUR5 M% UR(a[UR5UlUR(a[UR5Ulgg)N) locationoffsetlengthchecksumbasepath external_datasetattrkeyvalueint)selfrentrys S/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/external_data_helper.py__init__ExternalDataInfo.__init__s|     ))E D))U[[ 1* ;;dkk*DK ;;dkk*DK )rrrrr)__name__ __module__ __qualname____firstlineno__r r__static_attributes__r!rr r s+{+t+r!r rbase_dirrc[U5n[UR5n[RR X5n[ US5nUR(aURUR5 UR(a!URUR5Ul OUR5Ul SSS5 g!,(df  g=f)z Loads data from an external file for tensor. Ideally TensorProto should not hold any raw data but if it does it will be ignored. Arguments: tensor: a TensorProto object. base_dir: directory that contains the external data. rbN) r _sanitize_pathrospathjoinopenrseekrreadraw_data)rr(info file_locationexternal_data_file_path data_files rload_external_data_for_tensorr7 s F #D"4==1M ggll8C %t , ;; NN4;; ' ;;'nnT[[9FO'nn.FO - , ,s A4C  Cmodelc[U5HDn[U5(dM[X!5 [RUlUR SS2 MF g)z Loads external tensors into model Arguments: model: ModelProto to load external data to base_dir: directory that contains external data N)_get_all_tensorsuses_external_datar7r DEFAULT data_locationr)r8r(rs rload_external_data_for_modelr>7sB#5) f % % )& ;#.#6#6F $$Q' *r!rrrrrcURS5(d[SUR-S-5eURSS2 [R UlUUb [U5OSUb [U5OSUUS.R5H:upgUcM URR5nXhl [U5Ul M< g)Nr2zTensor zGdoes not have raw_data field. Cannot set external data for this tensor.)rrrrr) HasField ValueErrornamerr EXTERNALr=ritemsaddrstrr) rrrrrrkvrs rset_external_datarIHs ??: & & kk W X  Q&//F!'!3#f+!'!3#f+   eg  =((,,.EIa&EKr!all_tensors_to_one_filesize_thresholdconvert_attributec[U5nU(a [U5nU(a[[R"55nU(a1[ R RU5(a [S5eUnUHLnURS5(dM[R"UR5U:dMA[Xv5 MN gUHnURS5(dM[R"UR5U:dMAURn[U5(d[[R"55n[Xx5 M g)a Call to set all tensors with raw data as external data. This call should preceed 'save_model'. 'save_model' saves all the tensors data as external data after calling this function. Arguments: model (ModelProto): Model to be converted. all_tensors_to_one_file (bool): If true, save all tensors to one external file specified by location. If false, save each tensor to a file named with the tensor name. location: specify the external file relative to the model that all tensors to save to. Path is relative to the model path. If not specified, will use the model name. size_threshold: Threshold for size of data. Only when tensor's data is >= the size_threshold it will be converted to external data. To convert every tensor with raw data to external data set size_threshold=0. convert_attribute (bool): If true, convert all tensors to external data If false, convert only non-attribute tensors to external data zDlocation must be a relative path that is relative to the model path.r2N)_get_initializer_tensorsr:rFuuiduuid1r,r-isabsrAr@sys getsizeofr2rIrB_is_valid_filename) r8rJrrKrLtensors file_namertensor_locations rconvert_model_to_external_datarXfs.'u-G"5) % ww}}X&& Z!IF ++MM&//2nD!&4 F ++MM&//2nD"(++)/::&)$**,&7O!&:r!c[U5HZn[U5(dMURS5(d [S5eURSS2 [ R UlM\ g)z Call to set all tensors which use external data as embedded data. save_model saves all the tensors data as embedded data after calling this function. Arguments: model (ModelProto): Model to be converted. r2raw_data field doesn't exist.N)r:r;r@rArr r<r=)r8rs r convert_model_from_external_datar[sU#5) f % %??:.. !@AA$$Q'#.#6#6F *r! base_pathc*[U5n[RRXR5nUR S5(d [ S5e[RRU5(d[US5 SSS5 [US5nURSS5 URb\UR5nURU:a!URSURU- -5 URUR5 UR5nURUR5 [XRXdR5U- 5 SSS5 g!,(df  N=f!,(df  g=f) z Writes tensor data to an external file according to information in the `external_data` field. Arguments: tensor (TensorProto): Tensor object to be serialized base_path: System path of a folder where tensor data is to be stored r2rZabNzr+br)r r,r-r.rr@rAisfiler/r0rtellwriter2rI)rr\r3r5r6 file_sizers rsave_external_datares$ F #D ggll9mmD ??: & &899 77>>1 2 2 )4 0 1 %u -q! ;; "!(I{{Y&y)@ AB NN4;; '!(&--9IF9RS . - 1 0 . -sE3CF3 F Fonnx_model_protoc>[[U5[U55$)z=Scan an ONNX model for all tensors and return as an iterator.)rrN_get_attribute_tensorsrfs rr:r:s!  !12/0 r! attributefuncc## UR[R:XaU"UR5ShvN UR[R:Xa$UR HnU"U5ShvN M ggNGN 7f)zICreate an iterator through processing ONNX model attributes with functor.N)typerGRAPHgGRAPHSgraphs)rjrkgraphs r_recursive_attribute_processorrssi~~--- $$$~~...%%EE{ " "&/ % #s!4BA>  (,2 s!AA5AA AAc#J# [UR5ShvN gN7f)z:Create an iterator of initializer tensors from ONNX model.N)rxrrris rrNrNs23C3I3IJJJ #!#c## URHfnURHSnURS5(aURv URShvN [ U[ 5ShvN MU Mh gN%N7f)zJCreate an iterator of tensors from node attributes of an ONNX model graph.tN)rwrjr@r}rUrs!_get_attribute_tensors_from_graphrys rr~r~sq'++I!!#&&kk! (( ( (5<   (, ) s$AA>A:A>,A<-A><A>c#J# [UR5ShvN gN7f)zDCreate an iterator of tensors from node attributes of an ONNX model.N)r~rrris rrhrhs01A1G1GHHHr{r-c$URS5$)zRemove path components which would allow traversing up a directory tree from a base path. Note: This method is currently very basic and should be expanded. z/.)lstrip)r-s rr+r+s ;;t r!filenamecf[R"S5nURU5n[U5$)z8Utility to check whether the provided filename is valid.z^[^<>:;,?"*|/]+$)recompilematchbool)rexprs rrTrTs) **' (C IIh E ;r!clURS5=(a UR[R:H$)z?Returns true if the tensor stores data in an external location.r=)r@r=r rC)rs rr;r;s. ( 9  K$8$8 8r! field_keyc~[UR5H$up#URU:XdMURU M& g)z Removes a field from a Tensor's external_data key-value store. Modifies tensor object in place. Arguments: tensor (TensorProto): Tensor object from which value will be removed field_key (string): The key of the field to be removed N) enumeraterr)rrifields rremove_external_data_fieldrs5f223 99 !$$Q'4r!filepathc[U5HIn[U5(dMURS5(dM-[X!5 UR S5 MK U$)a Serializes data for all the tensors which have data location set to TensorProto.External. Note: This function also strips basepath information from all tensors' external_data fields. Arguments: model (ModelProto): Model object which is the source of tensors to serialize. filepath: System path to the directory which should be treated as base path for external data. Returns: ModelProto: The modified model object. r2)r:r;r@re ClearField)r8rrs rwrite_external_data_tensorsr,sK#5) f % %&//**E*E v 0   j )* Lr!)NNNN)TNiF)$r,rrRrO itertoolsrtypingrrr onnx.onnx_pbrrr r r rFr7r>rrIrrXr[rer:rsrxrNr~rhr+rTr;rrr'r!rrsj //LL++$/+///.( (c(d((! "" ! !! SM! SM ! sm ! sm ! !@%)"# 2; 2;!2;sm2; 2;  2;  2;j7J747" T{ Ts Tt TFzh{6K##%-zlH[SK &  k IZIH[