pytorch
10 строк · 439.0 Байт
1# flake8: noqa: F401
2r"""Quantizable Modules.
3
4This file is in the process of migration to `torch/ao/nn/quantizable`, and
5is kept here for compatibility while the migration process is ongoing.
6If you are adding a new entry/functionality, please, add it to the
7appropriate file under the `torch/ao/nn/quantizable/modules`,
8while adding an import statement here.
9"""
10from torch.ao.nn.quantizable.modules.activation import MultiheadAttention11