Table Batched Embedding (TBE) Training Module
=============================================

.. _tbe-ops-training-stable-api:

Stable API
----------

.. autoclass:: fbgemm_gpu.split_table_batched_embeddings_ops_training.SplitTableBatchedEmbeddingBagsCodegen
    :members: forward,
              split_embedding_weights,
              split_optimizer_states,
              set_learning_rate,
              update_hyper_parameters,
              set_optimizer_step

Other API
---------