/usr/local/lib64/python3.6/site-packages/caffe2/python/onnx/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/onnx/__pycache__/backend.cpython-36.pyc (28901B)
3
EgР @ sf d Z ddlZddlZddlZddlZddlZddlmZm Z m
Z
m Z ddl m
Z
ddlmZ ddlZddlZddlZddlmZ ddlZddlZddlZddlZddlmZmZm Z m Z ddl m Z dd l m!Z! ddl"j#j$Z%ddl&Z&d
d Z'd d
Z(G dd de)Z*dd Z+G dd de,Z-ej.ddddgZ/G dd deZ0e0j1Z1e0j2Z2e0j3Z3e0j4Z4e0j5Z5e0j6Z6dS )zaBackend for running ONNX on Caffe2
To run this, you will need to have Caffe2 installed as well.
N)core workspacernn_cellgru_cell) ModelHelper)
caffe2_pb2) TensorProto)BackendDevice
DeviceTypenamedtupledict) Workspace) Caffe2Repc C s$ y
| j dS tk
r | S X d S )Nzutf-8)decodeAttributeError)s r F/usr/local/lib64/python3.6/site-packages/caffe2/python/onnx/backend.py
force_unicode' s
r c C s( t jtjt jtji}tj|| j | j S )N)
r CPUr ZCUDAr Z
GpuDeviceTyper Z DeviceOptiontypeZ device_id)devicemr r r get_device_option- s r c @ s, e Zd ZdZedd Zdd fddZdS ) OnnxAttributeszx
This is a more convenient way to work with ONNX/Caffe2 attributes
that is not the protobuf representation.
c C s& t }x| D ]}t|||j< q W |S )N)r convertAttributeProtoname)argsdargr r r from_onnx8 s
zOnnxAttributes.from_onnxc C s | S )Nr )kr r r
? s zOnnxAttributes.c c s< x6| j D ]*\}}||dkr
tjjj|||V q
W d S )N )itemscaffe2pythonutils MakeArgument)selfkmapr! vr r r r% ? s zOnnxAttributes.caffe2N)__name__
__module__ __qualname____doc__ staticmethodr r% r r r r r 3 s r c C s | j dr| jS | j dr | jS | j dr0| jS | j dr@| jS | j drZtj| jtjS t | j
rnt | j
S t | j rt | j S t | j
rt | j
S t | jrg }x"| jD ]}|jtj|tj qW |S tdj| dS )z
Convert an ONNX AttributeProto into an appropriate Python object
for the type.
NB: Tensor attribute gets returned as the straight proto.
fir tgz Unsupported ONNX attribute: {}N)HasFieldr1 r2 r r3
Caffe2Backend
_graph_to_netr4 _known_opset_versionlenfloatslistintsstringsgraphsappend
ValueErrorformat)Zonnx_argretvalr4 r r r r E s,
r c @ s e Zd ZdZdd ZdS )OnnxNodeaX
Reimplementation of NodeProto from ONNX, but in a form
more convenient to work with from Python.
We may temporarily edit these nodes to get them into Caffe2 form,
before actually translating into the Caffe2 protobuf, since this
is easier than decomposing everything, and putting it back together
when we're ready.
c C sB t |j| _t |j| _tj|j| _t|j| _ t|j
| _ d S )N) strr op_typer r attributeattrsr; inputinputsoutputoutputs)r) noder r r __init__q s
zOnnxNode.__init__N)r, r- r. r/ rM r r r r rC g s rC Caffe2Opsopsinit_opsinterface_blobsc s e Zd ZdZi Zdddddddd d
dd d d
ddddddZddiZddiddiddiddddd id d id d d!d"d#d$d%Zd&d&d&d'd(d)d*d+Ze j
Z e d,d- Z
e d.ed/f fd0d1 Ze ded2d3Ze d4d5 Ze d6d7 Ze d8d9 Ze d:d; Ze dd? Ze d@dA Ze dBdC Ze dDdE Ze dFdG Ze dHdI Ze dJdK Ze dfdMdNZ e dgdOdPZ e dh fdQdR Z e dSdT Z dUdVhZ!e dWdX Z"e dYdZ Z#e d[d\ Z$e d]d^ Z%e d.efd_d`Z&e dadb Z'e di fdcdd Z( Z)S )jr6 ZMaxPoolZ AveragePoolZPadImageZNegativeZ SpatialBNZ InstanceNormZ BatchMatMulZ
ResizeNearestCopyZEQLTGT
ExpandDims ONNXWhileZ NumpyTileZ GaussianFillZ UniformFill)Z
GlobalMaxPoolZGlobalAveragePoolZPadZNegZBatchNormalizationInstanceNormalizationZMatMulUpsampleZIdentityrX ZEqualZLessZGreater UnsqueezeLoopTile RandomNormal
RandomUniformZ kernel_shapeZkernelsaxesdimsZpermr# )modeZscalesZoutput_paddingZadjsgammaZscalethen_netelse_net)Z then_branchZ else_branchminmax)lowhigh)SqueezerZ TransposerY Z
ConvTransposeZSeluIfr^ _create_rnn_variant _create_loop
_create_ifZ_create_upsampleZ_create_gaussian_fill)LSTMGRURNNr[ rk rY r] c C s
| j j S )N) _dummy_nameZnew_dummy_name)clsr r r
dummy_name s zCaffe2Backend.dummy_namer Nc s t t| j|||||d g }tt|}t tj|j t|t rx|j
D ]<\}} j || |j t
jj|t
jj| j | jdj qRW n~t|jt|kstdj|jt|jt|xLt|j|D ]<\}} j || |j t
jj|t
jj| j | jdj qW g }
tj| j } | j |j ||} xB| d | d D ].}
t j }|j |
|j!j"| |
j | qFW j#|
fdd|j$D }t%d|j$| S Q R X d S ) N)r outputs_info
opset_version)r elem_typeshapez{}: expected {} but got {}r c s g | ]} j |qS r )Z FetchBlob).0r )wsr r
s z*Caffe2Backend.run_node..ZOutputs)&superr6 run_noder r
r
r Z DeviceScope
isinstancedictr$ FeedBlobr? onnxhelperZmake_tensor_value_infomappingZNP_TYPE_TO_TENSOR_TYPEdtyperx SerializeToStringr9 rH AssertionErrorrA rE zipCrr convert_noder OperatorDefParseFromString
device_optionCopyFromZRunOperatorsOncerJ r )rs rL rI r rv ru value_infosr keyvaluerO cbackendZops_strr opZ
output_values) __class__)r{ r r~ s@
zCaffe2Backend.run_nodec C sl |s|j st|p|j }tj }|jj }d|_ dd }|jtjgkr^d|_ |j
j || n|jtj gkrd|_ |j
j || n|jtj
tjgkrd|_ |jj || n|jtjtjtjtjtjgkrd|_ |jj || nZ|jtjkr
d|_ |jj || n4|jtjkr.d |_ |jj |j ntd
j|j|jj }d |_ |jj |j |j j | |S ) z
Given an Onnx TensorProto, translate it into a Caffe2 operator
which produces the given tensor filling op.
valuesc S s t jj| j j S )N)r numpy_helperto_arrayflattentolist) onnx_tensorr r r tensor2list s z.tensor2listZGivenTensorFillZGivenTensorDoubleFillZGivenTensorInt64FillZGivenTensorIntFillZGivenTensorBoolFillZGivenTensorStringFillzunrecognized tensor type {}rx ) r r r r r addZ data_typer FLOATr r: extendZDOUBLEZINT64ZUINT32r< ZUINT8ZINT8ZUINT16ZINT16INT32ZBOOLSTRINGr= Z string_data RuntimeErrorrA r` rJ r? )rs r r c2_opZ c2_valuesr Zc2_shaper r r _create_tensor_filling_op sH
z'Caffe2Backend._create_tensor_filling_opc s x|D ]\}}} fdd|D xLt D ]@\}
} |
| |
d | f} t| f| \}
}|j|| |
|d q0W |r fdd|D }|j|df | j gdd qW d S ) Nc s g | ]}d |f qS )z%s/%s_%sr )rz prefix)r name_tor r r| + s z5Caffe2Backend._rnn_reform_weights..ry )startsendsc s g | ] } | qS r r )rz r2 )
gate_blobsr r r| 1 s z%s/%sr )axis) enumerater SliceConcatrt )rs reformsr hidden_sizeinit_netZgatesZreorder_indicesZ name_fromZ do_concatZ
extra_dimsr2 xZdim0r r Zreordered_gate_blobsr )r r r r _rnn_reform_weights( s z!Caffe2Backend._rnn_reform_weightsc "
C sX | j }| |
}d| | }| | }|j|||
|d| g|d| gd}
|j||| |d| g|d| gd}|j||| |d| dg|d| dgd}|j||| |d| dg|d| dgd}g }xB|D ]:\}}|j|jj||| | d ddg| d ddgd qW | dkr|d k r6|}n|jj||d } |jj| |d dgdgd} |jj| |d dgdgd} |jj| | g|d dd } |jj| | | j gdgd
|jj| |d tj j
d }||
|||||
| | dkr|jj ||g|d
} n|} |t ||| ||| |
|ddd }!| dkrT|jj |!d |g|d |!d< |!S )N r ry )r r z /input_shapez/batch_size_slicez/seq_len_slicez/dummy_sequence_lens)r )rx z/seq_lens_for_reverse)Ztoz/input-reversedFT)Z drop_statesZ forward_onlyz/output-reversedr r r r )
rt r r? netZShaper\ ReshapeZCastr ZDataTyper ZReversePackedSegsr; )"rs
input_blobBWRZinitial_states_and_names
sequence_lenspred_mhr
input_sizer Z num_gatesdirection_offsetBiBrW_R_reform make_cellZ keep_outputsr Zgates_hidden_sizeZ bias_offsetZ
weight_offsetZinitial_states_slicedZ
initial_stateZ name_suffixZseq_lens_for_reverseZ input_shapeZ
batch_sizeZseq_lenZdummy_sequence_lensrH rK r r r _make_rnn_direction4 sd
z!Caffe2Backend._make_rnn_directionc s |d k st d|d k s t dt|j}|jdt|jdd}|jdkrht|jdd*d j n|jd kr~|jd
d | st d t|j |d+kst d
|jd,kr|j \
n |jdkr|j \
dkrd xRt
j |j j
|j j|j j
|j jD ]&}|jkr|jjjjd j P qW tdt tjdjj gdd-dgd jdgd jj gdd.dgd jdgd j j gdd/gd j dgd |jdkr(dd fdd
f
dd}n|jd krrfd d fd d
f
d d}nF|jdkrfd dd d
fd!d}|dkrF|dx0t dt D ] } j j | |j | qW d k r"j j!d gd g j j"d g|j d gdgd n|d kr>|d}
|d} j j#|
d | d gj j gdd"\} }
d k rj j!| g| g j j| j j gddd0dgd\}}
j j$||j d dddd#gd$ xFt dt |j D ]2} j j#|
| | | g|j | j gdd" qW d%d& |j j%D d'd& |j j&D t' fd(d) fD }g }g }|r|n|j(j) j* |j(j) j* t+||t,j) j-S )1Nz4cannot convert RNNs without access to the full modelr directionforwardrq Z activationstanhr rp linear_before_resetz unsupported RNN attributes:
bidirectionalz"unsupported backwards RNN/GRU/LSTMro r# r z3best-effort shape inference for RNN/GRU/LSTM failedzinit-netry )rx )r` c W s d S )Nr )r r r r r s z1Caffe2Backend._create_rnn_variant..reformc s t j| d i|S )N
activation)r ZBasicRNN)r kwargs)r r r r s z4Caffe2Backend._create_rnn_variant..make_cellc s8 j dfg
d| dddd dd S ) Nz
/initial_hry z/i2h_bz
/gates_t_bz/i2h_wz
/gates_t_wc S s | S )Nr )r r r r r" s zECaffe2Backend._create_rnn_variant..make_rnn..)r )r )
r r r rs r r initial_hr r r r r r r r make_rnn s
z3Caffe2Backend._create_rnn_variant..make_rnnc sT |dddgf|dddgf| ddg f|ddg ff} j ||||d d
d gddd g d S )Ni2h_wTr ry Zgate_t_wFi2h_bZgate_t_bupdateresetrJ r r )r r r )r r )r )r r r r r r r r )rs r r r s
c s t j| d i|S )Nr )r rp )r r )r r r r s c s8 j dfg
d| dddd dd S ) Nz
/initial_h Z _bias_i2hZ _bias_gatesz
/i2h_w_prez/gates_t_w_prec S s | S )Nr )r r r r r" s zECaffe2Backend._create_rnn_variant..make_rnn..)r )r )
r r r rs r r r r r r r r r r r r s
c
sX |dddgf|dddgf| ddg f|ddg ff} j ||||dd d
d gdd dd
g d S )Nr Tr ry Z gates_t_wr Z gates_t_brH rJ Zforgetcellr r r )r r r )r r )r )r r r r r r r r )rs r r r s
c _ s t j| |S )N)r ro )r r r r r r s c s> j dfdfg
d| dddd
dd S )
Nz
/initial_hz
/initial_c z/i2h_bz
/gates_t_bz/i2h_wz
/gates_t_wc S s | d | d | d gS )Nr ry r r )r r r r r" s zECaffe2Backend._create_rnn_variant..make_rnn..)r )r )r r r rs r r initial_cr r r r r r r r r r s
)r r )r_ c S s h | ]
}|j qS r )r )rz r2 r r r s z4Caffe2Backend._create_rnn_variant..c S s h | ]}|j D ]}|qqS r )rJ )rz rL rJ r r r r s c 3 s | ]}| kp|kV qd S )Nr )rz r ) initializersrK r r s z4Caffe2Backend._create_rnn_variant..)r )r r )rq rp r r r r ).r r rG popr rE lowerrD keysrI itertoolschaingraphrH
value_infor r tensor_typerx dim dim_valuer r r ZNetr rt ri ranger9 r rS rK Z VariableLengthSequencePaddingrV r rj initializerrL allr ZProtor rN r; external_input)rs
init_model
pred_modelnrv rG r r r r2 Z outputs_fZ outputs_bZconcatted_output_Zreshaped_outputZhas_initializersZpred_opsrP r )r r r r rs r r r r r r r r r rK r r r r rl y s
&
&
&
$
(
*
z!Caffe2Backend._create_rnn_variantc C sH g }d|j kr |j|j d | j||||}|jj| t|gg g S )N__control_inputs)rG r _common_onnx_node_to_caffe2_opZ
control_inputrN )rs r r r rv control_inputsrL r r r _create_control_op s
z Caffe2Backend._create_control_opc C sp x<|j D ]2}x,t|jD ] \}}||kr|| |j|< qW qW x,t|jD ] \}}||krJ|| |j|< qJW d S )N)r r rJ external_output)rs r Z
remap_dictr r2 r outr r r _remove_ssa s zCaffe2Backend._remove_ssac C s | j ||||}|d d jdks&t|d d }d }}g } x>|jD ]4}
|
jdkrZ|
j}|
jdkrj|
j}|
jdkrF|
j} qFW |r|st|j} |j} |j}
t | t | kstt | t |
kstx>|jD ]4}
|
jdkr|
jj
j | |
jdkr|
jj
j | qW |S )Nr rk rc rd r ) r r r r r r r= r rJ r9 r r )rs r r r rv rO Zif_oprc rd r r Z
then_net_outsZ
else_net_outsZ
op_outputsr r r rn ! s0
zCaffe2Backend._create_ifc
C s | j ||||}|d d jdks&t|d d }|jjtjjjddg |jjtjjjddg |jjtjjjddg g }x |jD ]}|j dkr|j
}qW d} x6|jD ],}|j dkrt |j j
d } |j j
j| qW |jjtjjjd
| g |S ) Nr rW Zhas_trip_countTZhas_condZdisable_scopesr bodyr num_loop_carried_deps)r r r r r r% r& r' r( r r= r9 r r )
rs r r r rv rO Zwhile_opr r r r r r rm ? s&
zCaffe2Backend._create_loopc C sD |j dr@|jtdkr@|j|kr4tdj|jn ||j |_d S )Nraw_datas
__EXTERNALzBTensorProto for value {} referenced raw data but it was not found!)r5 r bytesr r rA )rs tpraw_values_dictr r r _substitute_raw_valueW s
z#Caffe2Backend._substitute_raw_valuec C s x|D ]}xz|j D ]p}|jdr.| j|j| x|jD ]}| j|| q6W |jdrd| j|jj| x |jD ]}| j|j| qlW qW qW d S )Nr3 r4 ) rF r5 r r3 Ztensors _visit_and_substitute_raw_valuesr4 rL r> )rs Znodesr rL attrr3 r4 r r r r _ s
z.Caffe2Backend._visit_and_substitute_raw_valuesc C s2 x |j jD ]}| j|| q
W | j|j j| d S )N)r r r r rL )rs modelr initr r r _external_value_resolution_passl s z-Caffe2Backend._external_value_resolution_passc C s* x$|D ] }|j |jtjj|| qW d S )N)r r r r r )rs r r{ r r r r r _direct_initialize_parameterst s
z+Caffe2Backend._direct_initialize_parametersc C s` xZ|D ]R}|j |krqtdd |jjjjD }|j|j tj|t j
j |jjj d| qW d S )Nc s s | ] }|j V qd S )N)r )rz r r r r r ~ s z:Caffe2Backend._direct_initialize_inputs..)r )
r r; r r rx r r npZonesr r ZTENSOR_TYPE_TO_NP_TYPErw )rs rI initializedr{ r r rx r r r _direct_initialize_inputsy s
z'Caffe2Backend._direct_initialize_inputsFc C s: ddddg}|r|j d |r(|j d tjj| |}|S )NZfuse_consecutive_transposesZeliminate_nop_transposeZfuse_transpose_into_gemmZlift_lexical_referencesZ
split_initZ
split_predict)r? r Z optimizeroptimize)rH r predictZpassesr r r r
optimize_onnx s
zCaffe2Backend.optimize_onnxc K s t j|ddr}|jdd}tj|}W d Q R X t|j td }i }x0|D ](} |j| d}
|
j || < W d Q R X qPW W d Q R X | j||fd|i|S )Nr)ra Z
__MODEL_PROTOr ) zipfileZipFileopenr loadsetnamelistreadprepare) rs filer r zr1 r Z
blob_namesr r Z blob_filer r r prepare_zip_archive s
$z!Caffe2Backend.prepare_zip_archivec
sp |j dds"tt| j||f| d}x`|jD ]V}|jd sH|jdkrr|j}|j| jkrt j
dj | j|j q.t j
dj |j q.W |dkr|j dkrt
d nd
}ytjj|}W n t
k
r t j
d Y nX t }tt|}| j|||d\} }
|r| j|| | j|jj|| d d
|jjD | j|jj || fdd|jjD } t| |
|| } | S )a
For Onnx Caffe2Backend, we require that init_graph don't initialize the actual input of the predict_graph,
for example, if "img" is the input blob for the predict_net, we require that in init_graph and in
initializer of the predict_graph, "img" is not initalized. We don't have a check for this, since
there is no way we can know which blob is the input of the predict_graph.
Zno_check_UNSAFEFNdomainr# a This version of onnx-caffe2 targets ONNX operator set version {}, but the model we are trying to import uses version {}. We will try to import it anyway, but if the model uses operators which had BC-breaking changes in the intervening versions, import will fail.z Unrecognized operator set {}r z^Model with IR version >= 3 did not specify ONNX operator set version (onnx-caffe2 requires it)ry zGShapeInferenceWarning: Inferred shape and existing shape differ in rankc S s h | ]
}|j qS r )r )rz r r r r r s z(Caffe2Backend.prepare..c s g | ]}|j kr|j qS r )r )rz r )r r r r| s z)Caffe2Backend.prepare..) r r} r6 r
Z opset_importr5 r versionr8 warningswarnrA Z
ir_versionr r Zshape_inferenceZ infer_shapesr
r r
_onnx_model_to_caffe2_netr r r r r rH r )
rs r r r r rv impr{ r r Z predict_netZ
uninitializedrB )r )r r r
sF
zCaffe2Backend.preparec C s^ t j| j}|j|jrg }xR|jD ]H}|d k r$x:tj|jj|jj |jj
D ] }|j |krL|j |j
qLW q$W |j|j
||} g }
x,| d D ] } tj } | j| |
j | qW g }
x,| d D ] } tj } | j| |
j | qW t|
|
g S |j| jkrt| | j|j }n| j}|||t||}
t|
tr<|
S t|
tjjsR|
g}
t|
g g S )Nr ry )r r6 rr Zsupport_onnx_importrE rH r r r rJ r r r? r r r r r rN _special_operatorsgetattrr rC r collectionsabcIterable)rs r r Znode_defrv r r r viZop_strsrP r r rO Z
translatorr r r _onnx_node_to_caffe2_op s>
z%Caffe2Backend._onnx_node_to_caffe2_opAddSubc
s
t j }|jj|j |jj|j |j|_|j j j
t d}||kr^t dj
|| jj
|_tj|jst dj
fdd}|jj|jj|d |dk r|j jkrd}x |jD ]} | jd krd
}qW |s|jjtjjjd d g |S ) a
This translator performs the basic translation of ONNX nodes into
Caffe2 operators. Besides doing a straightforward marshalling from
one format to another, it also does these extra things:
- Renames operators based on '_renamed_operators'
- Renames attributes based on '_global_renamed_attrs' and
'_per_op_renamed_attrs'
If you're writing a custom translator, consider calling this first,
and then fixing things up further.
ZInfzXDon't know how to translate op {} in ONNX operator set v{} (I only support prior to v{})z!Don't know how to translate op {}c s> j kr&| j kr& j | S | jkr: j| S | S )N)_per_op_renamed_attrs_global_renamed_attrs)r! )rs onnx_op_typer r r* , s
z:Caffe2Backend._common_onnx_node_to_caffe2_op..kmap)r* F broadcastTry )r r rH r rI rJ rK r rE _broken_operatorsgetfloatr@ rA _renamed_operatorsr r Z
IsOperatorr rG r% _broadcast_operatorsr& r' r( )
rs r r Z onnx_noderv r Zbroken_versionr* Zalready_broadcastr r )rs r" r r
s0
z,Caffe2Backend._common_onnx_node_to_caffe2_opc C sl | d krt S t }|jdd | jD |jdd | jD x&| jD ] }|j|j |j|j qHW |S )Nc s s | ] }|j V qd S )N)r )rz r r r r r H s z4Caffe2Backend._all_names_in_graph..c s s | ] }|j V qd S )N)r )rz r r r r r I s )r
r rH rJ rL )r namesrL r r r _all_names_in_graphB s z!Caffe2Backend._all_names_in_graphc C s t j }x||jD ]r}y| jd d ||}W n. tk
rV } ztd| wW Y d d }~X nX |jj|j |jj|j |j
j|j qW |j jdd |j
D |j
jdd |jD |S )Nz ONNX FATAL:c s s | ] }|j V qd S )N)r )rz r r r r r ] s z.Caffe2Backend._graph_to_net..c s s | ] }|j V qd S )N)r )rz r r r r r _ s )r NetDefrL r Exceptionprintr r rP rO r rQ r rJ rH )rs Z
onnx_graphrv r rL c2opser r r r7 O s
zCaffe2Backend._graph_to_netc # sV t t|}ytjj|}W nB tk
r: tjd Y n& tk
r^ tjdj tj
Y nX y j |dd} j |dd}W n. tk
r tjdj tj
|}|}Y nX t j
}t j
} |jjd |_|jjd | _|r|jj fd d
|jjD jj j|j j|jB g }
x ||f| |ffD ]\} } | jj| x| jjD ]}
y j|||
|}W nN tk
r } z0d j |
|}|
j| t d |t j d
wTW Y d d }~X nX |jj|j | jj|j | j!j|j" qTW | j#jdd
| jj$D | j!jdd
| jj%D q6W t&|
dkrNtdj t&|
dj'|
|| fS )NzGShapeInferenceWarning: Inferred shape and existing shape differ in rankz@ShapeInferenceWarning: utils module not found in ONNX version {}T)r )r z?OptimizerWarning: optimizer module not found in ONNX version {}_initZ_predictc 3 s | ]} j |V qd S )N)r )rz r )rs r r r s z:Caffe2Backend._onnx_model_to_caffe2_net..z.Error while processing node: {}. Exception: {}z ONNX FATAL:)r c s s | ] }|j V qd S )N)r )rz r r r r r s c s s | ] }|j V qd S )N)r )rz r r r r r s r z2ONNX conversion failed, encountered {} errors:
{}z
)(r r
r r' Z polish_modelr r r r rA __version__r r r, r r r r r rr r r+ r r rL r r- r? r. sysstderrrP rO r rQ r rJ rH r9 join)rs Z
onnx_modelr rv include_initializersr r r r Zpred_neterrorsr r rL r/ r0 msgr )rs r r b sZ
z'Caffe2Backend._onnx_model_to_caffe2_netc C s | j |||ddS )NT)r rv r6 )r )rs r r rv r r r onnx_graph_to_caffe2_net s z&Caffe2Backend.onnx_graph_to_caffe2_netc C s. t |}|jtjkrdS tj|jr*tjS dS )NTF)r
r r r r ZIsGPUDeviceTyper Zhas_gpu_support)rs Z
device_strr r r r supports_device s z Caffe2Backend.supports_devicec s> t tt| dr:ttt| jr:tt| j||f|s:dS dS )N
is_compatibleFT)hasattrr} r6 callabler; )rs r r r )r r r r; s
zCaffe2Backend.is_compatible)N)FF)r )r N)r )*r, r- r. r8 r% r( r! r r r Z DummyNamerr classmethodrt r~ r r r rl r r rn rm r r r r r r0 r r r
r r) r r+ r7 r r9 r: r;
__classcell__r r )r r r6 | s % 4 E
> + 5
> r6 )7r/ r r3 r r Z onnx.backendr Z
caffe2.pythonr r r r Zcaffe2.python.model_helperr Z caffe2.protor Zcaffe2.python.utilsr% Znumpyr r Zonnx.numpy_helperZ onnx.defsZonnx.shape_inferenceZ
onnx.utilsZonnx.backend.baser r
r r Z caffe2.python.onnx.workspacer
Z caffe2.python.onnx.backend_repr Z!caffe2.python._import_c_extensionr& Z_import_c_extensionr r r r r r r objectrC
namedtuplerN r6 r
r r~ Z run_modelr: r; r r r r sP " @