h7SSKrSSKrSSKrSSKrSSKJr SSKJr SSKJr SSK r \ \ \4r Sr SrS\ R4SjrSS\ RS\\ 4S jjr"S S 5rS rS r\S:Xa\"5 gg)N) OrderedDict)pprint)Anytuning_resultsc\[U5HupUR[:XdMUs $ g)N) enumeratekey_TUNING_RESULTS_KEY)metadata_propsidxprops Z/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/tools/offline_tuning.py_find_tuning_results_in_propsrs*~.  88* *J/ modelc[UR5nUS:agURUn[R"UR5$)Nr)rr jsonloadsvalue)rr tuning_results_props rextractrsC '(<(< =C Qw..s3 ::)// 00rc[UR5nU(d US::dS5eUS:aURRU5 URR5n[Ul[ R"U5UlU$)NrzDHrc,[UR5$r))hashr r/s r__hash__Merger.EpAndValidators.__hash__7s> !rctURUR:H=(a URUR:H$r))r&r )r/others r__eq__Merger.EpAndValidators.__eq__:s'77ehh&@488uyy+@ @r)r&r r'N) __name__ __module__ __qualname____firstlineno__strdictr0r5r9__static_attributes__rrEpAndValidatorsr$1s( ?s ?S#X ?  " ArrCc"[5Ulgr))r ev_to_resultsr4s rr0Merger.__init__=s (]rrc8UHnURU5 M gr)) _merge_one)r/rtrss rmerge Merger.merge@s!C OOC "rc/nURR5Hdup#0nURURUS.nUR5HuupgnUR U05n XU'M UR U5 Mf U$)N)r&r'results)rEr.r&r' setdefaultappend) r/rev flat_resultsrMrIop_sig params_sig kernel_id kernel_maps r get_mergedMerger.get_mergedDs $ 2 2 8 8 : BGee mm"C 4@3E3E3G/$i$//; )2:&4H  ! !# &!;rrIc[RUSUS5nURRU05nUSR 5H,upEUR 5HupgXF4U;dMXsXF4'M M. g)Nr&r'rM)r!rCrErNr.)r/rIrPrQrRrUrSrTs rrHMerger._merge_oneSs{  # #CIsrCr0list TuningResultsrJrVrHrArBrrr!r!0s7 A A+!D$7! CmCrr!c[R"5nURSSS9nURSSS9nUR S5 UR S5 URS S S9nUR S S S SS9 UR SSS9 UR SSS9 UR SSSS9 URSSS9nUR SSS9 UR SSSS9 URSSS9nUR SSS9 UR 5n[ [U55S:XaUR5 [S 5 U$)!NzCommand to executecmd)helpdestrz2Extract embedded tuning results from an onnx file.)r^ input_onnx output_jsonrz+Embed the tuning results into an onnx file.z--forcez-f store_truez+Overwrite the tuning results if it existed.)actionr^ output_onnxzPath of the output onnx file.zPath of the input onnx file. input_json+z5Path(s) of the tuning results file(s) to be embedded.)nargsr^rJz4Merge multiple tuning results files as a single one.z'Path of the output tuning results file.z/Paths of the tuning results files to be merged.rz Pretty print the tuning results. json_or_onnxz+A tuning results json file or an onnx file.rr) argparseArgumentParseradd_subparsers add_parser add_argument parse_argslenvars print_helpexit)parser sub_parsersextract_parser embed_parser merge_parser pprint_parserargss rrnrn\sd  $ $ &F''-A'NK ++I W 0;???K?atyya1?KN LL (Le..0$**= //0 W ;???K?atyya1?KN LL (L &##%tD,<,?N  ! (9(9:!(!)JJ$$'88a%bcHHRL  ! JJ   1 122cd e HHRL~ EL L     s1=)L..)L3?)L8,A0M8 MM MM__main__)F)rir*rr~ collectionsrrtypingrr|r@r?r[r r ModelProtorrZrr!rnrr;rBrrrs # S#X &14??1  $}2E )C)CX8- ` zFr