PhUddlZddlmZmZmZmZmZmZmZddl Z ddl m Z ddl m Z mZmZmZmZmZmZmZiZeeeefed<e j0j2dZde j6j8fdZ ddddd d d eed eed eedeedeedeededdfdZde de j6j@defdZ!de de j6j@dee jDjFfdZ$de de j6j@defdZ%de de j6j@dee jLfdZ'y)N)AnyDictIterableListOptionalTupleType)ExportedProgram)_register_pytree_nodeContextDumpableContext FlattenFuncFromDumpableContextFnToDumpableContextFn tree_flatten UnflattenFunc(SERIALIZED_DATACLASS_TO_PYTHON_DATACLASSc|t|\}}t|j|j|j|S)N)range_constraintsequality_constraints)r"_check_input_constraints_for_graphgraphrr)selfargskwargs flat_args_s ^C:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/_export/utils.py!_check_input_constraints_pre_hookrsE%LIq  - 00!66   rc*ddlmfd}|S)Nr)'_AddRuntimeAssertionsForConstraintsPassctjjitjj}jD]H}|j dk7r|j j|j}|j|_ J|j jd|}|J|j}||y)N placeholder) torchfx GraphModuleGraphnodesoprr$namemetaoutput graph_module) r_assertion_graphpnew_p_assertion_graph_resr"rrrs rinnerz1_check_input_constraints_for_graph..inner)s 88//EHHNN4DEAtt}$$**66qvv>EEJ  %%b) F     $////<<$r )@torch._export.passes.add_runtime_assertions_for_constraints_passr")rrrr4r"s``` @rrr"s " Lr F)serialized_type_nameto_dumpable_contextfrom_dumpable_contextreturn_none_fieldscls flatten_fn unflatten_fnr6r7r8r9returnc tjs Jdjdj t <dt dt tt tfffd }dtt dtdt fd}dtdtf fd } d tdtfd } ||n|}||n|}|du|duz rtd d ||n| }||n| }t|||||y)Nz7Only dataclasses can be registered with this function: .objr=cg}g}g}tj|D]\}|jt||j}}|r#|j ||j |L|j |^|||ffS)N) dataclassesfieldsr,getattrappend) r@ flattened flat_names none_namesfr,valr:r9s rdefault_flatten_fnz=register_dataclass_as_pytree_node..default_flatten_fnNs   ##C(AQVV 4#D"4  %!!$'!!$' )3 J777r valuescontextc p|\}}}|ditt|||Dcic]}|dc}Scc}w)Nr%)dictzip)rLrMtyprGrHks rdefault_unflatten_fnz?register_dataclass_as_pytree_node..default_unflatten_fn[sC&-#ZTT#j&12T 6S 1q$w 6STT6Ss 3 c|d|dfS)Nr%)rMserialized_types rdefault_to_dumpable_contextzFregister_dataclass_as_pytree_node..default_to_dumpable_context_sWQZ88r dumpable_contextc,t|d|d|dfS)NrrUrV)r)rYs rdefault_from_dumpable_contextzHregister_dataclass_as_pytree_node..default_from_dumpable_contextbs* 45Ea5H I Q  Q   r z7Both to_dumpable_context and from_dumpable_context for z must be None or registered.)r6r7r8) rB is_dataclass __module__ __qualname__rrrrr rr ValueErrorr ) r:r;r<r6r7r8r9rKrSrXr[rWs ` ` @r!register_dataclass_as_pytree_noder`=sW  # #  G @FG (#*:*:);r~sCCC )   BD($sDI~*>C 88>>:)-,0G +/9==A$G cG%G=)G #3- G ""56 G$$9:GG GTEoEUXX]]EtE    ((-- ehh  !  BBehhmmBB    ((-- ell r