Learn more  » Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Bower components Debian packages RPM packages NuGet packages

neilisaac / torch   python

Repository URL to install this package:

/ nn / intrinsic / qat / modules / __init__.py

from .linear_relu import LinearReLU
from .conv_fused import ConvBn1d, ConvBn2d, ConvBnReLU1d, ConvBnReLU2d, ConvReLU2d, \
    update_bn_stats, freeze_bn_stats

__all__ = [
    'LinearReLU',
    'ConvReLU2d',
    'ConvBn1d',
    'ConvBn2d',
    'ConvBnReLU1d',
    'ConvBnReLU2d',
    'update_bn_stats',
    'freeze_bn_stats'
]