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

.. _tbe-ops-inference-stable-api:

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

.. autoclass:: fbgemm_gpu.split_table_batched_embeddings_ops_inference.IntNBitTableBatchedEmbeddingBagsCodegen
    :members: forward,
              fill_random_weights,
              assign_embedding_weights,
              split_embedding_weights,
              split_embedding_weights_with_scale_bias,
              recompute_module_buffers

Other API
---------