class_docstr: Distributes pooled embeddings to be data-parallel.
class_docstr_tok: ["Distributes","pooled","embeddings","to","be","data-parallel","."]
docstr: No-op as pooled embeddings are already distributed in data-parallel fashion.
  
  Args:
      local_embs (torch.Tensor): output sequence embeddings.
  
  Returns:
      Awaitable[torch.Tensor]: awaitable of pooled embeddings tensor.
docstr_tok: ["No-op","as","pooled","embeddings","are","already","distributed","in","data-parallel","fashion",".","Args",":","local_embs","(","torch.Tensor",")",":","output","sequence","embeddings",".","Returns",":","Awaitable","[","torch.Tensor","]",":","awaitable","of","pooled","embeddings","tensor","."]
code: def forward(
      self,
      local_embs: torch.Tensor,
      sharding_ctx: Optional[EmbeddingShardingContext] = None,
  ) -> Awaitable[torch.Tensor]:
      """
      No-op as pooled embeddings are already distributed in data-parallel fashion.
  
      Args:
          local_embs (torch.Tensor): output sequence embeddings.
  
      Returns:
          Awaitable[torch.Tensor]: awaitable of pooled embeddings tensor.
      """
  
      return NoWait(local_embs)
  
imports: from typing import Any, cast, Dict, List, Optional, TypeVar
  import torch
  from torchrec.distributed.embedding_lookup import GroupedPooledEmbeddingsLookup
  from torchrec.distributed.embedding_sharding import BaseEmbeddingDist, BaseEmbeddingLookup, BaseSparseFeaturesDist, EmbeddingSharding, EmbeddingShardingContext, EmbeddingShardingInfo, group_tables
  from torchrec.distributed.embedding_types import BaseGroupedFeatureProcessor, EmbeddingComputeKernel, GroupedEmbeddingConfig, ShardedEmbeddingTable
  from torchrec.distributed.types import Awaitable, NoWait, ShardingEnv, ShardMetadata
  from torchrec.sparse.jagged_tensor import KeyedJaggedTensor
  from torchrec.streamable import Multistreamable
label_desc: Use image node_id 2 for calling the DpPooledEmbeddingDist obj's underlying member method code with example usage: obj.forward(local_embs, sharding_ctx) and returns: NoWait