IЦi fSSKJrJrJrJrJr SSKrSSKJr SSK J r SSK J r SSK JrJrJr 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))AnyDictList 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 a/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/fx/passes/graph_manipulation.py+replace_target_nodes_with..#s  c>TU$rrrs r!r"r#$s GAJr$c>TU$rrrs r!r"r#+s  r$N)r graphnodesoptargetr argskwargs isinstancetupledict create_nodename node_copy) rrrrr new_graphnoder+r,r s @r!rrsI "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$r!rr/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+r4s r!rr5sV )&&-%% 77h   +9;& r$r4chURRS5nU(d[SUS35eU$)N tensor_metazNode zQ has no tensor metadata associated with it! Check that shape propagation has run.)metaget RuntimeError)r4rEs r!rrGs=))-- .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_quantizedrK element_sizetensorr) rr4total_num_of_elemssubmodule_dict submodule parameters_nameprE output_elemsize_per_elem_bytesr8r7s r!rrTs  ww-i5578";;/ //1 "HE '') + #"$'K##))+K%#;; k'' ,. $ll2[5F5FGTTV$9J%3K k ..r$r)typingrrrrrrQtorch.fx._compatibilityrtorch.fx.graphr torch.fx.graph_moduler torch.fx.noder r r torch.fx.passes.shape_propr__all__strrrTensorrrrrr$r!rfs%88 1 -//0 e,         - 2e,- e,AE  "*4 +=">   - "e, $ 3 - e,/ /4/J/-/r$