[Th| ^SSKJrJrJr SSKrSSKJr SSKJr SSK J r SSK J r J r Jr SSKJr /SQr\"S S 9S \ S \S \S\S\4 Sj5r\"S S 9"SS\55r\"S S 9SS \ S\\\R,SS4Sjj5r\"S S 9S\ S\4Sj5r\"S S 9S \ S\ S\4Sj5rg))Any NamedTupleOptionalN) compatibility)Graph) GraphModule)map_argNodeTarget) ShapeProp)replace_target_nodes_with size_bytesget_size_of_all_nodesget_tensor_metaget_size_of_nodeF)is_backward_compatible fx_moduleold_op old_targetnew_op new_targetc^ [5n0m URRHnURU:XaURU:Xa[ UR U 4Sj5n[ URU 4Sj5n[U[5(de[U[5(deURX4XxUR5T U'MURUU 4Sj5T U'M XPlg)zModifies all nodes in fx_module.graph.nodes which match the specified op code and target, and updates them to match the new op code and targetc>TU$Nnval_maps Z/var/www/auris/envauris/lib/python3.13/site-packages/torch/fx/passes/graph_manipulation.py+replace_target_nodes_with..#s  c>TU$rrrs rr r!$s GAJr"c>TU$rrrs rr r!+s  r"N)rgraphnodesoptargetr argskwargs isinstancetupledict create_nodename node_copy) rrrrr new_graphnoder)r*rs @rr r sI "G%% 77f  !:499&:;DT[[*>?FdE** **fd++ ++%11D$))GDM&//6JKGDM& Or"c*\rSrSr%\\S'\\S'Srg)r/ output_size total_sizerN)__name__ __module__ __qualname____firstlineno__int__annotations____static_attributes__rr"rrr/s Or"rr)returncUb[U5R"U6 URRH%nURS:Xa g[ X5UlM' g)zGiven a fx graph module, update each node with its total size (weights + bias + output) and its output_size(output). For a non-module node, the total size is the output size. return total sizeNoutput)r propagater%r&r'rr)rr)r2s rrr5sV )&&-%% 77h   +9;& r"r2chURRS5nU(d[SUS35eU$)N tensor_metazNode zQ has no tensor metadata associated with it! Check that shape propagation has run.)metaget RuntimeError)r2rCs rrrGs=))-- .K D64 5  r"cSnURS:XaT[UR55nX1RnUR 5nUHupgX'R 5- nM [ U5nURR 5n X)- nUR(a.[R"/URS9R5n O-[R"/URS9R5n X-n X-n [X5$)zGiven a node with node.dtype and node.shape, return its total size and its output size. total_size = weights + bias + output_size r call_module)dtype)r'r- named_modulesr(named_parametersnumelrshape is_quantizedtorch_empty_affine_quantizedrI element_sizetensorr) rr2total_num_of_elemssubmodule_dict submodule parameters_nameprC output_elemsize_per_elem_bytesr6r5s rrrTs  ww-i5578";;/ //1 "HE '') + #"$'K##))+K%#;; k'' ,. $ll2[5F5FGTTV$9J%3K k ..r"r)typingrrrrOtorch.fx._compatibilityrtorch.fx.graphrtorch.fx.graph_moduler torch.fx.noder r r torch.fx.passes.shape_propr __all__strr rlistTensorrrrrr"rres%,, 1 -//0 e,         - 2e,- e,AE  "*4 +=">   - "e, $ 3 - e,/ /4/J/-/r"