/usr/local/lib64/python3.6/site-packages/caffe2/python/helpers/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/helpers/__pycache__/conv.cpython-36.pyc (5786B)
3
Eg( @ sd d dl mZ d dlmZ d dlmZ ddd Zdd
d Zdd d
ZdddZ dddZ
dddZ dS ) )core) initializers)
ParameterTagsN FNCHWc K sd g }|r t |ts|g}qN|}n.t |trDt|dks>td|}n
|gd }|jd}|d k r|
rv|dkrvtdn|
r|dkrtd|
rd|d< ||d< |r||d< d |kr|d rd
nd }|p| jj }|g}|d kr |jt || |j
| n |j
| |jt || t j | |d
i f} t j |
|di f}
| j
sXt j } t j }
| j|d || tjd}|r| j|d |g|
tjd}|r|||g}n||g}| d k r| | || |rd |d< d |kr|d = | dkr| |d< |r| jj||f||d|S t |trD| jj||f|d |d |d|S | jj||f||d|S d S )N z Conv support only a 2D kernel.engineCUDNNz?When use_cudnn=True, the only engine you can specify is "CUDNN"z;When use_cudnn=False, the only engine you can specify is ""exhaustive_searchws_nbytes_limitno_biasFTr
XavierFill ConstantFill_w)
param_nameshape initializertags_bfloat16_computer group)kernelsorderr )Zkernel_hZkernel_wr )kernelr )
isinstancelistlenAssertionErrorget
ValueErrornetNextNameappendintextendr Zupdate_initializer init_paramsZExternalInitializerZ create_paramr WEIGHTBIASConv)modelZis_ndblob_inblob_outdim_indim_outr weight_init bias_initWeightInitializerBiasInitializerr transform_inputs use_cudnnr cudnn_exhaustive_searchr r kwargsr Zrequested_engineuse_bias weight_shapeweightbiasinputs r; F/usr/local/lib64/python3.6/site-packages/caffe2/python/helpers/conv.py _ConvBase s
r= c
K s< | dkst dt| d||||||||| |
| f
d| i|
S )zBN-dimensional convolution for inputs with NCHW storage order.
r z'ConvNd only supported for NCHW storage.Tr )r r= )r) r* r+ r, r- r r. r/ r0 r1 r r2 r r5 r; r; r< conv_nd s r> c
K s$ t | d||||||||| |
| f
| S )z 2-dimensional convolution.
F)r= )
r) r* r+ r, r- r r. r/ r0 r1 r r2 r5 r; r; r< conv s r? c K s2 |r|ndi f}|r|ndi f}|p,| j j }| dkrB||||gn
||||g}
| jr| jj|d g |d fd|
i|d }| jj|d g |d fd|gi|d }n$tj|d | j}tj|d | j}| j|tj | j|tj
|rd | d
< |
| d < | r| | d < | j j |||g|f|| d
| S )zConvTranspose.
r
r r r r r r r r r r
r )r r ) r r! r% param_init_net __getattr__r ScopedBlobReference AddParameterr r& r' Z
ConvTranspose)r) r* r+ r, r- r r. r/ r3 r r4 r r5 r7 r8 r9 r; r; r< conv_transpose s@
rD c
K s" t | |||||f|||d| S )zGroup Convolution.
This is essentially the same as Conv with a group argument passed in.
We specialize this for backward interface compatibility.
)r. r/ r )r? )
r) r* r+ r, r- r r. r/ r r5 r; r; r<
group_conv s rE c
sN |r|ndi f}|r|ndi f}d|
kr4|
d r4dnd}| rhd|
d< | |
d< | rh| |
d < rht d
| rxt d | jj| fd d
tD fdd
tD |
d}|
dkr| ||gn| || g}dd
|D }g }x0tD ]"}| jr|| jj|d g d| fd|i|d }|r| jj|d g d| fdt| gi|d }n2tj d| | j}|rtj d| | j}| j
|t j |r| j
|t j
|r||g}n|g}d|
kr|
d= |j|| j| d| f||
d|
q W | jj| d d g|
d\}}|S ) zGroupConvolution's deprecated interface.
This is used to simulate a group convolution via split and concat. You
should always use the new group convolution in your new code.
r
r r FTr r r
r z$dim_in should be divisible by group.z%dim_out should be divisible by group.c s g | ]}d d t | qS )_Z
_gconv_split_)str).0i)r+ r; r<
0 s z)group_conv_deprecated..c s g | ]}t qS r; )r# )rH rI )r, r r; r< rJ 1 s )Z
dimensionsr r c S s g | ] }t |qS r; )r# )rH vr; r; r< rJ ; s r z _gconv_%d_wr r z _gconv_%d_bz _gconv_%d)r r rF Z _concat_dims)r )r r Z
DepthSplitranger% r@ rA r# r rB rC r r& r' r" r( ZConcat)r) r* r+ r, r- r r. r/ r r3 r r4 r r5 r6 Zsplitted_blobsr7 Z
conv_blobsrI r8 r9 r: concatZ concat_dimsr; )r+ r, r r< group_conv_deprecated
sx
rN ) NNNNr NFr FNF)NNNNr Nr )NNNNr N)NNFr FN)NNr )NNr Fr FN) Z
caffe2.pythonr Zcaffe2.python.modelingr Z%caffe2.python.modeling.parameter_infor r= r> r? rD rE rN r; r; r; r< sT
x
5