Name	Size	Mode	Actions
annotations.cpython-36.pyc	12774	0644	editdlrm
frontend.cpython-36.pyc	34480	0644	editdlrm
quantized.cpython-36.pyc	18517	0644	editdlrm
supported_ops.cpython-36.pyc	8224	0644	editdlrm
unsupported_tensor_ops.cpython-36.pyc	2295	0644	editdlrm
_async.cpython-36.pyc	4056	0644	editdlrm
_builtins.cpython-36.pyc	4966	0644	editdlrm
_check.cpython-36.pyc	6101	0644	editdlrm
_freeze.cpython-36.pyc	8831	0644	editdlrm
_fuser.cpython-36.pyc	2781	0644	editdlrm
_logging.cpython-36.pyc	371	0644	editdlrm
_monkeytype_config.cpython-36.pyc	6984	0644	editdlrm
_pickle.cpython-36.pyc	835	0644	editdlrm
_recursive.cpython-36.pyc	25478	0644	editdlrm
_script.cpython-36.pyc	48582	0644	editdlrm
_serialization.cpython-36.pyc	5912	0644	editdlrm
_state.cpython-36.pyc	3703	0644	editdlrm
_trace.cpython-36.pyc	35248	0644	editdlrm
__init__.cpython-36.pyc	6548	0644	editdlrm

3 Eg��� @s�dd
lZdd
l
Z
dd
lZdd
lZdd
lZddlmZ
ddlmZ
ddl m Z mZ
ddlm Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8
ddl9m:Z:m;Z;mZ>m?Z?
ddl@mAZAmBZBmCZC
dd
lDZd ZEydd
lFZFd ZEWneGk �
rr


YnXdZHdh
ZIeJejKejLejM�
ZNd d�ZOejPdejdejdej dej5dejdejdejQdejRdejSdej dej!diZTejPdejdejdej dej5dejdejd ejQd!ejRd"ejSd#ej d$ej!d%iZUeTjVejWd&ejXd'ejYd(ejZd)ej[d*i�

eUjVejWd+ejXd,ejYd-ejZd.ej[d/i�

e
j\dbk�r�eTjVej]d2i
�

Gd3d4�d4e^�Z_Gd5d6�d6e_�Z`Gd7d8�d8e`�ZaGd9d:�d:e_�Zbd;d<�Zcd=d>�Zdd?d@�ZedAdB�ZfdCdD�ZgdcdEdF�
ZhdGdH�ZiGdIdJ�dJej�ZkdKdL�ZldddMdN�
ZmdOZndedPdQ�
ZodfdRdS�
ZpdTdU�ZqdVdW�ZrdXdY�ZsGdZd[�d[ek�ZtGd\d]�d]ek�ZuGd^d_�d_ek�Zvev�Zweu�Zxet�Zydhd`da�
Zzd
S)i�N)
� namedtuple)
�dedent)�List�Tuple),�ClassDef�Ident�Stmt�Decl�Def�Var�EmptyTypeAnnotation�Param�ExprStmt�Assign�Delete�Return�Raise�Assert� AugAssign�While�For�If�Pass�Break�Continue�Apply�Dots�Select�TrueLiteral�FalseLiteral�NoneLiteral�Starred�ListLiteral�TupleLiteral�DictLiteral�Const� StringLiteral�ListComp� Attribute�BinOp�UnaryOp� SliceExpr� Subscript� TernaryIf�With�WithItem�Property�DictComp)�get_source_lines_and_file� parse_def�make_source_context)�monkeytype_trace�get_qualified_name)�should_drop�is_static_fn�FunctionModifiersFTZ__jit�printc

Cs|jt
�
p|tkS)
N)� startswith�_reserved_prefix�_reserved_names)
�name�r?�>/usr/local/lib64/python3.6/site-packages/torch/jit/frontend.py�is_reserved_name(s
rAzfunction definitionsz for loopszdel statementszclass definitionszwith statementszraise statementsZ assertionszimport statementszglobal variableszbreak statementszcontinue statements�def�for�del�class�with�raise�assert�import�from�global�break�continuezasync function definitionszasync for loopszasync with statementsz try blocksznonlocal variablesz async defz async forz async with�try�nonlocal��zannotated assignmentsc@seZ
dZd
d�Zdd�ZdS)� FrontendErrorcCs |
|_||_
tjj|j�
|_dS)
N)�source_range�msg�torch�_C�ErrorReport�error_report)�selfrSrTr?r?r@�__init__bs

zFrontendError.__init__c

Cs|j|j
j�j�S)
N)rTrX�what�lstrip)
rYr?r?r@�__str__js
zFrontendError.__str__N)�__name__� __module__�__qualname__rZr]r?r?r?r@rRas
rRc
@seZ
dZd
S)�NotSupportedErrorN)r^r_r`r?r?r?r@rans
racseZ
dZd�f
dd� Z�ZS)�UnsupportedNodeError�c slt|�
}t
tj|d
��
}|
j|j|j|j|�}tj||j�}dj ||rP|d
nd�}t t|�j||�
dS)N�
z{} {}aren't supportedrc) �type�len�node_start_tokens�get� make_range�lineno� col_offset�pretty_node_namesr^�format�superrbrZ) rY�ctxZoffending_node�reasonZ node_typeZ range_lenrSZfeature_namerT)
� __class__r?r@rZss






zUnsupportedNodeError.__init__)
rc)r^r_r`rZ� __classcell__r?r?)
rqr@rbrs
rbc
@seZ
dZd
S)�FrontendTypeErrorN)r^r_r`r?r?r?r@rss
rscs�f
d
d�|
D�
}
t|
�
S)Nc
sg|]}
t�|
��qSr?)
�build_withitem)�.0�
i)
ror?r@� �sz#build_withitems..)
�list)ro�itemsr?)
ror@�build_withitems�s

rzcs �f
d
d�|
D�
}
tt
d|
��
S)Nc
sg|]}
t�|
��qSr?)
� build_stmt)ru�
s)
ror?r@rw�szbuild_stmts..)rx�filter)roZstmtsr?)
ror@�build_stmts�s

r~cCs�tj
|d
d�d�}t|dg�}g}x�|D]�}|d|kr(t|dj�
r(t|djd|d�d�|
d �}|djr�t|djd|d�d �|
d �nd}|jt|j �t |j �|d�||��

q(W|S)aj
Get a list of Property objects representing the properties of a class. Args: cls: The class to get properties of. self_name: The name of the class that the properties should belong to. Returns: A list of Property objects corresponding to the properties of cls. Property here refers to the subclass of TreeView. c

Ss t|t
�S)
N)� isinstance�property)
�
mr?r?r@��sz&get_class_properties..)
� predicateZ__jit_unused_properties__r
�
�__Z_getter)
� self_nameZ_setterN)�inspect� getmembers�getattrr7�fget�get_jit_def�fset�appendr0�ranger)�clsr��propsZunused_properties� properties�prop�getter�setterr?r?r@�get_class_properties�s




.
*r�csXg���
fd
d�}x@|
jD]6}t
|tj�r8|tj|�
qt
|tj�r|tj|�
qW�S)Ncs.y�j|�
|
��

Wnt
k r(


YnXdS)
N)r�ra)Zbuilder�entry)�assignsror?r@�maybe_build_assign�s


z-get_class_assigns..maybe_build_assign)�bodyr�astr�StmtBuilder�build_Assign� AnnAssign�build_AnnAssign)roZcls_astr�r�r?)r�ror@�get_class_assigns�s





r�cs�tj
��f
d
d�d�}�f
dd��
�
�fdd�|D�
}t���}t�tjjj��\}}}dj|�
}t |�
}t j|�
} t|j d d �d�
t|j d d �d�
} t|||| d�}| jd}t|t j�s�t�
t||�} t||||�| �S) Nc

s.tj
|�
stj|�
o,t�|j�o,|j�jkS)
N)r��ismethod� isfunctionr8r^�__dict__)
r�)
r�r?r@r��s

z#get_jit_class_def..)
r�c

stj
|�
ot|d
d��kS)N�__self__)r�r�r�)
�fn)
r�r?r@�is_classmethod�s
z)get_jit_class_def..is_classmethodc
s$g|]\}
}t||
�
�|�
d��qS)
)r�r�)
r�)rur>�obj)r�r�r?r@rw�sz%get_jit_class_def..rc�
r�r
F)r�r�r�r2rUrVrWZ call_stack�joinrr��parserf�splitr4r�rr�AssertionErrorr��build_class_def)r�r��methodsr��sourcelines�file_lineno�filename�sourceZ dedent_srcZpy_astZleading_whitespace_lenroZ class_astr�r?)r�r�r�r@�get_jit_class_def�s"

 



(



r�cCs4
t|�
}t
jjj|j�
}|jjd
}|r^|jjd
j }tj |�d|���
jd
}|jjd
|�
t|�
r�tj d�
} t | j�
dks�t| jd
tj�r�td|j�d|j���
�
| jd
} | j|_d|j_|j_x(|jj|jjD]}| jjd
j|_q�Wt
jjj�}d} t�
rt|�
}|j|�
} t|j|||
|| d�S) aE Build a JIT AST (TreeView) from the given function. Args: fn: A function object to compile def_name: The name to give to the resulting AST object. This is not always the same as `fn.__name__`, for example: def _forward(self): ... forward = _forward In this case, the `__name__` attribute of the function object is "_forward", but we want the result AST to have the name "forward". self_name: If this function is a method, what the type name of `self` is. r
z = zLdef unused_fn(self: Any): raise RuntimeError("Cannot call @unused methods")r�z&Expected a single top-level function: �
:N)r�� pdt_arg_types)r3rU�jit�annotationsZ get_type_liner�r�r��args�argr��insertr7rfr�FunctionDef�RuntimeErrorr�r��kwarg�vararg� kwonlyargs� annotation�_scriptZ_get_type_trace_dbr5r6Zget_args_types� build_defro)r��def_namer�r�Z parsed_def� type_lineZfn_defZarg_nameZassign_stmtZ unused_fn_defZ unused_defr�Z type_trace_dbr��qualnamer?r?r@r��s,





"







r�c
Cszt|j
d
jtj�rv|j
d
jj}
t|
tj�rv|
j}|
j}|dkrvt|tj�rv|jdkrvt|jtj �rv|jj dkrvdSdS)Nr
Z_IgnoreContextManagerr�rUTF)rry�context_exprr��Call�funcr(�attr�value�Name�id)�stmt�function� attr_name� attr_valuer?r?r@�#is_torch_jit_ignore_context_manager
s






r�c@seZ
dZd
d�ZdS)�BuildercCs0t|d
|j
jd�}|dkr&t|
|��
||
|�S)NZbuild_)r�rqr^rb)rYro�node�methodr?r?r@�__call__
s



zBuilder.__call__N)r^r_r`r�r?r?r?r@r�
s
r�cCs:|j|
j
|
j|
jtd
�
�}tt||�dd�|D�
||�S)NrEc
Ssg|]}
t|
�
�qSr?)
r)rur�r?r?r@rw%
sz#build_class_def..)rirjrkrfrr)ro�py_defr�r�r�r��
rr?r?r@r�"
s


r�c Cs�|
j}|j
|
jt|
j�
|
j|
jtd
�
�}t||
j||�}d} t|
dd�dk r\t ||
j �} t||| �} |dk }|dk r�tj j|�
}tj j| ||�} tt||�| t||��S)NrB�returns)r�rirjrfZdecorator_listrk�build_param_listr�r�� build_exprr�r rUrVZparse_type_commentZmerge_type_from_type_commentr rr~) ror�r�r�r�r�r�r�Z param_list�return_type�declZ is_methodZtype_comment_declr?r?r@r�(
s












r�zfCompiled functions can't take variable number of arguments or use keyword-only arguments with defaultsc s
|
jdk r<|
j}�j
|j|jd
|jt|j�
�}t|t��
|
jdk rx|
j}�j
|j|jd
|jt|j�
�}t|t��
t|
j �
dkr�x.|
j D]$}|dk r�t �|�j�}t|t��
q�W�
f
dd�|
jD�
}�
f
dd�|
j D�
}��fdd�|D�
} | ��fdd�|D�
7} | S)Nr�r
c
s0g|](}
|
�r&t�|
j
�
r&�|
j
ndf�qS)
N)�boolr�)rur�)
r�r?r@rwT
s
z$build_param_list..c
s0g|](}
|
�r&t�|
j
�
r&�|
j
ndf�qS)
N)r�r�)rur�)
r�r?r@rwV
s
c
 s"g|]\}
}t�|
�
d|d
��qS)F)� kwarg_only�pdt_arg_type)
�build_param)rur��arg_type)ror�r?r@rwY
s
c
 s"g|]\}
}t�|
�
d|d
��qS)T)r�r�)
r�)rur�r�)ror�r?r@rw[
s
)r�rirjrkrfr�ra�_vararg_kwarg_errr��kw_defaultsr�r�r�r�) roZpy_argsr�r��exprZ ctx_ranger�Z arg_and_typesZarg_and_types_kwonlyargs�resultr?)ror�r�r@r�B
s,


"



"











r�cCs�|
j}|j
|
j|
j|
jt|�
�}t|
d
d�dk r@t||
j�}n<|rTtt ||��
}n(|dk rt|dkrttt ||��
}nt |�
}t|t ||�|�S)Nr�rY)r�rirjrkrfr�r�r�rrrr )ro�py_argr�r�r�r>r�Zannotation_exprr?r?r@r�`
s







r�csR
td
ddg��tdddg��
��
fdd�}dd�}d d �}dd�}||
j
d jj�
\}}d|||
�}d|} | ddjdd�|D�
�
d7} ||�
\} }| | d7} tj| �
jd }|
j|_tj|�
jd }|jj|�

dt j |�
} | dj||�7} t| �

||�
}dj|�
||�
d}t |�
d k�
r:|d|}n|}tj|�
jd }|S)N� InputTyper>�ann� OutputTypec
s~g}
g}xl|D]d}|j}t
jdkr,|jj}n|jj}|jd�
\}}|dkrZ|
j�||��

|dkr|j�
||��

qW|
|fS)NrP�r��inp�out)rPr�)r��sys�version_infor�r|r�r�)r��inputs�outputsr��var_nameZvar_annZ var_decl_type)r�r�r?r@�process_ins_outsr
s



 





z6build_ignore_context_manager..process_ins_outscSs$|jj
d
d�j
dd�dt|
j�
S)N�
.�
_�
/)r��replace�strrj)ror�r?r?r@�create_unique_name_ext�
sz.create_unique_name_extc
Ss�d
}
d}t|�
dkr|
d7}
t|�
dkrDd|dj
}
||dj7}t|�
dkr�d}
|
dd jd d�|D�
�
d7}
|d jd d�|D�
�
7}|
|fS)Nrczreturn r
z -> Noner�z -> z -> Tuple�
[z, c
Ssg|] }
|
j�qSr?)
r�)ru�varr?r?r@rw�
szObuild_ignore_context_manager..build_return_ann_stmt..�
]c
Ssg|] }
|
j�qSr?)
r>)rur�r?r?r@rw�
s)rfr�r>r�)r�Zreturn_type_annZreturn_statement_strr?r?r@�build_return_ann_stmt�
s











z;build_ignore_context_manager..build_return_ann_stmtc

Ssd
jdd�|D�
�
S)Nz, c
Ssg|] }
|
j�qSr?)
r>)rur�r?r?r@rw�
szDbuild_ignore_context_manager..build_args..)
r�)
r�r?r?r@� build_args�
s
z0build_ignore_context_manager..build_argsr
Zfunc_ignore_z def �
(z, c
Ssg|]}
|
jd|
j
�qS)
z :)r>r�)rur�r?r?r@rw�
sz0build_ignore_context_manager..�
)z: passz@torch.jit.ignore z globals()["{}"] = {}ztorch.jit.frontend.{}(z = )rryr��keywordsr�r�r�r�r�� astunparseZunparserm�execrf)ror�r�r�r�r�r�r�Zignore_function_nameZignore_function_strZ return_annZreturn_stmtZignore_functionZignore_func_strZassign_str_lhsZassign_str_rhsZ assign_str� assign_astr?)r�r�r@�build_ignore_context_managern
s4










r
c
Cs*|dkriStj
|�
}
d
d�|
jj�D�
S)Nc
Ss&i|]\}
}|jt
jjk r|j|
�qSr?)�defaultr�� Parameter�empty)ru�
k�
vr?r?r@� �
s
z$get_default_args..)r�� signature� parametersry)r�r
r?r?r@�get_default_args�
s

 r
c
s(tj
��f
d
d�d�}
dd�|
D�
}|S)aC
Get default arguments for all methods in a class (except for static methods). Args: cls: type - The class type to inspect for default arguments. Returns: A Dict[str, Dict[str, Any]] which maps each method name to a Dict[str, Any] that maps each argument name to its default value. c

s.tj
|�
stj|�
o,t�|j�o,|j�jkS)
N)r�r�r�r8r^r�)
r�)
r�r?r@r��
s

z,get_default_args_for_class..)
r�c
Ssi|]\}
}t|�
|
�qSr?)
r
)ru�method_nameZmethod_implr?r?r@r
�
sz.get_default_args_for_class..)r�r�)r�r��defaultsr?)
r�r@�get_default_args_for_class�
s 

r
c@seZ
dZed
d��
ZdS)�WithItemBuildercCsX|
jj
}|
jj}|tttj�
}|
j}|j|||�}t |t ||
j�|rRt ||�nd�S)
N)r�rjrkrfrlr�r.Z optional_varsrir/r�)ro�itemrj�start�endZop_varsr�r?r?r@rt�
s



zWithItemBuilder.build_withitemN)r^r_r`�staticmethodrtr?r?r?r@r
�
s
r
c@s
eZ
dZejd
ejdejdejdejdej dej dejdejd ej d ejdiZedd ��
Zedd��
Zedd��
Zedd��
Zedd��
Zedd��
Zedd��
Zedd��
Zedd��
Zedd��
Zed d!��
Zed"d#��
Zed$d%��
Zed&d'��
Zed(d)��
Zed*d+��
Z d,S)-r��
+�
-�
*r��
%�
|�
&�
^z<>z**cCs(|
j}|j
jd
krdStt||��
SdS)N�Str)r�rqr^rr�)ror�r�r?r?r@� build_Exprs
zStmtBuilder.build_Exprcs*t�|
j
�}�f
d
d�|
jD�
}t||�S)Nc
sg|]}
t�|
��qSr?)
r�)ru�
x)
ror?r@rwsz,StmtBuilder.build_Assign..)r�r��targetsr)ror��rhs�lhsr?)
ror@r�s

zStmtBuilder.build_AssigncCsJ|
jdkrt
||
d
d��
t||
j�}t||
j�}t||
j�}t|g
||�S)Nzwithout assigned value)
rp)r�rbr��targetr�r)ror�r
r!
Zthe_typer?r?r@r�s




zStmtBuilder.build_AnnAssigncs6�j|
j
|
j|
jtd
�
�}t|�f
dd�|
jD�
�S)NrDc
sg|]}
t�|
��qSr?)
r�)rur"
)
ror?r@rw!sz,StmtBuilder.build_Delete..)rirjrkrfrr
)ror�r�r?)
ror@�build_DeleteszStmtBuilder.build_DeletecCs<|j|
j
|
j|
jtd
�
�}t||
jdkr.dn t||
j��S)N�return)rirjrkrfrr�r�)ror�r�r?r?r@�build_Return#s
zStmtBuilder.build_ReturncCs2|j|
j
|
j|
jtd
�
�}t||
j�}t||�S)NrG)rirjrkrfr��excr)ror�r�r�r?r?r@�build_Raise(s

zStmtBuilder.build_RaisecCsN|j|
j
|
j|
jtd
�
�}t||
j�}|
jdk r>t||
j�nd}t|||�S)NrH)rirjrkrfr��testrTr)ror�r�r(
rTr?r?r@�build_Assert.s


zStmtBuilder.build_AssertcCsft||
j
�}t||
j�}t|
j�
}|tjkr8tj|}n"tt||j �j d
dd�d|j��
t|||�S)N�
=r�r
)
�offsetsz+unsupported kind of augumented assignment: �����)r,
r
) r�r"
r�re�opr�� augassign_mapra�find_beforer�r
r^r)ror�r!
r
r-
�op_tokenr?r?r@�build_AugAssign5s






zStmtBuilder.build_AugAssigncCsH|
jrt
dd
��
|j|
j|
j|
jtd�
�}t|t||
j�t ||
j ��S)Nz-else branches of while loops aren't supported�while)�orelserarirjrkrfrr�r(
r~r�)ror�r�r?r?r@�build_WhileBs 


zStmtBuilder.build_WhilecCsV|j|
j
|
j|
jtd
�
�}|
jr,t|d��
t|t||
j�g
t||
j �g
t ||
j��S)NrCz+else branches of for loops aren't supported)rirjrkrfr3
rarr�r"
�iterr~r�)ror�r�r?r?r@� build_ForLs



zStmtBuilder.build_ForcCsB|j|
j
|
j|
jtd
�
�}t|t||
j�t||
j�t||
j ��S)N�if) rirjrkrfrr�r(
r~r�r3
)ror�r�r?r?r@�build_IfVs


zStmtBuilder.build_IfcsZ�j|
j
|
j|
jtd
�
�}|
jr,t|d��
�f
dd�|
jD�
}ttt t |d
��
|g��
S)Nr:z?print statements with non-default destinations aren't supportedc
sg|]}
t�|
��qSr?)
r�)ru�val)
ror?r@rwbsz+StmtBuilder.build_Print..)rirjrkrf�destra�valuesrrrr)ror�r�r�r?)
ror@�build_Print]s 



zStmtBuilder.build_PrintcCs$|j|
j
|
j|
jtd
�
�}t|�
S)N�pass)rirjrkrfr)ror�r�r?r?r@� build_Passes
zStmtBuilder.build_PasscCs$|j|
j
|
j|
jtd
�
�}t|�
S)NrL)rirjrkrfr)ror�r�r?r?r@�build_Breakjs
zStmtBuilder.build_BreakcCs$|j|
j
|
j|
jtd
�
�}t|�
S)NrM)rirjrkrfr)ror�r�r?r?r@�build_Continueos
zStmtBuilder.build_ContinuecCs`|j|
j
|
j|
jtd
�
�}t|
�
rDts0td�
�
t||
�}t||�St |t ||
j�t||
j ��S)NrFz�torch.jit._IgnoreContextManager requires installing Python library `astunparse`, please install it in your Python environment)rirjrkrfr��_IS_ASTUNPARSE_INSTALLEDr�r
r{r.rzryr~r�)ror�r�r
r?r?r@� build_Withts





zStmtBuilder.build_WithN)!r^r_r`r��Add�Sub�Mult�Div�Mod�BitOr�BitAnd�BitXor�LShift�RShift�Powr.
r
r
r�r�r#
r%
r'
r)
r1
r4
r6
r8
r<
r>
r?
r@
rB
r?r?r?r@r��
s6









     r�c@s�
eZ
dZejd
ejdejdejdejdej dej dejdejd ej d ejdejdiZd eej<ejdejdejdiZejdejdiZejdejdejdejdejdejdejdej dej!dej"di Z#e$dd��
Z%e$dd��
Z&e$d d!��
Z'e$d"d#��
Z(e$d$d%��
Z)e$d&d'��
Z*e$d(d)��
Z+e$d*d+��
Z,e$d,d-��
Z-e$d.d/��
Z.e$d0d1��
Z/e$d2d3��
Z0e$d4d5��
Z1e$d6d7��
Z2e$d8d9��
Z3e$d:d;��
Z4e$dd?��
Z6e$d@dA��
Z7e$dBdC��
Z8e$dDdE��
Z9dFS)G�ExprBuilderr
r
r
r�z**r
z//r
r
r
z<>�
@�not�
~�and�orz==z!=z<=�
=�
>�iszis not�inznot incszt||
j
�}|jjd
�
��f
dd�}|j�jd}x||�
tjkrL|d7}q4W|t|
j �
}|j ||�}t|t||
j ��S)Nzutf-8c

st�|�
S)
N)
�chr)
�index)
r�r?r@�get_char�s
z-ExprBuilder.build_Attribute..get_charr�) r�r�r��encoder�r
�string� whitespacerfr��make_raw_rangerr)ror��baserZ
� start_pos�end_posZ name_ranger?)
r�r@�build_Attribute�s




zExprBuilder.build_Attributecs�t�|
j
�}�f
d
d�|
jD�
}t|
d�rP|
jrPt�|
j�}|t|j�|�g
7}g}xJ|
jD]@}t�|j�}|j s�t |j�d��
|jtt |j�|j �|��

q\Wt|||�S)Nc
sg|]}
t�|
��qSr?)
r�)rur�)
ror?r@rw�sz*ExprBuilder.build_Call..�starargsz&keyword-arg expansion is not supported)r�r�r��hasattrrc
r!r�r�r�r�rar�r(rr)ror�r�r�Zstararg_expr�kwargs�kwZkw_exprr?)
ror@� build_Call�s









zExprBuilder.build_CallcCs |j|
j
|
j|
jd
�}t|�
S)NrP)rirjrkr)ror�r�r?r?r@�build_Ellipsis�s
zExprBuilder.build_EllipsiscCs�|j|
j
|
j|
jt|
j�
�}|
jjt�
r8t|d
t��
|
jdkrJt|�
S|
jdkr\t |�
S|
jdkrnt |�
S|
jdkr�t|�
Stt ||
j��
S)Nz=names of variables used in JIT-ed functions can't start with �True�False�None�Ellipsis)rirjrkrfr�r;r<rarrr rrr)ror�r�r?r?r@� build_Name�s











zExprBuilder.build_NamecCs�|j|
j
|
j|
jtt|
j�
�
�}|
jd
kr4t|�
S|
jdkrFt|�
S|
jdkrXt|�
S|
jt krjt |�
Stdt|
j�
�
�
dS)NTFz!Name constant value unsupported: )rirjrkrfr�r�rrr rl
r� ValueError)ror�r�r?r?r@�build_NameConstant�s"







zExprBuilder.build_NameConstantcCs�t||
j
�}t||
j�}t|
j�
}|tjkrV|jrV|j|j �j |j �j�}t|d
��
t jj|�
}|dkr�|j|j �j |j �j�}t|d|j��
t|||�S)Nz�Division of ints in TorchScript uses Python 3 true division semantics. Please put `from __future__ import division` at the top of your filezunsupported binary operator: )r��left�rightrer-
r�rF
Zuses_true_divisionr^
r�r
r
rRrN
� binop_maprhrar^r))ror�r!
r
r-
� err_ranger0
r?r?r@�build_BinOp�s







zExprBuilder.build_BinOpcCsft||
j
�}t|
j�
}tjj|�
}|dkr>t|
j�d
|j ��
|j |
j|
j|
jt |�
�}t|||�S)Nzunsupported unary operator: )r��operandrer-
rN
�unop_maprhrar�r^rirjrkrfr*)ror��sub_exprr-
r0
r�r?r?r@� build_UnaryOp�s





zExprBuilder.build_UnaryOpcs�t|
j
�
d
kr$tdtt|
j
�
�
�
�
�f
dd�|
j
D�
}t|
j�
}tjj|�
}|dkr��j |dj �j|dj �j�}t |d|j��
|d}x |dd�D]}t|||�}q�W|S)N�z.expected at least 2 values in BoolOp, but got c
sg|]}
t�|
��qSr?)
r�)rurw
)
ror?r@rw sz,ExprBuilder.build_BoolOp..r
r�zunsupported boolean operator: )rfr;
r�r�rer-
rN
� boolop_maprhr^
r�r
r
rar^r))ror�� sub_exprsr-
r0
rs
r!
r
r?)
ror@�build_BoolOp s










zExprBuilder.build_BoolOpcCs$tt
||
j�t
||
j�t
||
j��S)
N)r-r�r(
r�r3
)ror�r?r?r@�build_IfExps

zExprBuilder.build_IfExpcs��f
d
d�|
jg
t
|
j�
D�
}d}x�t||
j|dd��D]�\}}}t|�
}tjj|�
}�j |j �j|j �j�} |dkr�t | d|j��
|tjkr�td||�} t| d| �}nt|||�}|dkr�|}q.r�z!unsupported comparison operator: rW
rP
rR
)rp
rxZcomparators�zip�opsrerN
� cmpop_maprhr^
r�r
r
rar^r��NotInr)r*)ror�Zoperandsr�r!
Zop_r
r-
r0
r�Zin_exprZcmp_exprr?)
ror@� build_Compares 

"




 


zExprBuilder.build_Comparec s�
d
d��
dd����
fdd�}t�|
j
�}t|
j�
}|tjkr�t|
jj
tj�r��f
dd�|
jj
jD�
}|s��j |
j |
jj
j|
jj
jd �}t|g�}|j |�

t||�St|t�|
jj
�g
�S�
n|tjkr�t|�
�||
j�g
�S|tjk�
rt||�||
j��Stjd k�
r�|tjk�
r�g}xD|
jjD]8}t|tj��
rN|j �
�||��

n|j t�|��

�
q(W|�
s��j |
j |
jj|
jjd �}t|g�}|j |�

t||�St|t�|
j�g
�St|j�d��
dS)NcSs`|jdk rt
||j�nd}|jdk r0t
||j�nd}|jdk rJt
||j�nd}t|
j�|||�S)
N)�lowerr��upper�stepr+r�)ror_
Z slice_exprr�
r�
r�
r?r?r@�build_SliceExpr9s



z4ExprBuilder.build_Subscript..build_SliceExprcSs(t|j
tj�rt|
j�d
��
t||j
�S)Nz9slicing multiple dimensions with tuples not supported yet)rr�r�rrar�r�)ror_
� index_exprr?r?r@�build_Index?s


z0ExprBuilder.build_Subscript..build_Indexcs�g}x�|jD]z}t
|�
}|tjkr6|j�||
|��

q|tjkrT|j�
||
|��

q|tjkrr|jt|
j��
�

qt |
j�d
j |�
��
qW|S)Nz1slicing multiple dimensions with {} not supported)Zdimsrer��Indexr��Slicerl
rr�rarm)ror_
Zextslicer{
r��sub_type)r�
r�
r?r@�build_ExtSliceFs











z3ExprBuilder.build_Subscript..build_ExtSlicec
sg|]}
t�|
��qSr?)
r�)rur�
)
ror?r@rw[sz/ExprBuilder.build_Subscript..ry
rP� zellipsis is not supported)rPr�
)r�r�re�slicer�r�
rr�eltsrirjrkr#r�r,r�
�ExtSlicer�r�rar�) ror�r�
r_
r�
�indicesr��tupr�
r?)r�
r�
ror@�build_Subscript7sH























zExprBuilder.build_Subscriptcs.t�j
|
j|
j|
jd
��f
dd�|
jD�
�S)Nr�c
sg|]}
t�|
��qSr?)
r�)rur~
)
ror?r@rw�sz*ExprBuilder.build_List..)r"rirjrkr�
)ror�r?)
ror@� build_List�s
zExprBuilder.build_Listcs.t�j
|
j|
j|
jd
��f
dd�|
jD�
�S)Nr�c
sg|]}
t�|
��qSr?)
r�)rur~
)
ror?r@rw�sz+ExprBuilder.build_Tuple..)r#rirjrkr�
)ror�r?)
ror@�build_Tuple�s
zExprBuilder.build_Tuplecs`�j|
j
|
j|
jd
�}|
jr4|
jdr4t|d��
t|�f
dd�|
jD�
�f
dd�|
jD�
�S)Nr�r
z1Dict expansion (e.g. `{**dict}`) is not supportedc
sg|]}
t�|
��qSr?)
r�)rur~
)
ror?r@rw�sz*ExprBuilder.build_Dict..c
sg|]}
t�|
��qSr?)
r�)rur~
)
ror?r@rw�s)rirjrk�keysrar$r;
)ror�r�r?)
ror@� build_Dict�s 



zExprBuilder.build_DictcCs0t|
j
�
}|j|
j|
j|
jt|�
�}t||�S)
N)r��
nrirjrkrfr%)ror�r�r�r?r?r@� build_Num�s

zExprBuilder.build_NumcCs�|
j}|dkst
|t�r$tj||
�St
|tttf�r@tj||
�St
|t �rVtj ||
�St
|tt�
�rptj ||
�S|j|
j|
j|
jtt |�
�
�}t|d
��
dS)Nz Unknown Constant expression type)r�rr�rN
ro
�int�float�complexr�
r�� build_Strrerl
rh
rirjrkrfrR)ror�r�Zerror_ranger?r?r@�build_Constant�s







zExprBuilder.build_ConstantcCs,t|
j
�
}|j|
j|
j|
jd
�}t||�S)Nr�)r�r|rirjrkr&)ror�r�r�r?r?r@r�
�s

zExprBuilder.build_StrcCs�d
}g}x�|
jD]�}|j
|j|j|jd�}t|tj�r||jdkrLt|d��
|j dk r`t|d��
|d7}|j t||j��

qt|tj �r�||j7}qt|d��
qW|j
|
j|
j|
jd�}ttt||�t|d��|g�S) Nrcr�z%Don't support conversion in JoinedStrz%Don't support formatting in JoinedStrz{}zUnsupported value in JoinedStrrmr,
)r;
rirjrkrr��FormattedValue� conversionra�format_specr�r�r�r
r|rrr&r)ror�r|r�r�r�r?r?r@�build_JoinedStr�s 












zExprBuilder.build_JoinedStrcCs�|j|
j
|
j|
j�}t|
j�
d
kr,t|d��
t|
jdj�
dkrJt|d��
t||
j�}t||
jdj �}t||
jdj �}t||||�S)Nr�z.Only a single generator is currently supportedr
z'Comprehension ifs are not supported yet)rirjrkrf� generatorsra�ifsr��eltr"
r5
r')ror�r�Zelt_expr�target_expr� iter_exprr?r?r@�build_ListComp�s




zExprBuilder.build_ListCompcCs�|j|
j
|
j|
j�}t|
j�
d
kr,t|d��
t|
jdj�
dkrJt|d��
t||
j�}t||
j �}t||
jdj �}t||
jdj�}t|||||�S)Nr�z.Only a single generator is currently supportedr
z'Comprehension ifs are not supported yet) rirjrkrfr�
rar�
r��keyr�r"
r5
r1)ror�r�Zkey_exprZ value_exprr�
r�
r?r?r@�build_DictComp�s





zExprBuilder.build_DictCompcCs*|j|
j
|
j|
jd
�}t|t||
j��S)Nr�)rirjrkr!r�r�)ror�r�r?r?r@� build_Starred�s
zExprBuilder.build_StarredN):r^r_r`r�rC
rD
rE
rF
rM
rG
�FloorDivrI
rJ
rH
rK
rL
rr
�MatMult�Not�USub�Invertrv
�And�Orrz
�Eq�NotEq�LtE�Lt�GtE�Gt�Is�IsNot�Inr�
r�
r
rb
rg
rh
rm
ro
rt
rx
r|
r}
r�
r�
r�
r�
r�
r�
r�
r�
r�
r�
r�
r�
r?r?r?r@rN
�sb










 











 KrN
cCs8|jd|
�j
|�
}|j||d
|t|�
|d�S)Nr
r�)r��rindexr^
rf)ro�pos�substrr+
�new_posr?r?r@r/
�s

r/
)rPrQ)NF)NN)
N)
N�r
r
)
r�
){rUr�r�r�r\
�collectionsr�textwrapr�typingrrZtorch._C._jit_tree_viewsrrrr r rrr rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1Ztorch._sourcesr2r3r4Ztorch.jit._monkeytype_configr5r6Ztorch._jit_internalr7r8r9Ztorch.jit.annotationsrA
r
�ImportErrorr<r=�set�ascii_lowercase�ascii_uppercase�digitsZ_identifier_charsrAr��Import� ImportFrom�Globalrlrg�update�AsyncFunctionDef�AsyncFor� AsyncWith�Try�Nonlocalr�r�� ExceptionrRrarbrsrzr~r�r�r�r�r��objectr�r�r�r�r�r�r
r
r
r
r�rN
r�r{rtr/
r?r?r?r@�
s�







�











































  ! 1   T l