lm_polygraph.stat_calculators.attention_forward_pass module

class lm_polygraph.stat_calculators.attention_forward_pass.AttentionForwardPassCalculator[source]

Bases: StatCalculator

For Whitebox model (lm_polygraph.WhiteboxModel), at input texts batch calculates: * attention masks across the model (if applicable)

static meta_info() → Tuple[List[str], List[str]][source]

Returns the statistics and dependencies for the calculator.