Gemfury - ao/nn/qat/modules/__init__.py · team-dmm / torch v2.4.0 - python package

team-dmm / torch python

Repository URL to install this package:

Details

torch / ao / nn / qat / modules / __init__.py

from .linear import Linear
from .conv import Conv1d
from .conv import Conv2d
from .conv import Conv3d
from .embedding_ops import EmbeddingBag, Embedding

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