def METHOD_NAME(
    self,
    ctx: EmbeddingBagCollectionContext,
    dist_input: KJTList,
) -> List[torch.Tensor]:
    fp_features = self.apply_feature_processors_to_kjt_list(dist_input)
    return self._embedding_bag_collection.METHOD_NAME(ctx, fp_features)