VhUddlZddlZddlmZmZddlmZddlmZej Zee e fe d<edZ edZ de d dfd Zd ee e fd ee e ffd Zy) N)CallableTypeVar) OrderedDict) ParamSpecsimple_call_counter_P_RlabelreturncHtj|d}|dzt|<yNr)r setdefault)r prevs B/home/dcms/DCMS/lib/python3.12/site-packages/torch/utils/_stats.py count_labelrs#  ) )% 3D!%fnctjdtjdtjdt ffd }|S)Nargskwargsr cjtvrdtj<tjdztj<|i|Sr ) __qualname__r)rrrs rwrapperzcount..wrappersM ??"5 534  0/B2??/SVW/WBOO,4"6""r) functoolswrapsrrrr )rrs` rcountrsB__R#rww#"))### Nr) collectionsrtypingrrrtyping_extensionsrrstrint__annotations__rr rrrrr%s$#'.E[-D-D-F[c*Ft_ T]*s*t*hr2v8BF#3r