Spaces:
Running
Running
File size: 497 Bytes
c61ccee |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
# flake8: noqa: F401
r"""QAT Modules.
This file is in the process of migration to `torch/ao/nn/qat`, 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/qat/modules`,
while adding an import statement here.
"""
from torch.ao.nn.qat.modules.conv import Conv1d
from torch.ao.nn.qat.modules.conv import Conv2d
from torch.ao.nn.qat.modules.conv import Conv3d
|