Jagged Tensor Operators
=======================

.. automodule:: fbgemm_gpu


.. autofunction:: torch.ops.fbgemm.jagged_2d_to_dense


.. autofunction:: torch.ops.fbgemm.jagged_1d_to_dense


.. autofunction:: torch.ops.fbgemm.dense_to_jagged


.. autofunction:: torch.ops.fbgemm.jagged_to_padded_dense


.. autofunction:: torch.ops.fbgemm.jagged_dense_elementwise_add


.. autofunction:: torch.ops.fbgemm.jagged_dense_elementwise_add_jagged_output


.. autofunction:: torch.ops.fbgemm.jagged_dense_dense_elementwise_add_jagged_output


.. autofunction:: torch.ops.fbgemm.jagged_dense_elementwise_mul


.. autofunction:: torch.ops.fbgemm.batched_dense_vec_jagged_2d_mul


.. autofunction:: torch.ops.fbgemm.stacked_jagged_1d_to_dense


.. autofunction:: torch.ops.fbgemm.stacked_jagged_2d_to_dense
