o "if @sFgZeGdddZGdddZddZiZd ddZd d Zd S) c@s"eZdZUeed<dZeed<dS) LevelInfolevelTaliveN)__name__ __module__ __qualname__int__annotations__rboolr r `/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/functorch/dim/dim.pyr s rc@sdeZdZddededeffddZddZedd Z e j defd d Z ed d Z d dZ dS)DimNnamesizecCs(||_d|_d|_|dur||_dSdSN)r _size _vmap_levelr)selfr rr r r __init__s  z Dim.__init__cCsl|jdur.dt|j_tdjs0ttdjkr4tttdjs2ttdjksdSdSdSdSdS)NF) rZ_vmap_active_levels _vmap_stackr _vmap_levelsZ current_levelrZ_vmap_decrement_nestingpoprr r r __del__s  z Dim.__del__cCs|jsJ|jSr)is_boundrrr r r r s zDim.sizecCsd|jdur||_t|d|_tt|_tt|jdS|j|kr0td|d|jd|dS)NZsamezDim 'z*' previously bound to a dimension of size z$ cannot bind to a dimension of size ) rZ_vmap_increment_nestingrlenrrappendrZDimensionBindError)rrr r r r%s    cCs |jduSr)rrr r r r2s z Dim.is_boundcCs|jSrr rr r r __repr__6sz Dim.__repr__r) rrrstrUnionrrrpropertyrsetterrrr r r r r s   r cCs|jdks |jdks J|jS)N STORE_FAST STORE_NAME)opnameargval)instr r r extract_name:sr(cst}|dus J|j}|dusJ|j|j}}||f}|tvrz|ddtt|j}|j dks>|j dkrS|j |dkrGt nt fddt|<n'|j dksZJ|j }t fd d t|Dt|fd dt|<t|S) Nr#r$r)cs dS)Nrr r )ctorr r r Rs zdims..UNPACK_SEQUENCEc3s$|] }td|VqdS)r+N)r().0i)first instructionsr r Vs zdims..cstfddtDS)Nc3s.|]\}}|krt|nt|dVqdS)rN)r DimList)r/r0n) first_listr r r3Zs  z)dims....)tuple enumerater )r6namesr r r-Zs )inspect currentframef_backf_codef_lasti_cachelistdisget_instructionsr%r&r r4r7ranger)listsframeZ calling_framecodelastikeyunpackZndimsr )r,r1r6r2r r9r dimsBs,     rJcsFfdd|dur St|ttfstfdd|DS|fS)Ncs$t|tr|St|tsJ|Sr) isinstancer r)a) positionalr r convertbs z_dim_set..convertc3s|]}|VqdSrr )r/rL)rNr r r3lsz_dim_set..)rKr rr7)rMargr )rNrMr _dim_setas  rPN)r))r dataclassrr r(r?rJrPr r r r s+