19 lines
855 B
Python
19 lines
855 B
Python
# flake8: noqa: F401
|
|
r"""Quantized Modules.
|
|
|
|
This file is in the process of migration to `torch/ao/nn/quantized`, and
|
|
is kept here for compatibility while the migration process is ongoing.
|
|
If you are adding a new entry/functionality, please, add it to the
|
|
appropriate file under the `torch/ao/nn/quantized/modules`,
|
|
while adding an import statement here.
|
|
"""
|
|
|
|
from torch.ao.nn.quantized.modules.activation import ELU
|
|
from torch.ao.nn.quantized.modules.activation import Hardswish
|
|
from torch.ao.nn.quantized.modules.activation import LeakyReLU
|
|
from torch.ao.nn.quantized.modules.activation import MultiheadAttention
|
|
from torch.ao.nn.quantized.modules.activation import PReLU
|
|
from torch.ao.nn.quantized.modules.activation import ReLU6
|
|
from torch.ao.nn.quantized.modules.activation import Sigmoid
|
|
from torch.ao.nn.quantized.modules.activation import Softmax
|