We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9a1dc43 commit ff8c3fdCopy full SHA for ff8c3fd
torch/__init__.py
@@ -295,6 +295,7 @@ def manager_path():
295
import torch.autograd
296
from torch.autograd import no_grad, enable_grad, set_grad_enabled # noqa: F401
297
import torch.nn
298
+import torch.nn.quantized
299
import torch.optim
300
import torch.multiprocessing
301
import torch.sparse
torch/nn/quantized/__init__.py
@@ -1 +1,12 @@
1
-from . import functional # noqa: F401
+r"""
2
+The quantized module contains the data structures and ops for implementation of
3
+the quantized models.
4
+"""
5
+
6
+from torch.nn.quantized.functional import add_relu
7
+from torch.nn.quantized.functional import relu
8
9
+__all__ = [
10
+ 'add_relu',
11
+ 'relu'
12
+]
0 commit comments