/usr/local/lib64/python3.6/site-packages/caffe2/python/helpers
NameSizeModeActions
__pycache__/-0755rm
algebra.py12960644editdlrm
arg_scope.py11140644editdlrm
array_helpers.py6480644editdlrm
control_ops.py6250644editdlrm
conv.py103960644editdlrm
db_input.py3810644editdlrm
dropout.py4120644editdlrm
elementwise_linear.py13850644editdlrm
fc.py64050644editdlrm
nonlinearity.py11450644editdlrm
normalization.py108910644editdlrm
pooling.py9240644editdlrm
quantization.py2610644editdlrm
tools.py10890644editdlrm
train.py21920644editdlrm
__init__.py00644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/helpers/dropout.py (412B)
## @package dropout # Module caffe2.python.helpers.dropout def dropout(model, blob_in, blob_out, use_cudnn=False, **kwargs): """dropout""" if use_cudnn: kwargs['engine'] = 'CUDNN' else: kwargs['engine'] = 'DEFAULT' assert 'is_test' in kwargs, "Argument 'is_test' is required" return model.net.Dropout( blob_in, [blob_out, "_" + blob_out + "_mask"], **kwargs)[0]