L i "ddlmZddlZddlmZddlmZddlmZddl m Z m Z m Z m Z ddlmZmZmZmZmZmZddlmZerdd lmZdd Zd d d  ddZd d d  ddZd dddZddZd d d  ddZd dddZy)) annotationsN) TYPE_CHECKING) assert_never)cpp)ArgNameBindingCType NamedCType)ArgumentFunctionSchemaReturn SelfArgumentTensorOptionsArgumentsType) concatMap)Sequencec,tj|SN)rname)funcs ]/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/torchgen/api/dispatcher.pyrr(s 88D>FT)remove_non_owning_ref_typessymintc6tj|||||S)N)mutablebindsrr)rargumenttype_type)trrrrs rrr,s&   $?  rcJt|j|j|||S)N)rrrr)rtypeis_write)arrrs r argument_typer$As(   $?  rrc0tj||SNr%)r returns_type)rsrs rr(r(Qs   Bv ..rc  dd}tt|tj|jj |jj |jjS)Nct|tr|gSt|tr |jgSt|tr.|j |j |j|jgSt|yr) isinstancer rargumentrdtypelayoutdevice pin_memoryr)r#s r to_argumentz"jit_arguments..to_argumentWs\ a "3J < (JJ<  1 2GGQXXqxx> > Or)r#z0Argument | TensorOptionsArguments | SelfArgumentreturnlist[Argument])listr itertoolschain arguments positional kwarg_onlyout)rr2s r jit_argumentsr<Vsb  ;      OO))4>>+D+DdnnFXFX   rc`tt||j|||j|S)N)rrr)nctyperr-)rr$r)r#rrs rr-r-ms5  &&(C  VV  rcTt|Dcgc]}t||c}Scc}wr')r<r-)rrr#s rr8r8|s"0=d0C D1HQv & DD Ds%)rr r3str) rrrboolrrrrArrAr3r ) r#r rrrrArrAr3r )r)zSequence[Return]rrAr3r )rr r3r4)r#r rrArrAr3r)rr rrAr3z list[Binding]) __future__rr6typingrtyping_extensionsr torchgen.apirtorchgen.api.typesrrr r torchgen.modelr r r rrrtorchgen.utilsrcollections.abcrrrr$r(r<r-r8rrrKs" *BB%(&).     "&   2).    "&      :>/ 09>d  15 GK   7;Er