Skip to content

Commit 2b2fdbc

Browse files
Deploying to gh-pages from @ 488c0de 🚀
1 parent 80abb8a commit 2b2fdbc

6 files changed

+45
-45
lines changed

fbgemm_gpu-python-api/pooled_embedding_modules.html

+3-3
Original file line numberDiff line numberDiff line change
@@ -456,7 +456,7 @@
456456
<span id="pooled-embedding-modules-stable-api"></span><h2>Stable API<a class="headerlink" href="#stable-api" title="Permalink to this heading"></a></h2>
457457
<dl class="py class">
458458
<dt class="sig sig-object py" id="fbgemm_gpu.permute_pooled_embedding_modules.PermutePooledEmbeddings">
459-
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">fbgemm_gpu.permute_pooled_embedding_modules.</span></span><span class="sig-name descname"><span class="pre">PermutePooledEmbeddings</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">embs_dims</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.List" title="(in Python v3.13)"><span class="pre">List</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.13)"><span class="pre">int</span></a><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">permute</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.List" title="(in Python v3.13)"><span class="pre">List</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.13)"><span class="pre">int</span></a><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://pytorch.org/docs/main/tensor_attributes.html#torch.device" title="(in PyTorch vmain (2.8.0a0+git1e37e5b ))"><span class="pre">device</span></a><span class="w"> </span><span class="p"><span class="pre">|</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.13)"><span class="pre">None</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/fbgemm_gpu/permute_pooled_embedding_modules.html#PermutePooledEmbeddings"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#fbgemm_gpu.permute_pooled_embedding_modules.PermutePooledEmbeddings" title="Permalink to this definition"></a></dt>
459+
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">fbgemm_gpu.permute_pooled_embedding_modules.</span></span><span class="sig-name descname"><span class="pre">PermutePooledEmbeddings</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">embs_dims</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.List" title="(in Python v3.13)"><span class="pre">List</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.13)"><span class="pre">int</span></a><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">permute</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.List" title="(in Python v3.13)"><span class="pre">List</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.13)"><span class="pre">int</span></a><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://pytorch.org/docs/main/tensor_attributes.html#torch.device" title="(in PyTorch vmain (2.8.0a0+git49570cb ))"><span class="pre">device</span></a><span class="w"> </span><span class="p"><span class="pre">|</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.13)"><span class="pre">None</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/fbgemm_gpu/permute_pooled_embedding_modules.html#PermutePooledEmbeddings"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#fbgemm_gpu.permute_pooled_embedding_modules.PermutePooledEmbeddings" title="Permalink to this definition"></a></dt>
460460
<dd><p>A module for permuting embedding outputs along the feature dimension</p>
461461
<p>An embedding output tensor contains the embedding outputs for all features
462462
in a batch. It is represented in a 2D format, where the rows are the batch
@@ -514,14 +514,14 @@
514514
<li><p><strong>permute</strong> (<em>List</em><em>[</em><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.13)"><em>int</em></a><em>]</em>) – A list that describes how each feature is
515515
permuted. <cite>permute[i]</cite> is to permute feature <cite>permute[i]</cite> to
516516
position <cite>i</cite>.</p></li>
517-
<li><p><strong>device</strong> (<em>Optional</em><em>[</em><a class="reference external" href="https://pytorch.org/docs/main/tensor_attributes.html#torch.device" title="(in PyTorch vmain (2.8.0a0+git1e37e5b ))"><em>torch.device</em></a><em>] </em><em>= None</em>) – The device to run this module
517+
<li><p><strong>device</strong> (<em>Optional</em><em>[</em><a class="reference external" href="https://pytorch.org/docs/main/tensor_attributes.html#torch.device" title="(in PyTorch vmain (2.8.0a0+git49570cb ))"><em>torch.device</em></a><em>] </em><em>= None</em>) – The device to run this module
518518
on</p></li>
519519
</ul>
520520
</dd>
521521
</dl>
522522
<dl class="py method">
523523
<dt class="sig sig-object py" id="fbgemm_gpu.permute_pooled_embedding_modules.PermutePooledEmbeddings.__call__">
524-
<span class="sig-name descname"><span class="pre">__call__</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pooled_embs</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://pytorch.org/docs/main/tensors.html#torch.Tensor" title="(in PyTorch vmain (2.8.0a0+git1e37e5b ))"><span class="pre">Tensor</span></a></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://pytorch.org/docs/main/tensors.html#torch.Tensor" title="(in PyTorch vmain (2.8.0a0+git1e37e5b ))"><span class="pre">Tensor</span></a></span></span><a class="reference internal" href="../_modules/fbgemm_gpu/permute_pooled_embedding_modules.html#PermutePooledEmbeddings.__call__"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#fbgemm_gpu.permute_pooled_embedding_modules.PermutePooledEmbeddings.__call__" title="Permalink to this definition"></a></dt>
524+
<span class="sig-name descname"><span class="pre">__call__</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pooled_embs</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://pytorch.org/docs/main/tensors.html#torch.Tensor" title="(in PyTorch vmain (2.8.0a0+git49570cb ))"><span class="pre">Tensor</span></a></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://pytorch.org/docs/main/tensors.html#torch.Tensor" title="(in PyTorch vmain (2.8.0a0+git49570cb ))"><span class="pre">Tensor</span></a></span></span><a class="reference internal" href="../_modules/fbgemm_gpu/permute_pooled_embedding_modules.html#PermutePooledEmbeddings.__call__"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#fbgemm_gpu.permute_pooled_embedding_modules.PermutePooledEmbeddings.__call__" title="Permalink to this definition"></a></dt>
525525
<dd><p>Performs pooled embedding output permutation along the feature dimension</p>
526526
<dl class="field-list simple">
527527
<dt class="field-odd">Parameters<span class="colon">:</span></dt>

fbgemm_gpu-python-api/tbe_ops_inference.html

+6-6
Large diffs are not rendered by default.

fbgemm_gpu-python-api/tbe_ops_training.html

+5-5
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)