code: def input_dist(
      self,
      ctx: EmbeddingBagCollectionContext,
      features: KeyedJaggedTensor,
  ) -> Awaitable[Awaitable[KJTList]]:
      return self._embedding_bag_collection.input_dist(ctx, features)
  
imports: from functools import partial
  from typing import Any, Dict, Iterator, List, Optional, Type, Union
  import torch
  from torch import nn
  from torchrec.distributed.embedding_types import BaseEmbeddingSharder, KJTList, ShardedEmbeddingModule
  from torchrec.distributed.embeddingbag import EmbeddingBagCollectionContext, EmbeddingBagCollectionSharder, ShardedEmbeddingBagCollection
  from torchrec.distributed.types import Awaitable, LazyAwaitable, ParameterSharding, QuantizedCommCodecs, ShardingEnv, ShardingType
  from torchrec.distributed.utils import append_prefix, init_parameters
  from torchrec.modules.feature_processor_ import FeatureProcessorsCollection
  from torchrec.modules.fp_embedding_modules import apply_feature_processors_to_kjt, FeatureProcessedEmbeddingBagCollection
  from torchrec.sparse.jagged_tensor import KeyedJaggedTensor, KeyedTensor
label_desc: Use image node_id 2 for calling the ShardedFeatureProcessedEmbeddingBagCollection obj's underlying member method code with example usage: obj.input_dist(ctx, features) and returns: self