[quant] add docs for embedding/embedding_bag

Summary: Test Plan: tested locally on mac Reviewers: Subscribers: Tasks: Tags: ghstack-source-id: e402b5f3cbe158426ed118f307cad7e4dfe741bd Pull Request resolved: #51770
pytorch · Feb 5, 2021 · 0f012f1 · 0f012f1
1 parent 1065c2d
commit 0f012f1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/docs/source/torch.nn.quantized.rst b/docs/source/torch.nn.quantized.rst
@@ -112,3 +112,13 @@ InstanceNorm3d
 ~~~~~~~~~~~~~~~
 .. autoclass:: InstanceNorm3d
     :members:
+
+Embedding
+~~~~~~~~~~~~~~~
+.. autoclass:: Embedding
+    :members:
+
+EmbeddingBag
+~~~~~~~~~~~~~~~
+.. autoclass:: EmbeddingBag
+    :members: