FM / helpers

/usr/local/lib64/python3.6/site-packages/caffe2/python/helpers/

/usr/local/lib64/python3.6/site-packages/caffe2/python/helpers

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
algebra.py	1296	0644	edit dl rm
arg_scope.py	1114	0644	edit dl rm
array_helpers.py	648	0644	edit dl rm
control_ops.py	625	0644	edit dl rm
conv.py	10396	0644	edit dl rm
db_input.py	381	0644	edit dl rm
dropout.py	412	0644	edit dl rm
elementwise_linear.py	1385	0644	edit dl rm
fc.py	6405	0644	edit dl rm
nonlinearity.py	1145	0644	edit dl rm
normalization.py	10891	0644	edit dl rm
pooling.py	924	0644	edit dl rm
quantization.py	261	0644	edit dl rm
tools.py	1089	0644	edit dl rm
train.py	2192	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/helpers/algebra.py (1296B)

## @package algebra
# Module caffe2.python.helpers.algebra

def transpose(model, blob_in, blob_out, use_cudnn=False, **kwargs):
    """Transpose."""
    if use_cudnn:
        kwargs['engine'] = 'CUDNN'
    return model.net.Transpose(blob_in, blob_out, **kwargs)

def sum(model, blob_in, blob_out, **kwargs):
    """Sum"""
    return model.net.Sum(blob_in, blob_out, **kwargs)

def reduce_sum(model, blob_in, blob_out, **kwargs):
    """ReduceSum"""
    return model.net.ReduceSum(blob_in, blob_out, **kwargs)

def sub(model, blob_in, blob_out, **kwargs):
    """Subtract"""
    return model.net.Sub(blob_in, blob_out, **kwargs)

def mat_mul(model, blob_in, blob_out, **kwargs):
    """Matrix multiplication"""
    return model.net.MatMul(blob_in, blob_out, **kwargs)

def arg_min(model, blob_in, blob_out, **kwargs):
    """ArgMin"""
    return model.net.ArgMin(blob_in, blob_out, **kwargs)

def batch_mat_mul(model, blob_in, blob_out,
                  enable_tensor_core=False, **kwargs):
    if enable_tensor_core:
        kwargs['engine'] = 'TENSORCORE'

return model.net.BatchMatMul(blob_in, blob_out, **kwargs)

def sparse_lengths_sum_4bit_rowwise_sparse(model, blob_in, blob_out, **kwargs):
    return model.net.SparseLengthsSum4BitRowwiseSparse(blob_in, blob_out, **kwargs)

## @package algebra # Module caffe2.python.helpers.algebra def transpose(model, blob_in, blob_out, use_cudnn=False, **kwargs): """Transpose.""" if use_cudnn: kwargs['engine'] = 'CUDNN' return model.net.Transpose(blob_in, blob_out, **kwargs) def sum(model, blob_in, blob_out, **kwargs): """Sum""" return model.net.Sum(blob_in, blob_out, **kwargs) def reduce_sum(model, blob_in, blob_out, **kwargs): """ReduceSum""" return model.net.ReduceSum(blob_in, blob_out, **kwargs) def sub(model, blob_in, blob_out, **kwargs): """Subtract""" return model.net.Sub(blob_in, blob_out, **kwargs) def mat_mul(model, blob_in, blob_out, **kwargs): """Matrix multiplication""" return model.net.MatMul(blob_in, blob_out, **kwargs) def arg_min(model, blob_in, blob_out, **kwargs): """ArgMin""" return model.net.ArgMin(blob_in, blob_out, **kwargs) def batch_mat_mul(model, blob_in, blob_out, enable_tensor_core=False, **kwargs): if enable_tensor_core: kwargs['engine'] = 'TENSORCORE' return model.net.BatchMatMul(blob_in, blob_out, **kwargs) def sparse_lengths_sum_4bit_rowwise_sparse(model, blob_in, blob_out, **kwargs): return model.net.SparseLengthsSum4BitRowwiseSparse(blob_in, blob_out, **kwargs)