pylibwholegraph.torch.embedding.WholeMemoryEmbedding
====================================================

.. currentmodule:: pylibwholegraph.torch.embedding

.. autoclass:: WholeMemoryEmbedding

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~WholeMemoryEmbedding.__init__
      ~WholeMemoryEmbedding.add_gradients
      ~WholeMemoryEmbedding.apply_gradients
      ~WholeMemoryEmbedding.dim
      ~WholeMemoryEmbedding.drop_all_cache
      ~WholeMemoryEmbedding.gather
      ~WholeMemoryEmbedding.get_embedding_tensor
      ~WholeMemoryEmbedding.get_optimizer_state
      ~WholeMemoryEmbedding.get_optimizer_state_names
      ~WholeMemoryEmbedding.load
      ~WholeMemoryEmbedding.need_grad
      ~WholeMemoryEmbedding.save
      ~WholeMemoryEmbedding.set_adjust_cache
      ~WholeMemoryEmbedding.writeback_all_cache
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~WholeMemoryEmbedding.shape