def call(self, wordpiece_ids_batch, num_wordpieces, application_span_begin,
           application_span_end, application_rule_idx):
    """Returns scores for a batch of anchored rule applications.

    Args:
      wordpiece_ids_batch: <int>[batch_size, max_num_wordpieces]
      num_wordpieces: <int>[batch_size, 1]
      application_span_begin: <int>[batch_size, max_num_applications]
      application_span_end: <int>[batch_size, max_num_applications]
      application_rule_idx: <int>[batch_size, max_num_applications]

    Returns:
      application_scores: <float>[batch_size, max_num_applications]
    """
    wordpiece_encodings_batch = self.get_wordpiece_encodings(
        wordpiece_ids_batch, num_wordpieces)
    application_scores_batch = self.application_score_layer(
        wordpiece_encodings_batch, application_span_begin, application_span_end,
        application_rule_idx)
    return application_scores_batch
Returns scores for a batch of anchored rule applications.

    Args:
      wordpiece_ids_batch: <int>[batch_size, max_num_wordpieces]
      num_wordpieces: <int>[batch_size, 1]
      application_span_begin: <int>[batch_size, max_num_applications]
      application_span_end: <int>[batch_size, max_num_applications]
      application_rule_idx: <int>[batch_size, max_num_applications]

    Returns:
      application_scores: <float>[batch_size, max_num_applications]
    
language/compgen/nqg/model/parser/nqg_model.py