/usr/local/lib64/python3.6/site-packages/caffe2/python/helpers
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]