Name	Size	Mode	Actions
backend.cpython-36.pyc	28901	0644	editdlrm
backend_cpp_rep.cpython-36.pyc	1869	0644	editdlrm
backend_rep.cpython-36.pyc	2061	0644	editdlrm
error.cpython-36.pyc	450	0644	editdlrm
frontend.cpython-36.pyc	10344	0644	editdlrm
helper.cpython-36.pyc	3629	0644	editdlrm
onnxifi.cpython-36.pyc	1631	0644	editdlrm
test_onnxifi.cpython-36.pyc	6131	0644	editdlrm
workspace.cpython-36.pyc	2590	0644	editdlrm
__init__.cpython-36.pyc	154	0644	editdlrm

3 Eg>�@szdd
lm
Z

ddlmZ
ddlmZ
ddlZddlZeje �
Z dd�Zddd�
Zd d �Z ddd �
Zddd�
Zddd�
ZdS)�)
� caffe2_pb2)
�namedtupledict)
� WorkspaceNcs�t��t
|
t�r8xf|
j�D]\}}�j|||j�
qWn@t|j�
t|
�
ksNt�
x(t |j|
�D]\}}�j|||j�
q\W�j |�

|j}�f
d
d�|D�
}�td|�|�fS)Nc
sg|]}
�j|
�
�qS�)
� FetchBlob)�.0�name)
�wsr�E/usr/local/lib64/python3.6/site-packages/caffe2/python/onnx/helper.py� sz$c2_native_run_op..�Outputs) r� isinstance�dict�items�FeedBlob� device_option�len�input�AssertionError�zipZRunOperatorOnce�outputr)Zop_def�inputs�key�value�output_names� output_valuesr)
r r �c2_native_run_ops





 

rc sj
t��|r�j
|�

t|t�rFx$|j�D]\}}�j|||
j�
q(Wn��f
d
d�|
jD�
}t|�
t|�
kr�x�t ||�D]\}}�j|||
j�
qvWn�t|�
t|
j�
kr�t dt|�
dt|
j�
�
t d|�
t dt|
�
d|
�
t dt|�
d|�
t|�
t|
j�
k
�
st�
x0t t|�
�
D] }�j|
j||||
j�
�
qW�j
|
�

|
j}�f
d d�|D�
} �td |�| �fS)Nc
sg|]}
�j|
�
s|
�qSr)
ZHasBlob)rZ input_name)
r rr r-s

z%c2_native_run_net..z&c2_native_run_net assert. len(inputs)=z len(predict_net.external_input)=zdebug_arg: zpredict_net �
:zinputs c
sg|]}
�j|
�
�qSr)
r)rr)
r rr rFsr)r� RunNetOncer rrrrZexternal_inputrr�print�typer�rangeZexternal_outputr) �init_net�predict_netrZ debug_argrrZ uninitialized�
irrr)
r r �c2_native_run_net$s4

















 

r%c
 Cs0tj
�}
t|d
��}|
j|j��

WdQRX|
S)N�rb)rZNetDef�openZParseFromString�read)�file�net�
frrr �load_caffe2_netJs



r,FcCsTt|
d
��}|j
|j��

WdQRX|rPt|
dd��}|j
t|�
�

WdQRXdS)N�wb�txt�
w)r'�writeZSerializeToString�str)r*r)Z output_txtr+rrr �save_caffe2_netQs




r2�� TcCs:t�}|r|j
|�

|j|
�

|j|
j|||�}~|d
S)zw Run the benchmark net on the target model. Return the execution time per iteration (millisecond). r
)rrZ CreateNetZBenchmarkNetr)r"r#�warmup_iters� main_itersZ layer_detailsr �resultsrrr �benchmark_caffe2_modelYs





r8c Cs~xt|�
D]}||
�
q Wd
}x4t|�
D](}t
j
�}||
�
t
j
�} || |7}q(Wtjdj||d||��

|d|S)z� Run the model several times, and measure the execution time. Return the execution time per iteration (millisecond). gzRThe PyTorch model execution time per iter is {} milliseconds, {} iters per second.i�)r!�time�log�info�format) �modelrZtrainingr5r6�verboseZ_iZtotal_pytorch_time�ts�terrr �benchmark_pytorch_modelgs










rA)
N)
F)r3r4T)Fr3r4F)Zcaffe2.protorZonnx.backend.baserZcaffe2.python.onnx.workspacer�loggingr9� getLogger�__name__r:rr%r,r2r8rArrrr �s


  &