Name	Size	Mode	Actions
algebra.cpython-36.pyc	1503	0644	editdlrm
arg_scope.cpython-36.pyc	972	0644	editdlrm
array_helpers.cpython-36.pyc	686	0644	editdlrm
control_ops.cpython-36.pyc	649	0644	editdlrm
conv.cpython-36.pyc	5786	0644	editdlrm
db_input.cpython-36.pyc	451	0644	editdlrm
dropout.cpython-36.pyc	486	0644	editdlrm
elementwise_linear.cpython-36.pyc	1109	0644	editdlrm
fc.cpython-36.pyc	3492	0644	editdlrm
nonlinearity.cpython-36.pyc	1082	0644	editdlrm
normalization.cpython-36.pyc	5978	0644	editdlrm
pooling.cpython-36.pyc	903	0644	editdlrm
quantization.cpython-36.pyc	357	0644	editdlrm
tools.cpython-36.pyc	818	0644	editdlrm
train.cpython-36.pyc	1965	0644	editdlrm
__init__.cpython-36.pyc	157	0644	editdlrm

3 Egã@sHdd
d„
Zdd„Z
dd„Zdd„Zd d „Zdd„Zdd d„
Zdd„ZdS)FcKs|rd
|d<|jj
|
|f|Ž
S)z Transpose.ZCUDNNÚengine)ÚnetZ Transpose)ÚmodelÚblob_inÚblob_outZ use_cudnnÚkwargs©rúI/usr/local/lib64/python3.6/site-packages/caffe2/python/helpers/algebra.pyÚ transpose s

r cKs|jj
|
|f|Ž
S)
ÚSum)rr )rrrrrrrÚsumsrcKs|jj
|
|f|Ž
S)
Ú ReduceSum)rr)rrrrrrrÚ reduce_sumsr cKs|jj
|
|f|Ž
S)
ZSubtract)rÚSub)rrrrrrrÚsubsrcKs|jj
|
|f|Ž
S)
zMatrix multiplication)rZMatMul)rrrrrrrÚmat_mulsrcKs|jj
|
|f|Ž
S)
ÚArgMin)rr)rrrrrrrÚarg_min$srcKs|rd
|d<|jj
|
|f|Ž
S)NZ TENSORCOREr
)rZBatchMatMul)rrrZenable_tensor_corerrrrÚ batch_mat_mul(s
rcKs|jj
|
|f|Ž
S)
N)rZ!SparseLengthsSum4BitRowwiseSparse)rrrrrrrÚ&sparse_lengths_sum_4bit_rowwise_sparse/s
rN)
F)
F)r rr rrrrrrrrrÚ s