IЦiP SSKrSSKrSSKJrJrJrJrJrJrJ r SSK r SSK J s J s Js Jr SSKJ s J s Jr SSKJ r SSKJrJr SSKJr SSKJr SSKJr SSKJr SS KJ r J!r! \ RDRr#"S S \RH5r%S \S \S\S\\&\\ 4S\\%\%44 Sjr'S \S \S\\&\\ 4S\\\ \ RP\)4\ \ RP\*444Sjr+S \S \S\4Sjr,S \S \S\*4Sjr-S \S\\*4Sjr.S \S \S\&4Sjr/S\!S\&S\!4Sjr0S\!SS4Sjr1Sr2\2S\ RPS\ RPS\ RP4Sj5r3\2S\ RPS\ RPS\ RP4Sj5r4\2S\ RPS\ RPS\ RP4S j5r5S \S\64S!jr7S \S \S"\*S\4S#jr8g)$N)CallableDictListOptionalSetTupleUnion)FakeQuantizeBase ObserverBase)_is_activation_post_process)getattr_from_fqn) GraphModule)Node)NSNodeTargetType NSResultsTypec\rSrSr\R "5r\R "5r\R "5r\R "5r \R "5r Sr g)NodeInputOrOutputTypeN) __name__ __module__ __qualname____firstlineno__enumautoFP32INT8FP16UNKNOWN FP32_OR_INT8__static_attributes__rS/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/ao/ns/fx/utils.pyrrs; 99;D 99;D 99;DiikG 99;Lr#rnodegm logger_clsnode_type_to_io_type_mapreturnc^USnUSnUSnUSnUSnUSn USn USn URS :XaURU;a [R[R4$URU;a [R[R4$URU;a [R [R 4$URU;a5[ XS 5n [U [5(de[XX#5un nX4$[R[R4$URS :XGaMURS :Xde[UR[5(de[XR5m[U4S jU 55n[TU[[45(dU(a5[ XS 5n [U [5(de[XX#5un nX4$[U4S jU55n[U4SjU 55nU(a [R[R4$U(a [R [R 4$[R[R4$URS:XGa8URS:XaE[ XS 5n[U[5(de[UXU5un nU[R4$URS:Xan[ XS 5n[U[5(de[UXU5un n[ XS5nU[ R"Ld US35eU[R4$URU ;a5[ XS 5n [U [5(de[XX#5un nX4$[R[R4$[R[R4$)Nfuns_io_type_fp32funs_io_type_fp16funs_io_type_int8funs_io_type_fp32_or_int8mods_io_type_fp32mods_io_type_int8mods_io_type_fp32_or_int8meths_io_type_fp32_or_int8 call_functionr call_modulec3<># UHn[TU5v M g7fN isinstance.0 target_typemods r$ 7get_node_first_input_and_output_type..Ns1 # UHn[TU5v M g7fr6r7r9s r$r=r>`) # UHn[TU5v M g7fr6r7r9s r$r=r>crAr? call_method dequantizetorz handling needs to be added)optargetrrrrget_normalized_nth_inputr8r$get_node_first_input_and_output_typer strr anyr r torchfloat16)r%r&r'r(FUNS_IO_TYPE_FP32FUNS_IO_TYPE_FP16FUNS_IO_TYPE_INT8FUNS_IO_TYPE_FP32_OR_INT8MODS_IO_TYPE_FP32MODS_IO_TYPE_INT8MODS_IO_TYPE_FP32_OR_INT8METHS_IO_TYPE_FP32_OR_INT8 first_arg_prev_node_input_typeprev_node_output_type"is_known_fp32_or_int8_input_moduleis_known_fp32_input_moduleis_known_int8_input_module prev_nodecur_node_dtype_targetr<s @r$rIrI&s11DE01DE01DE 89T U01DE01DE 89T U!9:V!W ww/! ;;+ +)..0E0J0JK K ;;+ +)..0E0J0JK K [[- -)..0E0J0JK K [[5 501=Ii.. ..5z %%*A A)113H3P3PQ Q M !ww-'''$++s++++r;;/-01 [R@[RB[$RD[$RF[$RH[$R&[$RJ[$RL45(aTRNTRP4$[SU4S jU55nU(a [UX1U5$g) zo Returns the qparams (scale, zero_point) of the first input to `node`, if they can be inferred from the graph. rNr1cd[XU5n[XU5n[U[5(a[UR[5(de[U[5(a[UR[5(de[ XR5n[ XR5nXg4$r6)rHr8rrGrJr )r%r& scale_arg_idx zp_arg_idx scale_nodezp_node scale_objzp_objs r$ _get_scale_zp_from_function_args@get_node_input_qparams.._get_scale_zp_from_function_argss-d F *4Z@*d++ :;L;Lc0R0RRR'4((Z-L-LLL$R):):; !"nn5""r#r3rr4c3<># UHn[TU5v M g7fr6r7)r:r; module_objs r$r=)get_node_input_qparams..s1 C\KJz; / /C\r?)+rHr8rrFrGrLquantize_per_tensortoqaddadd_relumulmul_relurJr nnqLinearConv1dConv2dnniq ConvReLU2dConv3d BatchNorm2d BatchNorm3dConvTranspose1dConvTranspose2dELU GroupNormInstanceNorm1dInstanceNorm2dInstanceNorm3d LayerNorm Hardswish LeakyReLUReLU6BNReLU2dBNReLU3d ConvReLU1d ConvReLU3d LinearReLUscale zero_pointrKget_node_input_qparams)r%r&r(r\rTrfrYrks @r$rrs)15I i & & 89T U#||&   u88 83I1aH H   #''3<<#,,!O O3I1aH H  &)**C0000%b*:*:;      #### """"""      1   :$$j&;&;< <-01 C\1 . * .))9QR R r#c6URS:XGa[XR5n[U5(a[ UR 5S:Xde[ UR S[5(deUR Sn[ UR[5(de[XR5n[U5(aN[ UR 5S:Xde[ UR S[5(deUR SnU$)ak If node is not an observer, returns it. If node is an observer, navigates up the graph and returns the first parent which is not an observer. For example, graph: (node_non_obs), node = node_non_obs : returns node_non_obs graph: (node_non_obs -> obs0), node = obs0 : returns node_non_obs graph: (node_non_obs -> obs0 -> fq0), node = fq0 : returns node_non_obs r4rr) rFr rGr lenargsr8rrJr%r&node_objs r$return_first_non_observer_noders ww-#B 4 &x 0 0tyy>Q& &&diilD11 1199QUtp#n[U[5(a[U[5(d*[U[5(aJ[U[5(a5/n[X#5H"upgXg/UQ7nUR T "U0UD65 M$ U$[U[ R 5(aa[U[ R 5(aBUR(aUR5nUR(aUR5nUR[ R:wdUR[ R:wagX#/UQ7nT "U0UD6$r6) r8tuplelistzipappendrLTensor is_quantizedrDdtypefloat) rkwargsa0a1a_otherrel0el1new_argsfinners r$rGmaybe_dequantize_first_two_tensor_args_and_handle_tuples..inners r5 ! !jU&;&; r4 ZD%9%9GK/w/uh9&9:(N ELL ) )jU\\.J.J]]_]]_ 88u{{ "bhh%++&=%W%(%f%%r#r)rrs`@r$8maybe_dequantize_first_two_tensor_args_and_handle_tuplesrs&2 Lr#xyc[R"U5n[R"X- 5nS[R"X#- 5-$)z Computes the SQNR between `x` and `y`. Args: x: Tensor or tuple of tensors y: Tensor or tuple of tensors Return: float or tuple of floats )rLnormlog10)rrPsPns r$ compute_sqnrrs7 AB AE B  BG$ $$r#c|[R"X- S-R5US-R5- 5$)z Computes the normalized L2 error between `x` and `y`. Args: x: Tensor or tuple of tensors y: Tensor or tuple of tensors Return: float or tuple of floats rh)rLsqrtsumrrs r$compute_normalized_l2_errorrs1 ::!|((*adZZ\9 ::r#cURSS5nURSS5n[RRR X5$)z Computes the cosine similarity between `x` and `y`. Args: x: Tensor or tuple of tensors y: Tensor or tuple of tensors Return: float or tuple of floats r)reshaperLr functionalcosine_similarityrs r$compute_cosine_similarityrs? !RA !RA 88   0 0 66r#cURS:XakUR[R[R[ R[ R[R [R4;agg)Nr3FT)rFrGrLrorqrcatstack)r%s r$op_type_supports_shadowingrsQ ww/! ;; II II LL LL II KK    r#idxcJURUSS9nUbPUupE[U5[U5-U:deU[U5:aXB$[UR55U$[UR5[UR 5-U:deU[UR5:aURU$U[UR5-n[UR R55U$![ a [UR5[UR 5-U:deU[UR5:aURUs$U[UR5-n[UR R55Us$f=f)zi Given a node, gets the n'th input to that node, normalizing args and kwargs to the best of its ability. T)normalize_to_only_use_kwargs)normalized_argumentsrrrrr RuntimeError)r%r&rnorm_args_and_kwargs norm_args norm_kwargs kwargs_idxs r$rHrHsp :#88 T 9  +%9 "Iy>C $44s: ::S^# ~%K..01#66tyy>C $44s: ::S^#yy~% 3tyy>1 DKK..01*==  :499~DKK 003666 TYY 99S> !s499~-J **,-j9 9 :s,AC< C<%AC<>=C<F"!F")9rrtypingrrrrrrr rLtorch.ao.nn.intrinsic.quantizedaor intrinsicrrwtorch.ao.nn.quantizedrstorch.nntorch.ao.quantizationr r torch.ao.quantization.observerr torch.ao.quantization.utilsr torch.fxrtorch.fx.graphrns_typesrrrrnEnumrrJrIrrintrrrrrrrrrrrboolrrHrr#r$rs DDD ..##@F8 5 ii DII xN xNxNxN#3,<(=#=> xN  "7 78 xNvM MM#3,<(=#=>MeE%,,-.ellC6G0HHIJ M`  :     44DI0 d   $ $$$N6M6d6D:: %ELL %U\\ %ell %: % : ;5<< ;ELL ;U\\ ;: ;:77%,,75<<7:7& T d !:4!:[!:s!:t!:r#