h.DSSKJr SSKJr SSKrSSKJr "SS5rg)) annotations)dequeN) ONNXModelc\rSrSrSrSSjrSSjrSSjrSr\ SSj5r \ SSj5r \ SS j5r \ SS j5r SS jrSSS jjrSS S jjrS!SjrS/4S"SjjrSS/S4S#SjjrS$S%SjjrS&SjrS'S(SjjrSrg))Fusionz Base class for fusions. cX0lX lXl/Ul/UlURS-UR-S-UlSUlg)N_fused__)search_op_type fused_op_typemodelnodes_to_remove nodes_to_add_new_node_name_prefix_new_node_name_suffix)selfrrr s a/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/quantization/fusions/fusion.py__init__Fusion.__init__sO#1"/ % %'"$%)%7%7)%CdFYFY%Y\_%_"%)"c[e)zm Interface function for derived fusion classes. Tries to fuse a node sequence containing the specified node. )NotImplementedError)rnodeinput_name_to_nodesoutput_name_to_nodes rfuse Fusion.fuses "!rc2URR5nURR5nURR5H1nURUR :XdMUR X1U5 M3 URRUR5 URRUR5 [UR=(d UR5nU(aURR5 U$)z/ Apply graph fusion on the entire model graph. ) rrrnodesop_typer r remove_nodesr add_nodesrboolremove_unused_constant)rrrr graph_updateds rapply Fusion.apply*s#jj<<>"jj<<>JJ$$&D||t222 $5HI'  4 45 T../T11FT5F5FG  JJ - - /rcURnURc%URRU5nUS-UlUUR<3nU=RS- slU$)N)rrrget_largest_node_name_suffix)rprefixlargest_suffixnew_names rcreate_unique_node_nameFusion.create_unique_node_name?se++  % % -"&**"I"I&"QN)7!);D &Xd88;< ""a'"rcUH8nURH%nXQ;aM XR;dMX%H nX`;dM g M' M: gNFT)output)r keep_outputsrrnode_to_removeoutput_to_remove impacted_nodes ris_safe_to_fuse_nodesFusion.is_safe_to_fuse_nodesKsJ.N$2$9$9 #3#:)<)N (?#(*O %:.rcURH6nURU:XdM[RR U5nUs $ g)N) attributenameonnxhelperget_attribute_value)rattribute_nameattrvalues rget_node_attributeFusion.get_node_attribute^s:NNDyyN* 77= #rcR[UR5Hup#X0:XdM Us $ g)N) enumerateinput) node_output child_nodeindex input_names r input_indexFusion.input_indexfs*!*:+;+;!< E( "=rc2/nURRHznURS5(aURUR5 M6URS5(aURUR 5 MiURS5 M| U$)N dim_value dim_param?)shapedimHasFieldappendrQrR) tensor_type shape_listds rtensor_shape_to_listFusion.tensor_shape_to_listmsu ""&&Azz+&&!!!++.K((!!!++.!!#& 'rc[UR5H(up#URRU5nUcM%X$4s $ g)NNN)rHrIrget_constant_value)rriinprCs rget_constant_inputFusion.get_constant_inputys= +FAJJ11#6E x, rcvURU5upEUb#URS:Xa[XR- 5U:aU$g)Nr+rG)rbsizeabs)rrexpected_valuedeltar`rCs rfind_constant_inputFusion.find_constant_inputs<**40  qS9O5PSX5XHrc,URXU5S:$Nr)ri)rrrgrhs rhas_constant_inputFusion.has_constant_inputs''eDIIrcvURRU5nUcg[UR5U:waggr3)rr_lenrT)r output_namerankrCs ris_constant_with_specified_rank&Fusion.is_constant_with_specified_ranks5 --k: = u{{ t #rNcUcURR5n[UR5H,upVXc;dM X6nURU:XdM"Xt;dM)Xu4s $ g)a Find parent node based on constraints on op_type. Args: node: current node. parent_op_type (str): constraint of parent node op_type. output_name_to_node (dict): dictionary with output name as key, and node as value. exclude (list): list of nodes that are excluded (not allowed to match as parent). Returns: parent: The matched parent node. None if not found. index: The input index of matched parent node. None if not found. r^)rrrHrIr")rrparent_op_typerexcluder`raparents rmatch_first_parentFusion.match_first_parents^(  &"&**"@"@"B  +FA),1>>^38M!9$ , rcXUceUbUS:deUcURR5nUc*URXXE5upxUbURU5 U$U[ UR 5:agURR XU5nUbURU:XaXu;aU$g)a Find parent node based on constraints on op_type and index. When input_index is None, we will find the first parent node based on constraints, and return_indice will be appended the corresponding input index. Args: node (str): current node name. parent_op_type (str): constraint of parent node op_type. input_index (int or None): only check the parent given input index of current node. output_name_to_node (dict): dictionary with output name as key, and node as value. exclude (list): list of nodes that are excluded (not allowed to match as parent). return_indice (list): a list to append the input index when input_index is None. Returns: parent: The matched parent node. Nr)rrryrWrprI get_parentr") rrrvrNrrw return_indicerxrLs r match_parentFusion.match_parents2"kQ&666  &"&**"@"@"B    33DJ]gMF($$U+M #djj/ )&&t:MN  &..N"BvG\Mrc Ub[U5[U5:XdeUcURR5nUn/n[U5H8upUR UU UbX8OSU/US9n U c gUR U 5 U nM: U$)a Find a sequence of input edges based on constraints on parent op_type and index. When input_index is None, we will find the first parent node based on constraints, and return_indice will be appended the corresponding input index. Args: node (str): current node name. parent_op_types (str): constraint of parent node op_type of each input edge. parent_input_index (list): constraint of input index of each input edge. None means no constraint. output_name_to_node (dict): dictionary with output name as key, and node as value. return_indice (list): a list to append the input index When there is no constraint on input index of an edge. Returns: parents: a list of matched parent node. N)rwr})rprrrHr~rW) rrparent_op_typesparent_input_indexrr} current_nodematched_parentsr`r"matched_parents rmatch_parent_pathFusion.match_parent_paths0  ))*c/.BB BB  &"&**"@"@"B  #O4JA!..);)G"%T#+ /N%  " "> 2)L5rc|[U5H-upE/nURXSUSX65nU(dM)XGU4s $ g)z0 Find a matching parent path to the given node. rr+)rGNN)rHr)rrpathsrr`pathr}matcheds rmatch_parent_pathsFusion.match_parent_pathssL!'GAM,,T7DGEXhGw=00 ( rcLURRX5n[U5n[U5S:aoUR 5nUR U:XaU$U(a5URRXs5nUHnUR U5 M [U5S:aMogrl)r get_childrenrrppopr" appendleft) rr child_typer recursivechildrendqrchilds rfind_first_child_by_typeFusion.find_first_child_by_type$s::**4E 8_"gk668L##z1##::22<U%EMM%(&"gkr)rrrrrrr )rrrstrr r)ronnx.NodeProtordict[str, list[onnx.NodeProto]]rdict[str, onnx.NodeProto])returnr%) rlist[onnx.NodeProto]r5 list[str]rrrrrr%)rrrAr)rJrrKrrint)rz list[int])rr)gư>)rrrgfloatrhrrr)rrrgrrhrrr%)rqrrrrrr%) rrrvrr dict[str, onnx.NodeProto] | Nonerwrrz(tuple[onnx.NodeProto | None, int | None])rrrvrrNz int | Nonerrrwrr}list[int] | Noneronnx.NodeProto | None)NNN) rrrrrrrrr}rrzlist[onnx.NodeProto] | None)rrrz!list[tuple[list[str], list[int]]]rrrz9tuple[int, list[onnx.NodeProto] | None, list[int] | None])NT) rrrrrz&dict[str, list[onnx.NodeProto]] | Nonerr%rr)__name__ __module__ __qualname____firstlineno____doc__rrr(r0 staticmethodr9rDrNr[rbrirmrsryr~rrr__static_attributes__rrrrs* " "= "7 "* -=7   $   JAE(* >  &  2 F#'@D(**.--- - > - & -(- -f04@D*. //#/- / > / ( / %/b17  C (GK D     rr) __future__r collectionsrr> onnx_modelrrrrrrs # "hhr