nn/qat/__init__.py · duality-group / torch v2.1.2+cpu - python package | Gemfury

Gemfury

Why Gemfury? Push, build, and install RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

duality-group / torch python

Repository URL to install this package:

Details

torch / nn / qat / __init__.py

# flake8: noqa: F401
r"""QAT Dynamic Modules

This package is in the process of being deprecated.
Please, use `torch.ao.nn.qat.dynamic` instead.
"""
from . import dynamic  # noqa: F403
from . import modules  # noqa: F403
from .modules import *  # noqa: F403

__all__ = [
    "Linear",
    "Conv1d",
    "Conv2d",
    "Conv3d",
    "Embedding",
    "EmbeddingBag",
]