Skip to content

Commit

Permalink
Deploying to gh-pages from @ 2c8b1fa 🚀
Browse files Browse the repository at this point in the history
  • Loading branch information
facebook-github-bot committed Jun 28, 2023
1 parent 2591a2b commit ee68354
Show file tree
Hide file tree
Showing 5 changed files with 8 additions and 8 deletions.
2 changes: 1 addition & 1 deletion cpp-api/sparse_ops.html
Original file line number Diff line number Diff line change
Expand Up @@ -337,7 +337,7 @@ <h1>Sparse Data Operators<a class="headerlink" href="#sparse-data-operators" tit
<h2>Sparse Data CUDA Operators<a class="headerlink" href="#sparse-data-cuda-operators" title="Permalink to this headline"></a></h2>
<dl class="cpp function">
<dt id="_CPPv431expand_into_jagged_permute_cudaRKN2at6TensorERKN2at6TensorERKN2at6TensorE7int64_t">
<span id="_CPPv331expand_into_jagged_permute_cudaRKN2at6TensorERKN2at6TensorERKN2at6TensorE7int64_t"></span><span id="_CPPv231expand_into_jagged_permute_cudaRKN2at6TensorERKN2at6TensorERKN2at6TensorE7int64_t"></span><span id="expand_into_jagged_permute_cuda__at::TensorCR.at::TensorCR.at::TensorCR.int64_t"></span><span class="target" id="group__sparse-data-cuda_1ga6c201cc917e8a241f860df46971c41fe"></span>at::Tensor <code class="sig-name descname">expand_into_jagged_permute_cuda</code><span class="sig-paren">(</span><em class="property">const</em> at::Tensor &amp;<em>permute</em>, <em class="property">const</em> at::Tensor &amp;<em>input_offsets</em>, <em class="property">const</em> at::Tensor &amp;<em>output_offsets</em>, int64_t <em>output_size</em><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv431expand_into_jagged_permute_cudaRKN2at6TensorERKN2at6TensorERKN2at6TensorE7int64_t" title="Permalink to this definition"></a><br /></dt>
<span id="_CPPv331expand_into_jagged_permute_cudaRKN2at6TensorERKN2at6TensorERKN2at6TensorE7int64_t"></span><span id="_CPPv231expand_into_jagged_permute_cudaRKN2at6TensorERKN2at6TensorERKN2at6TensorE7int64_t"></span><span id="expand_into_jagged_permute_cuda__at::TensorCR.at::TensorCR.at::TensorCR.int64_t"></span><span class="target" id="group__sparse-data-cuda_1ga2402de1c0102b21af5f2bd5a50d30309"></span>at::Tensor <code class="sig-name descname">expand_into_jagged_permute_cuda</code><span class="sig-paren">(</span><em class="property">const</em> at::Tensor &amp;<em>permute</em>, <em class="property">const</em> at::Tensor &amp;<em>input_offsets</em>, <em class="property">const</em> at::Tensor &amp;<em>output_offsets</em>, int64_t <em>output_size</em><span class="sig-paren">)</span><a class="headerlink" href="#_CPPv431expand_into_jagged_permute_cudaRKN2at6TensorERKN2at6TensorERKN2at6TensorE7int64_t" title="Permalink to this definition"></a><br /></dt>
<dd><p>expand_into_jagged_permute expand the sparse data permute index from table dimension to batch dimension, for cases where the sparse features has different batch sizes across ranks.</p>
<p><dl class="simple">
<dt><strong>Return</strong></dt><dd><p>The output follows the following formula: <div class="highlight-cpp notranslate"><div class="highlight"><pre><span></span><span class="n">output_permute</span><span class="p">[</span><span class="n">table_offset</span><span class="p">[</span><span class="n">permute</span><span class="p">[</span><span class="n">table</span><span class="p">]]</span><span class="w"> </span><span class="o">+</span><span class="w"> </span><span class="n">batch</span><span class="p">]</span><span class="w"> </span><span class="o">&lt;-</span><span class="w"> </span><span class="n">bag_offset</span><span class="p">[</span><span class="n">batch</span><span class="p">]</span>
Expand Down
10 changes: 5 additions & 5 deletions group__sparse-data-cuda.html
Original file line number Diff line number Diff line change
Expand Up @@ -77,20 +77,20 @@
<table class="memberdecls">
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a id="func-members" name="func-members"></a>
Functions</h2></td></tr>
<tr class="memitem:ga6c201cc917e8a241f860df46971c41fe"><td class="memItemLeft" align="right" valign="top">at::Tensor&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__sparse-data-cuda.html#ga6c201cc917e8a241f860df46971c41fe">expand_into_jagged_permute_cuda</a> (const at::Tensor &amp;permute, const at::Tensor &amp;input_offsets, const at::Tensor &amp;output_offsets, int64_t output_size)</td></tr>
<tr class="separator:ga6c201cc917e8a241f860df46971c41fe"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ga2402de1c0102b21af5f2bd5a50d30309"><td class="memItemLeft" align="right" valign="top">at::Tensor&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="group__sparse-data-cuda.html#ga2402de1c0102b21af5f2bd5a50d30309">expand_into_jagged_permute_cuda</a> (const at::Tensor &amp;permute, const at::Tensor &amp;input_offsets, const at::Tensor &amp;output_offsets, int64_t output_size)</td></tr>
<tr class="separator:ga2402de1c0102b21af5f2bd5a50d30309"><td class="memSeparator" colspan="2">&#160;</td></tr>
</table>
<a name="details" id="details"></a><h2 class="groupheader">Detailed Description</h2>
<p>The following are CUDA operators </p>
<h2 class="groupheader">Function Documentation</h2>
<a id="ga6c201cc917e8a241f860df46971c41fe" name="ga6c201cc917e8a241f860df46971c41fe"></a>
<h2 class="memtitle"><span class="permalink"><a href="#ga6c201cc917e8a241f860df46971c41fe">&#9670;&#160;</a></span>expand_into_jagged_permute_cuda()</h2>
<a id="ga2402de1c0102b21af5f2bd5a50d30309" name="ga2402de1c0102b21af5f2bd5a50d30309"></a>
<h2 class="memtitle"><span class="permalink"><a href="#ga2402de1c0102b21af5f2bd5a50d30309">&#9670;&#160;</a></span>expand_into_jagged_permute_cuda()</h2>

<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">at::Tensor expand_into_jagged_permute_cuda </td>
<td class="memname">DLL_PUBLIC Tensor expand_into_jagged_permute_cuda </td>
<td>(</td>
<td class="paramtype">const at::Tensor &amp;&#160;</td>
<td class="paramname"><em>permute</em>, </td>
Expand Down
Binary file modified objects.inv
Binary file not shown.
2 changes: 1 addition & 1 deletion search/all_2.js
Original file line number Diff line number Diff line change
Expand Up @@ -2,5 +2,5 @@ var searchData=
[
['embedding_20cpu_20operators_0',['Embedding CPU Operators',['../group__embedding-cpu.html',1,'']]],
['embedding_20cuda_20operators_1',['Embedding CUDA Operators',['../group__embedding-cuda.html',1,'']]],
['expand_5finto_5fjagged_5fpermute_5fcuda_2',['expand_into_jagged_permute_cuda',['../group__sparse-data-cuda.html#ga6c201cc917e8a241f860df46971c41fe',1,'fbgemm_gpu']]]
['expand_5finto_5fjagged_5fpermute_5fcuda_2',['expand_into_jagged_permute_cuda',['../group__sparse-data-cuda.html#ga2402de1c0102b21af5f2bd5a50d30309',1,'fbgemm_gpu']]]
];
2 changes: 1 addition & 1 deletion search/functions_1.js
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
var searchData=
[
['expand_5finto_5fjagged_5fpermute_5fcuda_0',['expand_into_jagged_permute_cuda',['../group__sparse-data-cuda.html#ga6c201cc917e8a241f860df46971c41fe',1,'fbgemm_gpu']]]
['expand_5finto_5fjagged_5fpermute_5fcuda_0',['expand_into_jagged_permute_cuda',['../group__sparse-data-cuda.html#ga2402de1c0102b21af5f2bd5a50d30309',1,'fbgemm_gpu']]]
];

0 comments on commit ee68354

Please sign in to comment.