/usr/local/lib64/python3.6/site-packages/torch/jit/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/jit/__pycache__/annotations.cpython-36.pyc (12774B)
3
Eg> @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZddlmZm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZmZmZ ddlmZmZmZ ddlmZ d dl 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+m,Z,m-Z-m.Z. d dl/m0Z0 d d l1m2Z2 d d
l3m4Z4 ej5j6j7 r:dd lm8Z8m9Z9 d d l m:Z: G d
d de;Z
dd Z?dd Z@dd ZAdd ZBdd ZCd d ZDd d ZEd!d" ZFe4ejG d#d$d%ZHd&d' ZId(d) ZJd*d+ ZKd,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@ddddd d d d"d)d+g ZLdS )A N )ListTupleis_tupleis_listDictis_dictOptional is_optional_qualified_nameAnyFuture is_future
is_ignored_fnUnionis_union)BroadcastingList1BroadcastingList2BroadcastingList3 )_get_script_class)
TensorType TupleType FloatTypeIntType ComplexTypeListType
StringTypeDictTypeBoolType OptionalType
InterfaceTypeAnyTypeNoneType
DeviceObjType
StreamObjType
FutureTypeEnumType UnionType)dedent)get_source_lines_and_file)Type)RRefis_rref)RRefTypec @ s e Zd Zdd Zdd ZdS )Modulec C s || _ || _d S )N)namemembers)selfr0 r1 r3 A/usr/local/lib64/python3.6/site-packages/torch/jit/annotations.py__init__ s zModule.__init__c C s< y
| j | S tk
r6 td| j d| d Y nX d S )NzModule z has no member called )r1 KeyError RuntimeErrorr0 )r2 r0 r3 r3 r4 __getattr__ s
zModule.__getattr__N)__name__
__module__ __qualname__r5 r8 r3 r3 r3 r4 r/ s r/ c
@ sL e Zd Zeddejiejeddeieeee e
e d Z dd Z
dd Zd
S ) EvalEnvtorchTensortypingr ) r= r> r? r r r r r r
c C s || _ tjjj r t| jd< d S )Nr, )rcbr= distributedrpc is_availabler, env)r2 r@ r3 r3 r4 r5 4 s zEvalEnv.__init__c C s4 || j kr| j | S | jd k r(| j|S tt|d S )N)rD r@ getattrbuiltins)r2 r0 r3 r3 r4 __getitem__9 s
zEvalEnv.__getitem__N)r9 r: r; r/ r= r> r r r r r r
rD r5 rG r3 r3 r3 r4 r< ' s r< c C s t | |}|d k r2|r2|\}}|dd }||f}|d krd\}}y"tdjt| d }t|}W n tk
rx Y nX |d k rt|||}|S )Nr r )NN)try_real_annotationsr) joinr*
get_type_line TypeErrorparse_type_line) fnr@ locZ is_method signatureZ param_types return_type type_linesourcer3 r3 r4
get_signature@ s
rT c C s t j| pt j| S )N)inspect
isfunctionismethod) the_callabler3 r3 r4 is_function_or_methodZ s rY c C s: t | rt| dr| j} t | r2tj| jd k S dS d S )N__call__F)rY hasattrrZ rU getfullargspecvarargs)rX r3 r3 r4 is_vararg` s
r^ c C s` t | r$t| dr$t | jr$| j} t | rJt| r>tj| } tj| jS dd t|D S d S )NrZ c S s g | ] }t |qS r3 )str).0ir3 r3 r4
x s z#get_param_names..) rY r[ rZ r rU unwrapr\ argsrange)rN Zn_argsr3 r3 r4 get_param_namesl s
rf c
C s yt djt| d }W n ttfk
r2 d S X |d kr@d S tj|}t|jdkrt |jd tj
rt j j
j|d|jd j dt|jdkst |jd tj rt j j
j|dd S )NrH r r zCannot instantiate class 'z' in a script functionz$Expected a single top-level function)r) rJ r* rL IOErrorastparselenbody
isinstanceClassDefr= jitZfrontendZ
FrontendErrorr0 FunctionDef)rN rO rS Zpy_astr3 r3 r4 check_fn{ s
"rp c s t | \}}yt|i t|}W n0 ttfk
rP } ztd|W Y dd}~X nX t|tsb|f}yt|i t|}W n0 ttfk
r } ztd|W Y dd}~X nX fdd|D }|t| fS )zParses a type annotation specified as a comment.
Example inputs:
# type: (Tensor, torch.Tensor) -> Tuple[Tensor]
# type: (Tensor, Tuple[Tensor, Tensor]) -> Tensor
z6Failed to parse the argument list of a type annotationNz4Failed to parse the return type of a type annotationc s g | ]}t | qS r3 ) ann_to_type)r` ann)rO r3 r4 rb s z#parse_type_line..) split_type_lineevalr< NameError SyntaxErrorr7 rl tuplerq ) rR r@ rO Z arg_ann_strZ ret_ann_strZarg_anneZret_ann arg_typesr3 )rO r4 rM s
rM c sl d | j d}dd t|D }tt fdd|}tjdttfdd|}t|d krtjd
ttfd d|}t|d krtd t|d d d
d d dS t|dkr|d d j S d}g }x6|D ].\}}d|kr||f}P q |kr|j
| qW |dkr.c s | d kS )Nr r3 )r| ) type_commentr3 r4 s z get_type_line..z!# type:\ ignore(\[[a-zA-Z-]+\])?$c s j | d S )Nr )search)r| ) type_patternr3 r4 r~ s r z&#[ ]*type[ ]*(?!: ignore(\[.*\])?$):c s j | d S )Nr )r )r| )wrong_type_patternr3 r4 r~ s z The annotation prefix in line z* is probably invalid.
It must be '# type:'zl
See PEP 484 (https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)z
for examplesNr z# type: (...) -> z_Return type line '# type: (...) -> ...' not found on multiline type annotation
for type lines:
c S s g | ] }|d qS )r r3 )r` r| r3 r3 r4 rb s zl
(See PEP 484 https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)c s" | | j t d }|j S )N)findrj strip)r| Z item_type)r} r3 r4 get_parameter_type s z)get_type_line..get_parameter_typez, z...)split enumeratelistfilterrecompilerj r7 r_ r appendrJ mapreplace) rS linesZ
type_linesZwrong_type_linesZ return_lineZparameter_type_linesr{ r| r typesZparameter_typesr3 )r} r r r4 rK s@
"
rK c C sZ t d}y| jd}W n tk
r4 tddY nX | || j | |d d j fS )a Splits the comment with the type annotation into parts for argument and return types.
For example, for an input of:
# type: (Tensor, torch.Tensor) -> Tuple[Tensor, Tensor]
This function will return:
("(Tensor, torch.Tensor)", "Tuple[Tensor, Tensor]")
z# type:z->z3Syntax error in type annotation (cound't find `->`)Nr )rj index
ValueErrorr7 r )rR Z start_offsetZ arrow_posr3 r3 r4 rs s
rs c s yt j| W n tk
r" dS X jgdd jj D }tfdd|D rZdS fddjj D }tj }||fS )z:Tries to use the Py3.5+ annotation syntax to get the type.Nc S s g | ]
}|j qS r3 )
annotation)r` pr3 r3 r4 rb s z(try_real_annotations..c 3 s | ]}| j kV qd S )N)empty)r` rr )sigr3 r4 s z'try_real_annotations..c s g | ]}t |j qS r3 )rq r )r` r )rO r3 r4 rb s )rU rP r return_annotation
parametersvaluesallrq )rN rO Z
all_annotsry rQ r3 )rO r r4 rI s
rI )rx c sJ t | }|s td| j ddd |D } fdd|D }tjj|S )Nz No enum values defined for: ''c S s h | ]}t |jqS r3 )typevalue)r` vr3 r3 r4 s z&get_enum_value_type..c s g | ]}t | qS r3 )try_ann_to_type)r` t)rO r3 r4 rb s z'get_enum_value_type..)r r __class__r= _CZunify_type_list)rx rO Z enum_valuesr Zir_typesr3 )rO r4 get_enum_value_type s r c C sP t | tjrdS t | tjtjtjtjtjtjtj tj
tj f rLt j
d dS dS )NTzTorchScript will treat type annotations of Tensor dtype-specific subtypes as if they are normal Tensors. dtype constraints are not enforced in compilation either.F)
issubclassr= r> Z
LongTensorZ DoubleTensorZ FloatTensorZ IntTensorZ ShortTensorZ
HalfTensorZ
CharTensorZ
ByteTensorZ
BoolTensorwarningswarn)rr r3 r3 r4 is_tensor s
r c s | t jjkrtj S | d kr$tj S t j| r>t| r>tj S t | rt
| j dkrj| j d f krjt g S t fdd| j D S t
| rt| j d }|rt|S t| r(t| j d }t| j d }|d krtd| j d d j |d kr td| j d d j t||S t| rt| j d td rT| j d }n
| j d }t| }d}|st|jt| t|t|S t| rg }x^| j D ]T} | d kr|j tj t| }
d}|
st|jt| t|
|j |
qW t |S t j j j! r4t"| r4t#t| j d S t$| rRt%t| j d S | t&krdt'j S | t(krvt)j S | t*krt+j S | t,krt-j S | t.krt/j S | t0krt1j S | td krtj S t j| rt2| drt3| j4S | t j5kr
t6j S | t j7kr t8j S | t j9kr2t+j S t j| rt| t:j;rt<| d krtt j=j>j?| } | j@ } ntA| } tB| tC| tD| S t j| rt<| }
|
d k r|
S t jEjF| rt j=j>j?| S d d
}t jGjH| |S ) Nr r c s g | ]}t | qS r3 )r )r` a)rO r3 r4 rb 5 s z#try_ann_to_type..zUnknown type annotation: 'z' at zQUnsupported annotation {} could not be resolved because {} could not be resolved.__torch_script_interface__c S s d S )Nr3 )keyr3 r3 r4 fake_rcb s z!try_ann_to_type..fake_rcb)IrU Signaturer r Z getInferredr# getisclassr r rj __args__r r r r r r highlightr r
r r AssertionErrorformatreprr r r r( r= rA rB rC r- r. r r& floatr complexr intr r_ r boolr r r" r[ r! r Zdevicer$ ZStreamr% ZdtypeenumEnumr rn _scriptZ_recursive_compile_classZqualified_namer r' r r
_jit_internalZcan_compile_classr Z_resolve_type_from_object)rr rO Z elem_typer r Z containedZ
valid_typemsginnerr Z
maybe_typeZscripted_classr0 Zmaybe_script_classr r3 )rO r4 r * s
r c C s2 t | |}|d k r|S td| d|j d S )NzUnknown type annotation: 'z' at )r r r )rr rO Zthe_typer3 r3 r4 rq s
rq r r r r r r r r r r r
r r r r r r r r r r" )Mrh r rU r rF r= r r r r r r r r r r
r r r
r r r r r r r _stater Ztorch._Cr r r r r r r r r r r! r" r# r$ r% r& r' r( textwrapr) Ztorch._sourcesr* r? r+ rA rB rC r, r- r. objectr/ r< rT rY r^ rf rp rM rK rs rI r r r r rq __all__r3 r3 r3 r4 s| D P >]