h jSSKrSSKJr SSKrSSKJrJrJrJ r Sr Sr Sr Sr \S:Xa\ "5 gg) N)Path) load_model numpy_helperonnx_pb save_model cUR/nU(Ga=URS5nU"U5 URGHnURHnUR[ R RR:XaF[UR[ R5(deURUR5 UR[ R RR:XdMURH5n[U[ R5(deURU5 M7 M GM U(aGM<gg)Nr)graphpopnode attributetyperAttributeProto AttributeTypeGRAPH isinstanceg GraphProtoappendGRAPHSgraphs)modelfunc graph_queuer r attrrs [/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/tools/onnx_randomizer.pygraph_iteratorrs;;-K " U JJD99 6 6 D D J JJ%dffg.@.@AAAA&&tvv.99 6 6 D D K KK![[)!W-?-?@@@@#**1-) ' +cURHn[R"U5nUR[:dM/[ R RUR5UR5URS9RUR5n[R"X1R5nURU5 M g)N)size) initializerrto_arrayr SIZE_THRESHOLDnprandomuniformminmaxshapeastypedtype from_arraynameCopyFrom)r i_tensorarray random_arrayo_tensors rrandomize_graph_initializerr3%s%%%%h/ :: &99,,UYY[%))+EKK,X__`e`k`klL#..|]]KH   h ' &rc[R"SS9nURS[SSS9 URS[SSS9 URS S S S S 9 URSS S SS 9 UR 5nSnUR (a~[ UR5R[ UR5R:Xa [S5eUR(a"[ UR5RS-n[ UR5RRSSS9 [URUR S9n[U[ 5 [#UURUR URUS9 g)Nz&Randomize the weights of an ONNX model) descriptionz-mTzinput onnx model path)rrequiredhelpz-ozoutput onnx model pathz--use_external_data_formatF store_truez%Store or Save in external data format)r6actionr7z--all_tensors_to_one_filezSave all tensors to one filezNPlease specify output directory with different parent path to input directory.z.data)parentsexist_ok)load_external_data)save_as_external_dataall_tensors_to_one_filelocation)argparseArgumentParser add_argumentstr parse_argsuse_external_data_formatrmparento RuntimeErrorr>r-mkdirrrr3r)parserargs data_path onnx_models rmainrO/s[  $ $1Y ZF 3rVsA>>  . ("J zFr