| """Ordered tensor contract shared by extraction conversion and verification.""" | |
| FEATURE_NAMES = ( | |
| "text_states", | |
| "query_states", | |
| "boundary_states", | |
| "boundary_mask", | |
| "start_logits", | |
| "end_logits", | |
| "inside_prefix", | |
| "inside_prefix_mean", | |
| "pool_start_projection", | |
| "pool_end_projection", | |
| "null_logits", | |
| "count_log_rates", | |
| "classification_logits", | |
| ) | |
| SCORE_INPUT_NAMES = ( | |
| "text_states", | |
| "text_mask", | |
| "query_states", | |
| "query_mask", | |
| "boundary_states", | |
| "start_logits", | |
| "end_logits", | |
| "inside_prefix", | |
| "inside_prefix_mean", | |
| "candidate_indices", | |
| "candidate_mask", | |
| "candidate_compatibility", | |
| ) | |