From dbf1ac830415637ca8aed4ad7970bbec2d8f7886 Mon Sep 17 00:00:00 2001 From: Supriya Rao Date: Thu, 4 Feb 2021 21:48:48 -0800 Subject: [PATCH] [quant] add docs for embedding/embedding_bag Summary: Test Plan: tested locally on mac Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned] --- docs/source/torch.nn.quantized.rst | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/docs/source/torch.nn.quantized.rst b/docs/source/torch.nn.quantized.rst index aeb3b55cd5fd..ef53d99e7490 100644 --- 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: