Skip to content

Commit

Permalink
Deploying to gh-pages from @ c5e1cde 🚀
Browse files Browse the repository at this point in the history
  • Loading branch information
facebook-github-bot committed Jan 29, 2025
1 parent 83831e4 commit e60c7dd
Show file tree
Hide file tree
Showing 11 changed files with 66 additions and 61 deletions.
1 change: 1 addition & 0 deletions doxygen_crawl.html
Original file line number Diff line number Diff line change
Expand Up @@ -598,6 +598,7 @@
<a href="namespacefbgemm.html#a0bfa499fd1b485bc3e457842343bca57"/>
<a href="namespacefbgemm.html#a0e2b50c7d828e56f78cc0b8368dee35a"/>
<a href="namespacefbgemm.html#a0f66af5e8e787dc1ff6893ac75ae161f"/>
<a href="namespacefbgemm.html#a150beacb26a913c4a012ec1d4a092eaf"/>
<a href="namespacefbgemm.html#a1671cc912f6aa4bab678a0d255c8a690"/>
<a href="namespacefbgemm.html#a18832d0507cb6d8bce78371b97f66479"/>
<a href="namespacefbgemm.html#a19217d3a8551a7b251ecb1eba79669bc"/>
Expand Down
6 changes: 3 additions & 3 deletions fbgemm_gpu-python-api/pooled_embedding_modules.html
Original file line number Diff line number Diff line change
Expand Up @@ -452,7 +452,7 @@
<span id="pooled-embedding-modules-stable-api"></span><h2>Stable API<a class="headerlink" href="#stable-api" title="Permalink to this heading"></a></h2>
<dl class="py class">
<dt class="sig sig-object py" id="fbgemm_gpu.permute_pooled_embedding_modules.PermutePooledEmbeddings">
<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.7.0a0+git5a4d959 ))"><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>
<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.7.0a0+git2f24f2e ))"><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>
<dd><p>A module for permuting embedding outputs along the feature dimension</p>
<p>An embedding output tensor contains the embedding outputs for all features
in a batch. It is represented in a 2D format, where the rows are the batch
Expand Down Expand Up @@ -510,14 +510,14 @@
<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
permuted. <cite>permute[i]</cite> is to permute feature <cite>permute[i]</cite> to
position <cite>i</cite>.</p></li>
<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.7.0a0+git5a4d959 ))"><em>torch.device</em></a><em>] </em><em>= None</em>) – The device to run this module
<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.7.0a0+git2f24f2e ))"><em>torch.device</em></a><em>] </em><em>= None</em>) – The device to run this module
on</p></li>
</ul>
</dd>
</dl>
<dl class="py method">
<dt class="sig sig-object py" id="fbgemm_gpu.permute_pooled_embedding_modules.PermutePooledEmbeddings.__call__">
<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.7.0a0+git5a4d959 ))"><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.7.0a0+git5a4d959 ))"><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>
<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.7.0a0+git2f24f2e ))"><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.7.0a0+git2f24f2e ))"><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>
<dd><p>Performs pooled embedding output permutation along the feature dimension</p>
<dl class="field-list simple">
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
Expand Down
12 changes: 6 additions & 6 deletions fbgemm_gpu-python-api/tbe_ops_inference.html

Large diffs are not rendered by default.

10 changes: 5 additions & 5 deletions fbgemm_gpu-python-api/tbe_ops_training.html

Large diffs are not rendered by default.

1 change: 1 addition & 0 deletions namespacemembers.html
Original file line number Diff line number Diff line change
Expand Up @@ -186,6 +186,7 @@ <h3><a id="index_f" name="index_f"></a>- f -</h3><ul>
<li>FloatToFloat16_avx512()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a5305ff58517ca3669ec41c7f1d4817a4">fbgemm</a></li>
<li>FloatToFloat16_ref()&#160;:&#160;<a class="el" href="namespacefbgemm.html#af44c584c974f95f4866806cee3798742">fbgemm</a></li>
<li>FloatToFloat16_simd()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a53a50b113345c09b89b45834f31d52df">fbgemm</a></li>
<li>FloatToFloat16_sve2()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a150beacb26a913c4a012ec1d4a092eaf">fbgemm</a></li>
<li>FloatToFloat8_ref()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a32a45639603e1584965b471846fd067f">fbgemm</a></li>
<li>Fused8BitRowwiseQuantizedSBFloatToFloatOrHalf()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a42aefaaf238d065625a64a757f998eef">fbgemm</a></li>
<li>Fused8BitRowwiseQuantizedSBFloatToFloatOrHalfRef()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a19217d3a8551a7b251ecb1eba79669bc">fbgemm</a></li>
Expand Down
1 change: 1 addition & 0 deletions namespacemembers_func.html
Original file line number Diff line number Diff line change
Expand Up @@ -186,6 +186,7 @@ <h3><a id="index_f" name="index_f"></a>- f -</h3><ul>
<li>FloatToFloat16_avx512()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a5305ff58517ca3669ec41c7f1d4817a4">fbgemm</a></li>
<li>FloatToFloat16_ref()&#160;:&#160;<a class="el" href="namespacefbgemm.html#af44c584c974f95f4866806cee3798742">fbgemm</a></li>
<li>FloatToFloat16_simd()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a53a50b113345c09b89b45834f31d52df">fbgemm</a></li>
<li>FloatToFloat16_sve2()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a150beacb26a913c4a012ec1d4a092eaf">fbgemm</a></li>
<li>FloatToFloat8_ref()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a32a45639603e1584965b471846fd067f">fbgemm</a></li>
<li>Fused8BitRowwiseQuantizedSBFloatToFloatOrHalf()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a42aefaaf238d065625a64a757f998eef">fbgemm</a></li>
<li>Fused8BitRowwiseQuantizedSBFloatToFloatOrHalfRef()&#160;:&#160;<a class="el" href="namespacefbgemm.html#a19217d3a8551a7b251ecb1eba79669bc">fbgemm</a></li>
Expand Down
Loading

0 comments on commit e60c7dd

Please sign in to comment.