nn/intrinsic/quantized/modules/__init__.py · turingmotors / torch v2.4.1 - python package | Gemfury

turingmotors / torch python

Repository URL to install this package:

Details

torch / nn / intrinsic / quantized / modules / __init__.py

from .linear_relu import LinearReLU
from .conv_relu import ConvReLU1d, ConvReLU2d, ConvReLU3d
from .bn_relu import BNReLU2d, BNReLU3d

__all__ = [
    'LinearReLU',
    'ConvReLU1d',
    'ConvReLU2d',
    'ConvReLU3d',
    'BNReLU2d',
    'BNReLU3d',
]