o <&i%@sJddlmZmZddlmZddlmZmZmZddl m Z m Z m Z m Z mZmZmZdZdedefdd Zdedeeeeeffd d Zd d dedededeeeeeeeffddZdedededeeeeeffddZdededededeeeeeff ddZdedede dedeeeeeff ddZdS))ListTuple)cpp)BindingCppSignatureGroupCType)ArgumentBaseTyBaseTypeListTypeNativeFunction OptionalTypeTypez freturncCs |jjS)N)funcnameZunambiguous_name)rr@C:\wamp64\www\opt\env\Lib\site-packages\torchgen/api/unboxing.pyrds rc stj|ddfddttDdg}g}D]2}t|jts.t d||j}t |j |j |j d\}}}}|||||||q||fS)NF)methodc s.g|]}d|jd|dtdqS)z c10::IValue z = std::move(peek(stack, , z));)rlen).0iargsrr ps z%convert_arguments..z7Unexpected argument type, expecting `Argument` but got mutable)rZfrom_native_functionZmost_faithful_signature argumentsranger isinstanceargumentr Exceptionargumenttype_ivalue_converttyperZis_writeextendappend with_name) rZ code_listZ binding_listargr#Z unboxed_name_codedeclrrrconvert_argumentsis2      r.Frtarg_namercCstj|||ddj}t|tr|d}t|||d\}}n4t|tr3|d}t||||d\}}nt|trH|d}t ||||d\}}n t d|d |||||fS) NF)r/rZbindsZsymint_baser0out_namectypeZ_opt_out)r0r3r/r4Z _list_outzCannot handle type z . arg_name: ) rZargumenttype_typer&r"r _gen_code_base_typer _gen_code_optional_typer _gen_code_list_typer$)r/r0rr4r3r,r-rrrr%s8        r%r3r4c Cs2|jddd|d|d|jdddggfS)NTZ strip_ref  = z.to();)cpp_typer2rrrr5s*r5c Cs|d}t|j|\}}}}d|d|d|jddd|d|d |d |d t|d |d |jddd |d|d |jdddd|fS)NZ_opt_inz c10::optional z_opt = z.toOptional(); Tr8r9z; if (z*_opt.has_value()) { const c10::IValue r:z_opt.value();  (z); } else { z(); }  )r%elemr; connectorjoinsplit) r0r3r/r4in_nameres_namer+res_coder-rrrr6sD     r6c Cs|d}|d}d|d|dg}t|j|\}}} } t|jtrS|jjtjkrS|jrS|d|j ddd |d |j ddd |jd |d d|| fSt|jt r|d|j ddd |d|d|dt | d|d|d d|| fS|d} | d|j ddd| d|d|d|dt | d| d|d|j ddd |d| d  d|| fS)NZ_list_in_elemzconst c10::List r:z .toList();r>Tr8r9z = as_array(z); z; for (c10::IValue z: z) { r<z .push_back(z); } Z_vecz std::vector ;z for (c10::IValue z); } r=)r%r?r"r rr boolsizer'r;rBr r@rAr() r0r3r/r4rCZ elem_namer,rDZ res_ctyperEr-Zvec_namerrrr7s     $     r7N)typingrrZ torchgen.apirZtorchgen.api.typesrrrZtorchgen.modelrr r r r r rr@strrr.rHr%r5r6r7rrrrs` $["" "