Table Batched Embedding (TBE) Operators
=======================================

.. automodule:: fbgemm_gpu

.. autofunction:: fbgemm_gpu.split_table_batched_embeddings_ops.SplitTableBatchedEmbeddingBagsCodegen