| transformer_lens/ActivationCache.py |
_normalize_projection_to_2d |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache._batch_size |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.remove_batch_dim |
9 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.__repr__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/ActivationCache.py |
ActivationCache.__getitem__ |
8 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.__len__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.to |
3 |
3 |
0 |
0 |
0 |
0% |
| transformer_lens/ActivationCache.py |
ActivationCache.toggle_autodiff |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.keys |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.values |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.items |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.__iter__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.apply_slice_to_batch_dim |
8 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.accumulated_resid |
25 |
0 |
0 |
14 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.logit_attrs |
22 |
0 |
0 |
16 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.decompose_resid |
34 |
0 |
0 |
22 |
2 |
96% |
| transformer_lens/ActivationCache.py |
ActivationCache.compute_head_results |
15 |
4 |
0 |
10 |
1 |
64% |
| transformer_lens/ActivationCache.py |
ActivationCache.ssm_layers |
11 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache._over_ssm_layers |
9 |
1 |
0 |
4 |
1 |
85% |
| transformer_lens/ActivationCache.py |
ActivationCache.compute_ssm_effective_attention |
9 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.compute_ssm_effective_attention._call |
10 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.compute_ssm_state |
10 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.compute_ssm_state._call |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.stack_head_results |
27 |
0 |
0 |
16 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.stack_activation |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/ActivationCache.py |
ActivationCache.get_neuron_results |
18 |
0 |
0 |
12 |
2 |
93% |
| transformer_lens/ActivationCache.py |
ActivationCache._get_cached_ln_scale |
11 |
2 |
0 |
4 |
0 |
87% |
| transformer_lens/ActivationCache.py |
ActivationCache._stack_neuron_results_apply_ln_projected |
23 |
3 |
0 |
6 |
2 |
83% |
| transformer_lens/ActivationCache.py |
ActivationCache.stack_neuron_results |
55 |
3 |
0 |
34 |
4 |
92% |
| transformer_lens/ActivationCache.py |
ActivationCache.apply_ln_to_stack |
27 |
1 |
0 |
20 |
1 |
96% |
| transformer_lens/ActivationCache.py |
ActivationCache.get_full_resid_decomposition |
60 |
2 |
0 |
38 |
4 |
94% |
| transformer_lens/ActivationCache.py |
ActivationCache.get_full_resid_decomposition._ln_then_project |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/ActivationCache.py |
(no function) |
42 |
0 |
2 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
TensorLike.ndim |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
TensorLike.shape |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
TensorLike.size |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
TensorLike.unsqueeze |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
TensorLike.squeeze |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
TensorLike.broadcast_to |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
TensorLike.__matmul__ |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__init__ |
16 |
2 |
0 |
0 |
0 |
88% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__matmul__ |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__matmul__ |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__matmul__ |
8 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__rmatmul__ |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__rmatmul__ |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__rmatmul__ |
8 |
1 |
0 |
6 |
1 |
86% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__mul__ |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__rmul__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.AB |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.BA |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.T |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix._svd_cached |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.svd |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.U |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.S |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.V |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.Vh |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.eigenvalues |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix._convert_to_slice |
8 |
0 |
0 |
4 |
1 |
92% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__getitem__ |
13 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.norm |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.__repr__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.make_even |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.get_corner |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.ndim |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.collapse_l |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.collapse_r |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.unsqueeze |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
FactoredMatrix.pair |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/FactoredMatrix.py |
(no function) |
46 |
0 |
47 |
0 |
0 |
100% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.setup |
9 |
9 |
0 |
6 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.hook_points |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.remove_all_hook_fns |
2 |
2 |
0 |
2 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.clear_contexts |
2 |
2 |
0 |
2 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.reset_hooks |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.check_and_add_hook |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.check_hooks_to_add |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.add_hook |
7 |
7 |
0 |
6 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.add_perma_hook |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule._enable_hook_with_name |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule._enable_hooks_for_points |
3 |
3 |
0 |
4 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule._enable_hook |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.hooks |
10 |
10 |
0 |
6 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.run_with_hooks |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.add_caching_hooks |
21 |
21 |
0 |
16 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.add_caching_hooks.save_hook |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.run_with_cache |
7 |
7 |
0 |
2 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.get_caching_hooks |
27 |
27 |
0 |
18 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.get_caching_hooks.save_hook |
14 |
14 |
0 |
10 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.cache_all |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
HookedRootModule.cache_some |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/HookedRootModule.py |
(no function) |
36 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/SVDInterpreter.py |
SVDInterpreter.__init__ |
6 |
2 |
0 |
2 |
1 |
62% |
| transformer_lens/SVDInterpreter.py |
SVDInterpreter.get_singular_vectors |
14 |
1 |
0 |
8 |
1 |
91% |
| transformer_lens/SVDInterpreter.py |
SVDInterpreter._get_singular_vectors_from_matrix |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/SVDInterpreter.py |
SVDInterpreter._get_OV_matrix |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/SVDInterpreter.py |
SVDInterpreter._get_w_in_matrix |
9 |
1 |
0 |
4 |
1 |
85% |
| transformer_lens/SVDInterpreter.py |
SVDInterpreter._get_w_out_matrix |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/SVDInterpreter.py |
SVDInterpreter._raise_unsupported_mlp_weight |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/SVDInterpreter.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/__init__.py |
__getattr__ |
6 |
3 |
0 |
4 |
2 |
50% |
| transformer_lens/__init__.py |
__dir__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/__init__.py |
(no function) |
22 |
2 |
0 |
2 |
1 |
88% |
| transformer_lens/benchmarks/__init__.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/activation_cache.py |
benchmark_run_with_cache |
28 |
9 |
0 |
20 |
8 |
65% |
| transformer_lens/benchmarks/activation_cache.py |
benchmark_activation_cache |
27 |
23 |
0 |
12 |
1 |
13% |
| transformer_lens/benchmarks/activation_cache.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/audio.py |
_prepare_audio_encoder_input |
17 |
2 |
0 |
6 |
1 |
87% |
| transformer_lens/benchmarks/audio.py |
_prepare_audio_text_inputs |
16 |
12 |
0 |
2 |
1 |
28% |
| transformer_lens/benchmarks/audio.py |
benchmark_audio_text_forward |
21 |
14 |
0 |
10 |
1 |
39% |
| transformer_lens/benchmarks/audio.py |
benchmark_audio_forward |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/benchmarks/audio.py |
benchmark_audio_cache |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/benchmarks/audio.py |
benchmark_audio_representation_stability |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/benchmarks/audio.py |
benchmark_audio_feature_extractor |
26 |
26 |
0 |
14 |
0 |
0% |
| transformer_lens/benchmarks/audio.py |
benchmark_audio_ctc_decode |
26 |
26 |
0 |
8 |
0 |
0% |
| transformer_lens/benchmarks/audio.py |
run_audio_benchmarks |
21 |
21 |
0 |
14 |
0 |
0% |
| transformer_lens/benchmarks/audio.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/backward_gradients.py |
needs_fp32_gradients |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/backward_gradients.py |
gradient_mismatch_stats |
8 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/benchmarks/backward_gradients.py |
gradient_mismatch_is_numerical_noise |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/backward_gradients.py |
benchmark_backward_hooks |
69 |
49 |
0 |
32 |
4 |
28% |
| transformer_lens/benchmarks/backward_gradients.py |
benchmark_backward_hooks.within_noise_band |
5 |
5 |
0 |
2 |
0 |
0% |
| transformer_lens/benchmarks/backward_gradients.py |
benchmark_critical_backward_hooks |
53 |
33 |
0 |
30 |
3 |
33% |
| transformer_lens/benchmarks/backward_gradients.py |
benchmark_gradient_computation |
30 |
15 |
0 |
16 |
5 |
43% |
| transformer_lens/benchmarks/backward_gradients.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/component_benchmark.py |
benchmark_all_components |
30 |
21 |
0 |
18 |
3 |
25% |
| transformer_lens/benchmarks/component_benchmark.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentTestResult.get_failure_severity |
11 |
11 |
0 |
10 |
0 |
0% |
| transformer_lens/benchmarks/component_outputs.py |
BenchmarkReport.pass_rate |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/benchmarks/component_outputs.py |
BenchmarkReport.print_summary |
19 |
9 |
0 |
10 |
2 |
41% |
| transformer_lens/benchmarks/component_outputs.py |
BenchmarkReport._print_component_result |
16 |
16 |
0 |
8 |
0 |
0% |
| transformer_lens/benchmarks/component_outputs.py |
BenchmarkReport.get_component_type_summary |
10 |
1 |
0 |
6 |
1 |
88% |
| transformer_lens/benchmarks/component_outputs.py |
BenchmarkReport.get_failure_by_severity |
7 |
7 |
0 |
6 |
0 |
0% |
| transformer_lens/benchmarks/component_outputs.py |
BenchmarkReport.print_detailed_analysis |
22 |
22 |
0 |
12 |
0 |
0% |
| transformer_lens/benchmarks/component_outputs.py |
_is_ssm_mixer_internal |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentBenchmarker._is_delegated_block |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentBenchmarker.__init__ |
32 |
11 |
0 |
10 |
4 |
64% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentBenchmarker.benchmark_all_components |
39 |
3 |
0 |
28 |
7 |
85% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentBenchmarker._test_component_recursive |
79 |
12 |
0 |
52 |
15 |
79% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentBenchmarker._test_component |
56 |
13 |
0 |
30 |
5 |
77% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentBenchmarker._add_direct_attention_mask_if_needed |
14 |
3 |
0 |
8 |
3 |
73% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentBenchmarker._run_component |
64 |
31 |
0 |
30 |
6 |
54% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentBenchmarker._get_test_input_for_component |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentBenchmarker._generate_test_inputs |
15 |
3 |
0 |
4 |
2 |
74% |
| transformer_lens/benchmarks/component_outputs.py |
ComponentBenchmarker._compare_outputs |
11 |
1 |
0 |
2 |
1 |
85% |
| transformer_lens/benchmarks/component_outputs.py |
benchmark_model |
17 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/benchmarks/component_outputs.py |
(no function) |
48 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/encoder_common.py |
extract_encoder_states |
7 |
7 |
0 |
6 |
0 |
0% |
| transformer_lens/benchmarks/encoder_common.py |
benchmark_encoder_forward |
27 |
11 |
0 |
14 |
3 |
56% |
| transformer_lens/benchmarks/encoder_common.py |
benchmark_encoder_cache |
25 |
5 |
0 |
10 |
3 |
77% |
| transformer_lens/benchmarks/encoder_common.py |
benchmark_encoder_representation_stability |
20 |
17 |
0 |
4 |
1 |
17% |
| transformer_lens/benchmarks/encoder_common.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/forward_pass.py |
_compute_self_target_loss |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/forward_pass.py |
_is_encoder_decoder |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/benchmarks/forward_pass.py |
_get_decoder_input_ids |
11 |
11 |
0 |
8 |
0 |
0% |
| transformer_lens/benchmarks/forward_pass.py |
benchmark_forward_pass |
50 |
38 |
0 |
24 |
4 |
22% |
| transformer_lens/benchmarks/forward_pass.py |
benchmark_loss_equivalence |
12 |
8 |
0 |
6 |
1 |
28% |
| transformer_lens/benchmarks/forward_pass.py |
benchmark_logits_equivalence |
12 |
12 |
0 |
6 |
0 |
0% |
| transformer_lens/benchmarks/forward_pass.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/generation.py |
resolve_text_generator |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/benchmarks/generation.py |
benchmark_generation |
24 |
5 |
0 |
12 |
5 |
72% |
| transformer_lens/benchmarks/generation.py |
benchmark_generation_with_kv_cache |
12 |
12 |
0 |
6 |
0 |
0% |
| transformer_lens/benchmarks/generation.py |
benchmark_multiple_generation_calls |
16 |
16 |
0 |
8 |
0 |
0% |
| transformer_lens/benchmarks/generation.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/hook_registration.py |
benchmark_hook_registry |
20 |
14 |
0 |
10 |
3 |
30% |
| transformer_lens/benchmarks/hook_registration.py |
benchmark_forward_hooks |
44 |
23 |
0 |
26 |
5 |
43% |
| transformer_lens/benchmarks/hook_registration.py |
benchmark_gated_hooks_fire |
58 |
13 |
0 |
26 |
4 |
80% |
| transformer_lens/benchmarks/hook_registration.py |
benchmark_critical_forward_hooks |
47 |
29 |
0 |
28 |
3 |
33% |
| transformer_lens/benchmarks/hook_registration.py |
benchmark_hook_functionality |
16 |
5 |
0 |
4 |
2 |
65% |
| transformer_lens/benchmarks/hook_registration.py |
benchmark_hook_functionality.ablation_hook |
14 |
8 |
0 |
4 |
1 |
39% |
| transformer_lens/benchmarks/hook_registration.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/main_benchmark.py |
_adapter_applicable_phases |
10 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/benchmarks/main_benchmark.py |
_phase_enabled |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/benchmarks/main_benchmark.py |
get_auto_model_class |
7 |
2 |
0 |
0 |
0 |
71% |
| transformer_lens/benchmarks/main_benchmark.py |
_fixup_custom_model |
30 |
26 |
0 |
20 |
2 |
12% |
| transformer_lens/benchmarks/main_benchmark.py |
_hf_forward_with_mask_fallback |
10 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/benchmarks/main_benchmark.py |
run_comparison_benchmarks |
111 |
56 |
0 |
52 |
15 |
43% |
| transformer_lens/benchmarks/main_benchmark.py |
run_comparison_benchmarks.add_result |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/benchmarks/main_benchmark.py |
run_benchmark_suite |
528 |
367 |
0 |
284 |
64 |
28% |
| transformer_lens/benchmarks/main_benchmark.py |
run_benchmark_suite.get_memory_mb |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/benchmarks/main_benchmark.py |
run_benchmark_suite.should_run_phase |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/main_benchmark.py |
run_benchmark_suite.add_result |
5 |
1 |
0 |
4 |
2 |
67% |
| transformer_lens/benchmarks/main_benchmark.py |
run_benchmark_suite.cleanup_tensors |
17 |
17 |
0 |
16 |
0 |
0% |
| transformer_lens/benchmarks/main_benchmark.py |
run_benchmark_suite.cleanup_model |
92 |
35 |
0 |
74 |
29 |
55% |
| transformer_lens/benchmarks/main_benchmark.py |
run_benchmark_suite._cleanup_bridge_unprocessed |
3 |
0 |
0 |
2 |
1 |
80% |
| transformer_lens/benchmarks/main_benchmark.py |
update_model_registry |
27 |
2 |
0 |
8 |
1 |
91% |
| transformer_lens/benchmarks/main_benchmark.py |
main |
13 |
13 |
0 |
2 |
0 |
0% |
| transformer_lens/benchmarks/main_benchmark.py |
(no function) |
30 |
1 |
0 |
2 |
1 |
94% |
| transformer_lens/benchmarks/multimodal.py |
_create_test_image |
5 |
2 |
0 |
0 |
0 |
60% |
| transformer_lens/benchmarks/multimodal.py |
_prepare_test_inputs |
34 |
5 |
0 |
18 |
4 |
83% |
| transformer_lens/benchmarks/multimodal.py |
benchmark_multimodal_forward |
20 |
20 |
0 |
10 |
0 |
0% |
| transformer_lens/benchmarks/multimodal.py |
benchmark_multimodal_generation |
22 |
22 |
0 |
12 |
0 |
0% |
| transformer_lens/benchmarks/multimodal.py |
benchmark_multimodal_cache |
17 |
17 |
0 |
8 |
0 |
0% |
| transformer_lens/benchmarks/multimodal.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/text_quality.py |
load_judge |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/text_quality.py |
_judge_perplexity |
18 |
3 |
0 |
6 |
1 |
83% |
| transformer_lens/benchmarks/text_quality.py |
_compute_repetition_penalty |
18 |
3 |
0 |
12 |
4 |
77% |
| transformer_lens/benchmarks/text_quality.py |
_ratio_to_score |
5 |
1 |
0 |
4 |
1 |
78% |
| transformer_lens/benchmarks/text_quality.py |
_wrong_language |
15 |
2 |
0 |
8 |
2 |
83% |
| transformer_lens/benchmarks/text_quality.py |
_length_penalty |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/benchmarks/text_quality.py |
_build_caption_test_images |
10 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/benchmarks/text_quality.py |
_generate_image_conditioned_captions |
25 |
5 |
0 |
8 |
3 |
76% |
| transformer_lens/benchmarks/text_quality.py |
_architecture_id |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/text_quality.py |
_resolve_lang_code |
21 |
9 |
0 |
12 |
1 |
52% |
| transformer_lens/benchmarks/text_quality.py |
_forced_bos_for_target |
19 |
6 |
0 |
8 |
2 |
70% |
| transformer_lens/benchmarks/text_quality.py |
_build_model_input |
9 |
4 |
0 |
6 |
2 |
60% |
| transformer_lens/benchmarks/text_quality.py |
benchmark_text_quality |
149 |
28 |
0 |
68 |
11 |
80% |
| transformer_lens/benchmarks/text_quality.py |
(no function) |
25 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/text_quality_profiles.py |
ProfileSpec.parse |
11 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/benchmarks/text_quality_profiles.py |
ProfileSpec.__str__ |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/benchmarks/text_quality_profiles.py |
is_default_profile |
6 |
3 |
0 |
2 |
1 |
50% |
| transformer_lens/benchmarks/text_quality_profiles.py |
p4_pass_threshold |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/text_quality_profiles.py |
extract_languages |
14 |
0 |
0 |
12 |
1 |
96% |
| transformer_lens/benchmarks/text_quality_profiles.py |
_marian_pair_from_model_id |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/benchmarks/text_quality_profiles.py |
_id_says_chat |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/benchmarks/text_quality_profiles.py |
_first_covered_language |
4 |
2 |
0 |
4 |
1 |
38% |
| transformer_lens/benchmarks/text_quality_profiles.py |
profile_from_hf_signals |
18 |
2 |
0 |
12 |
2 |
87% |
| transformer_lens/benchmarks/text_quality_profiles.py |
resolve_profile |
57 |
17 |
2 |
38 |
5 |
73% |
| transformer_lens/benchmarks/text_quality_profiles.py |
prompts_for |
20 |
3 |
0 |
18 |
3 |
84% |
| transformer_lens/benchmarks/text_quality_profiles.py |
(no function) |
58 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/utils.py |
deterministic_rng |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/utils.py |
is_tiny_test_model |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/utils.py |
filter_expected_missing_hooks |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/benchmarks/utils.py |
build_modality_input |
16 |
1 |
0 |
8 |
1 |
92% |
| transformer_lens/benchmarks/utils.py |
build_modality_input._from_config |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/benchmarks/utils.py |
safe_allclose |
7 |
4 |
0 |
4 |
2 |
45% |
| transformer_lens/benchmarks/utils.py |
BenchmarkResult.__str__ |
12 |
12 |
0 |
6 |
0 |
0% |
| transformer_lens/benchmarks/utils.py |
BenchmarkResult.print_immediate |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/benchmarks/utils.py |
make_capture_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/utils.py |
make_capture_hook.hook_fn |
6 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/benchmarks/utils.py |
make_grad_capture_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/utils.py |
make_grad_capture_hook.hook_fn |
6 |
0 |
0 |
6 |
2 |
83% |
| transformer_lens/benchmarks/utils.py |
_squeeze_batch_dim |
7 |
1 |
0 |
6 |
1 |
85% |
| transformer_lens/benchmarks/utils.py |
compare_activation_dicts |
15 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/benchmarks/utils.py |
compare_tensors |
16 |
11 |
0 |
8 |
4 |
38% |
| transformer_lens/benchmarks/utils.py |
compare_scalars |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/benchmarks/utils.py |
format_results |
38 |
38 |
0 |
14 |
0 |
0% |
| transformer_lens/benchmarks/utils.py |
bridge_self_target_loss |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/utils.py |
(no function) |
43 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/vision.py |
benchmark_vision_forward |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/vision.py |
benchmark_vision_cache |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/vision.py |
benchmark_vision_representation_stability |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/vision.py |
benchmark_vision_embeddings |
29 |
29 |
0 |
16 |
0 |
0% |
| transformer_lens/benchmarks/vision.py |
benchmark_vision_classification_decode |
29 |
21 |
0 |
8 |
1 |
35% |
| transformer_lens/benchmarks/vision.py |
run_vision_benchmarks |
21 |
21 |
0 |
14 |
0 |
0% |
| transformer_lens/benchmarks/vision.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/benchmarks/weight_processing.py |
benchmark_weight_modification |
49 |
49 |
0 |
16 |
0 |
0% |
| transformer_lens/benchmarks/weight_processing.py |
benchmark_layer_norm_folding |
33 |
33 |
0 |
14 |
0 |
0% |
| transformer_lens/benchmarks/weight_processing.py |
benchmark_attention_output_centering |
37 |
37 |
0 |
12 |
0 |
0% |
| transformer_lens/benchmarks/weight_processing.py |
benchmark_mlp_output_centering |
34 |
34 |
0 |
26 |
0 |
0% |
| transformer_lens/benchmarks/weight_processing.py |
benchmark_unembed_centering |
15 |
15 |
0 |
6 |
0 |
0% |
| transformer_lens/benchmarks/weight_processing.py |
benchmark_value_bias_folding |
20 |
20 |
0 |
12 |
0 |
0% |
| transformer_lens/benchmarks/weight_processing.py |
benchmark_no_nan_inf |
19 |
19 |
0 |
12 |
0 |
0% |
| transformer_lens/benchmarks/weight_processing.py |
benchmark_weight_magnitudes |
41 |
41 |
0 |
28 |
0 |
0% |
| transformer_lens/benchmarks/weight_processing.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/cache/__init__.py |
(no function) |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/cache/key_value_cache.py |
TransformerLensKeyValueCache.init_cache |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/cache/key_value_cache.py |
TransformerLensKeyValueCache.freeze |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/cache/key_value_cache.py |
TransformerLensKeyValueCache.unfreeze |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/cache/key_value_cache.py |
TransformerLensKeyValueCache.append_attention_mask |
5 |
5 |
0 |
2 |
0 |
0% |
| transformer_lens/cache/key_value_cache.py |
TransformerLensKeyValueCache.__getitem__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/cache/key_value_cache.py |
(no function) |
18 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/cache/key_value_cache_entry.py |
TransformerLensKeyValueCacheEntry.init_cache_entry |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/cache/key_value_cache_entry.py |
TransformerLensKeyValueCacheEntry.append |
6 |
0 |
0 |
2 |
1 |
88% |
| transformer_lens/cache/key_value_cache_entry.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/config/__init__.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/config/transformer_bridge_config.py |
TransformerBridgeConfig.__init__ |
74 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/config/transformer_bridge_config.py |
TransformerBridgeConfig.__setattr__ |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/config/transformer_bridge_config.py |
TransformerBridgeConfig.__getstate__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/config/transformer_bridge_config.py |
TransformerBridgeConfig.__setstate__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/config/transformer_bridge_config.py |
TransformerBridgeConfig._bind_bridge |
6 |
0 |
0 |
4 |
1 |
90% |
| transformer_lens/config/transformer_bridge_config.py |
TransformerBridgeConfig._set_bridge_managed_hook_flag |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/config/transformer_bridge_config.py |
TransformerBridgeConfig.__post_init__ |
9 |
1 |
0 |
10 |
2 |
84% |
| transformer_lens/config/transformer_bridge_config.py |
TransformerBridgeConfig.head_dim |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/config/transformer_bridge_config.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/config/transformer_lens_config.py |
TransformerLensConfig.__post_init__ |
8 |
1 |
0 |
8 |
1 |
88% |
| transformer_lens/config/transformer_lens_config.py |
TransformerLensConfig.unwrap |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/config/transformer_lens_config.py |
TransformerLensConfig.from_dict |
12 |
2 |
0 |
4 |
1 |
81% |
| transformer_lens/config/transformer_lens_config.py |
TransformerLensConfig.to_dict |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/config/transformer_lens_config.py |
TransformerLensConfig.__repr__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/config/transformer_lens_config.py |
(no function) |
39 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/__init__.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/__init__.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/arithmetic_tensor_conversion.py |
ArithmeticTensorConversion.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/arithmetic_tensor_conversion.py |
ArithmeticTensorConversion.handle_conversion |
9 |
0 |
0 |
8 |
1 |
94% |
| transformer_lens/conversion_utils/conversion_steps/arithmetic_tensor_conversion.py |
ArithmeticTensorConversion.revert |
11 |
11 |
0 |
8 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/arithmetic_tensor_conversion.py |
ArithmeticTensorConversion.__repr__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/arithmetic_tensor_conversion.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/attention_auto_conversion.py |
AttentionAutoConversion.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/attention_auto_conversion.py |
AttentionAutoConversion.handle_conversion |
19 |
11 |
0 |
10 |
2 |
34% |
| transformer_lens/conversion_utils/conversion_steps/attention_auto_conversion.py |
AttentionAutoConversion.revert_conversion |
13 |
13 |
0 |
10 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/attention_auto_conversion.py |
AttentionAutoConversion.clear_state |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/attention_auto_conversion.py |
AttentionAutoConversion.get_conversion_info |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/attention_auto_conversion.py |
AttentionAutoConversion.__repr__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/attention_auto_conversion.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/base_tensor_conversion.py |
BaseTensorConversion.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/base_tensor_conversion.py |
BaseTensorConversion.convert |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/base_tensor_conversion.py |
BaseTensorConversion.handle_conversion |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/base_tensor_conversion.py |
BaseTensorConversion.revert |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/base_tensor_conversion.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/callable_tensor_conversion.py |
CallableTensorConversion.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/callable_tensor_conversion.py |
CallableTensorConversion.handle_conversion |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/callable_tensor_conversion.py |
CallableTensorConversion.__repr__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/callable_tensor_conversion.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/chain_tensor_conversion.py |
ChainTensorConversion.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/chain_tensor_conversion.py |
ChainTensorConversion.handle_conversion |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/chain_tensor_conversion.py |
ChainTensorConversion.revert |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/chain_tensor_conversion.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/rearrange_tensor_conversion.py |
RearrangeTensorConversion.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/rearrange_tensor_conversion.py |
RearrangeTensorConversion.handle_conversion |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/rearrange_tensor_conversion.py |
RearrangeTensorConversion.revert |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/rearrange_tensor_conversion.py |
RearrangeTensorConversion.__repr__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/rearrange_tensor_conversion.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/repeat_tensor_conversion.py |
RepeatTensorConversion.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/repeat_tensor_conversion.py |
RepeatTensorConversion.handle_conversion |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/repeat_tensor_conversion.py |
RepeatTensorConversion.__repr__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/repeat_tensor_conversion.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/split_tensor_conversion.py |
SplitTensorConversion.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/split_tensor_conversion.py |
SplitTensorConversion.handle_conversion |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/split_tensor_conversion.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/tensor_conversion_set.py |
TensorConversionSet.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/tensor_conversion_set.py |
TensorConversionSet.get_component |
11 |
11 |
0 |
6 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/tensor_conversion_set.py |
TensorConversionSet.handle_conversion |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/tensor_conversion_set.py |
TensorConversionSet.process_conversion_action |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/tensor_conversion_set.py |
TensorConversionSet.process_conversion |
7 |
4 |
0 |
4 |
1 |
36% |
| transformer_lens/conversion_utils/conversion_steps/tensor_conversion_set.py |
TensorConversionSet.get_conversion_action |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/tensor_conversion_set.py |
TensorConversionSet.__repr__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/tensor_conversion_set.py |
(no function) |
14 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/ternary_tensor_conversion.py |
TernaryTensorConversion.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/ternary_tensor_conversion.py |
TernaryTensorConversion.handle_conversion |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/ternary_tensor_conversion.py |
TernaryTensorConversion.handle_primary_conversion |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/ternary_tensor_conversion.py |
TernaryTensorConversion.handle_fallback_conversion |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/ternary_tensor_conversion.py |
TernaryTensorConversion.find_context_field |
5 |
0 |
0 |
4 |
1 |
89% |
| transformer_lens/conversion_utils/conversion_steps/ternary_tensor_conversion.py |
TernaryTensorConversion.__repr__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/ternary_tensor_conversion.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/transpose_tensor_conversion.py |
TransposeTensorConversion.handle_conversion |
5 |
2 |
0 |
4 |
2 |
56% |
| transformer_lens/conversion_utils/conversion_steps/transpose_tensor_conversion.py |
TransposeTensorConversion.revert |
5 |
2 |
0 |
4 |
2 |
56% |
| transformer_lens/conversion_utils/conversion_steps/transpose_tensor_conversion.py |
TransposeTensorConversion.__repr__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/transpose_tensor_conversion.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/zeros_like_conversion.py |
ZerosLikeConversion.handle_conversion |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/conversion_steps/zeros_like_conversion.py |
ZerosLikeConversion.__repr__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/conversion_utils/conversion_steps/zeros_like_conversion.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/helpers/__init__.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/helpers/find_property.py |
find_property |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/conversion_utils/helpers/find_property.py |
(no function) |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/helpers/merge_quantiziation_fields.py |
merge_quantization_fields |
15 |
1 |
0 |
10 |
1 |
92% |
| transformer_lens/conversion_utils/helpers/merge_quantiziation_fields.py |
(no function) |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/hook_conversion_utils.py |
get_weight_conversion_field_set |
14 |
3 |
0 |
10 |
1 |
75% |
| transformer_lens/conversion_utils/hook_conversion_utils.py |
model_info_cfg |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/conversion_utils/hook_conversion_utils.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/param_processing_conversion.py |
ParamProcessingConversion.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/param_processing_conversion.py |
ParamProcessingConversion._resolve_key |
5 |
1 |
0 |
2 |
1 |
71% |
| transformer_lens/conversion_utils/param_processing_conversion.py |
ParamProcessingConversion.convert |
5 |
1 |
0 |
2 |
1 |
71% |
| transformer_lens/conversion_utils/param_processing_conversion.py |
ParamProcessingConversion.revert |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/conversion_utils/param_processing_conversion.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/evals.py |
sanity_check |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/evals.py |
make_wiki_data_loader |
5 |
5 |
0 |
0 |
0 |
0% |
| transformer_lens/evals.py |
make_owt_data_loader |
5 |
5 |
0 |
0 |
0 |
0% |
| transformer_lens/evals.py |
make_pile_data_loader |
5 |
5 |
0 |
0 |
0 |
0% |
| transformer_lens/evals.py |
make_code_data_loader |
5 |
5 |
0 |
0 |
0 |
0% |
| transformer_lens/evals.py |
make_mmlu_data_loader |
21 |
4 |
0 |
10 |
1 |
84% |
| transformer_lens/evals.py |
evaluate_on_dataset |
10 |
10 |
0 |
4 |
0 |
0% |
| transformer_lens/evals.py |
induction_loss |
11 |
11 |
0 |
4 |
0 |
0% |
| transformer_lens/evals.py |
evaluate |
9 |
9 |
0 |
4 |
0 |
0% |
| transformer_lens/evals.py |
IOIDataset.__init__ |
10 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/evals.py |
IOIDataset.__len__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/evals.py |
IOIDataset.__getitem__ |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/evals.py |
IOIDataset.get_sample |
13 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/evals.py |
IOIDataset.get_default_names |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/evals.py |
IOIDataset.get_default_templates |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/evals.py |
IOIDataset.get_default_nouns |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/evals.py |
ioi_eval |
29 |
0 |
0 |
8 |
1 |
97% |
| transformer_lens/evals.py |
ioi_eval.collate |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/evals.py |
mmlu_eval |
46 |
3 |
0 |
16 |
3 |
90% |
| transformer_lens/evals.py |
(no function) |
40 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/factories/__init__.py |
(no function) |
0 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/factories/activation_function_factory.py |
ActivationFunctionFactory.pick_activation_function |
9 |
3 |
0 |
6 |
3 |
60% |
| transformer_lens/factories/activation_function_factory.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/factories/architecture_adapter_factory.py |
ArchitectureAdapterFactory.register_adapter |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/factories/architecture_adapter_factory.py |
ArchitectureAdapterFactory.discover_entry_points |
16 |
2 |
0 |
6 |
0 |
91% |
| transformer_lens/factories/architecture_adapter_factory.py |
ArchitectureAdapterFactory.select_architecture_adapter |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/factories/architecture_adapter_factory.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/head_detector.py |
detect_head |
33 |
2 |
0 |
20 |
2 |
92% |
| transformer_lens/head_detector.py |
get_previous_token_head_detection_pattern |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/head_detector.py |
get_duplicate_token_head_detection_pattern |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/head_detector.py |
get_induction_head_detection_pattern |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/head_detector.py |
get_supported_heads |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/head_detector.py |
compute_head_attention_similarity_score |
18 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/head_detector.py |
(no function) |
24 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/hook_points.py |
_ScaledGradientTensor.__init__ |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/hook_points.py |
_ScaledGradientTensor.sum |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/hook_points.py |
_ScaledGradientTensor.__getattr__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/hook_points.py |
_ScaledGradientTensor.__repr__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/hook_points.py |
_HookFunctionProtocol.__call__ |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/hook_points.py |
_AliasedHookPoint.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/hook_points.py |
_AliasedHookPoint.name |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/hook_points.py |
_AliasedHookPoint.ctx |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/hook_points.py |
_AliasedHookPoint.hook_conversion |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/hook_points.py |
_AliasedHookPoint.layer |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/hook_points.py |
HookPoint.__init__ |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/hook_points.py |
HookPoint.__repr__ |
6 |
1 |
0 |
4 |
1 |
80% |
| transformer_lens/hook_points.py |
HookPoint.add_perma_hook |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/hook_points.py |
HookPoint.add_hook |
25 |
3 |
0 |
14 |
4 |
82% |
| transformer_lens/hook_points.py |
HookPoint.add_hook.full_hook |
23 |
1 |
0 |
18 |
2 |
93% |
| transformer_lens/hook_points.py |
HookPoint.has_hooks |
8 |
1 |
0 |
6 |
1 |
86% |
| transformer_lens/hook_points.py |
HookPoint.has_hooks._has_hooks_in_direction |
7 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/hook_points.py |
HookPoint.remove_hooks |
7 |
1 |
0 |
6 |
1 |
85% |
| transformer_lens/hook_points.py |
HookPoint.remove_hooks._remove_hooks |
8 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/hook_points.py |
HookPoint.clear_context |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/hook_points.py |
HookPoint.enable_reshape |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/hook_points.py |
HookPoint.forward |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/hook_points.py |
HookPoint.layer |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/hook_points.py |
HookIntrospectionMixin.list_hooks |
24 |
0 |
0 |
18 |
0 |
100% |
| transformer_lens/hook_points.py |
__getattr__ |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/hook_points.py |
(no function) |
55 |
0 |
4 |
0 |
0 |
100% |
| transformer_lens/lit/__init__.py |
serve |
26 |
26 |
0 |
16 |
0 |
0% |
| transformer_lens/lit/__init__.py |
LITWidget.__init__ |
10 |
10 |
0 |
6 |
0 |
0% |
| transformer_lens/lit/__init__.py |
LITWidget.url |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/__init__.py |
LITWidget.render |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/__init__.py |
LITWidget.stop |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/__init__.py |
(no function) |
21 |
1 |
0 |
2 |
1 |
91% |
| transformer_lens/lit/constants.py |
(no function) |
80 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/lit/dataset.py |
_ensure_lit_available |
2 |
2 |
0 |
2 |
0 |
0% |
| transformer_lens/lit/dataset.py |
SimpleTextDataset.__init__ |
6 |
6 |
0 |
4 |
0 |
0% |
| transformer_lens/lit/dataset.py |
SimpleTextDataset.examples |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
SimpleTextDataset.__len__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
SimpleTextDataset.__iter__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
SimpleTextDataset.description |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
SimpleTextDataset.spec |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
SimpleTextDataset.from_strings |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
SimpleTextDataset.from_file |
9 |
9 |
0 |
4 |
0 |
0% |
| transformer_lens/lit/dataset.py |
PromptCompletionDataset.__init__ |
6 |
6 |
0 |
4 |
0 |
0% |
| transformer_lens/lit/dataset.py |
PromptCompletionDataset._add_example |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/lit/dataset.py |
PromptCompletionDataset.examples |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
PromptCompletionDataset.__len__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
PromptCompletionDataset.__iter__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
PromptCompletionDataset.description |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
PromptCompletionDataset.spec |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
PromptCompletionDataset.from_pairs |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
IOIDataset.__init__ |
3 |
3 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
IOIDataset.examples |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
IOIDataset.__len__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
IOIDataset.__iter__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
IOIDataset.description |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
IOIDataset.spec |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
IOIDataset.add_example |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
IOIDataset.generate |
9 |
9 |
0 |
2 |
0 |
0% |
| transformer_lens/lit/dataset.py |
InductionDataset.__init__ |
3 |
3 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
InductionDataset.examples |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
InductionDataset.__len__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
InductionDataset.__iter__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
InductionDataset.description |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
InductionDataset.spec |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
InductionDataset.add_example |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
InductionDataset.generate_simple |
11 |
11 |
0 |
2 |
0 |
0% |
| transformer_lens/lit/dataset.py |
LITDatasetWrapper.__init__ |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
LITDatasetWrapper.init_spec |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
LITDatasetWrapper.spec |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
LITDatasetWrapper.description |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
LITDatasetWrapper.examples |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
LITDatasetWrapper.__len__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
LITDatasetWrapper.__iter__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
wrap_for_lit |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/dataset.py |
(no function) |
92 |
16 |
3 |
4 |
2 |
81% |
| transformer_lens/lit/model.py |
_ensure_lit_available |
2 |
2 |
0 |
2 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT.__init__ |
13 |
13 |
0 |
6 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT.supports_concurrent_predictions |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT.description |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT.init_spec |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT.input_spec |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT.output_spec |
20 |
20 |
0 |
10 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT._get_embedding_layers |
8 |
8 |
0 |
6 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT.predict |
2 |
2 |
0 |
2 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT._predict_single |
20 |
20 |
0 |
8 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT._get_top_k_per_position |
14 |
14 |
0 |
8 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT._extract_embeddings |
12 |
12 |
0 |
2 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT._extract_attention |
6 |
6 |
0 |
2 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT._compute_gradients |
23 |
23 |
0 |
6 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT.max_minibatch_size |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT.get_embedding_table |
14 |
14 |
0 |
4 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLIT.from_pretrained |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLITBatched.__init__ |
3 |
3 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLITBatched.description |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLITBatched.init_spec |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLITBatched.input_spec |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLITBatched.output_spec |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLITBatched.max_minibatch_size |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLITBatched.predict_minibatch |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
TransformerLensLITBatched.from_pretrained |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/model.py |
(no function) |
69 |
17 |
3 |
8 |
4 |
73% |
| transformer_lens/lit/utils.py |
check_lit_installed |
5 |
1 |
0 |
0 |
0 |
80% |
| transformer_lens/lit/utils.py |
tensor_to_numpy |
7 |
1 |
0 |
6 |
1 |
85% |
| transformer_lens/lit/utils.py |
numpy_to_tensor |
10 |
3 |
0 |
8 |
3 |
67% |
| transformer_lens/lit/utils.py |
get_tokens_from_model |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/lit/utils.py |
clean_token_string |
7 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/lit/utils.py |
clean_token_strings |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/lit/utils.py |
extract_attention_from_cache |
6 |
6 |
0 |
4 |
0 |
0% |
| transformer_lens/lit/utils.py |
extract_embeddings_from_cache |
11 |
11 |
0 |
8 |
0 |
0% |
| transformer_lens/lit/utils.py |
compute_token_gradients |
13 |
13 |
0 |
2 |
0 |
0% |
| transformer_lens/lit/utils.py |
get_top_k_predictions |
8 |
8 |
0 |
2 |
0 |
0% |
| transformer_lens/lit/utils.py |
validate_input_example |
5 |
5 |
0 |
4 |
0 |
0% |
| transformer_lens/lit/utils.py |
batch_examples |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/lit/utils.py |
unbatch_outputs |
15 |
2 |
0 |
10 |
2 |
84% |
| transformer_lens/lit/utils.py |
get_hook_name_for_layer |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/utils.py |
filter_cache_by_pattern |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/lit/utils.py |
get_model_info |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/lit/utils.py |
(no function) |
22 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/__init__.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
ln_rule_grad |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_LNRule.forward |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_LNRule.backward |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
ln_rule |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_IdentityRule.forward |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_IdentityRule.backward |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
identity_rule |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_HalfRule.forward |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_HalfRule.backward |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
half_rule |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_ScaleGradient.forward |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_ScaleGradient.backward |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
scale_gradient |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_RelevanceRuleCapable._enable_relevance_rule |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_RelevanceRuleCapable._disable_relevance_rule |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_iter_canonical_mount_candidates |
8 |
0 |
0 |
6 |
1 |
93% |
| transformer_lens/model_bridge/_relevance_rules.py |
_acquire_rule |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
_release_rule |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
use_relevance_rules |
23 |
0 |
0 |
14 |
0 |
100% |
| transformer_lens/model_bridge/_relevance_rules.py |
(no function) |
51 |
0 |
7 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.native_sampler_kwargs |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.__init__ |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._check_required_libraries |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._merge_default_config |
3 |
0 |
0 |
4 |
1 |
86% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._gated_mlp |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.components |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._ungated_mlp |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._qkvo_attention_bridge |
9 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._build_attention_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._canonical_layer_types |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._extract_vision_dims |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._set_rms_rotary_defaults |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.apply_output_logits_transform |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.validate_output_logits_transform |
0 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._qkvo_weight_conversions |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._reprefix_components |
4 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.preprocess_weights |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.postprocess_weights |
0 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.get_component_mapping |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.get_remote_component |
20 |
6 |
0 |
14 |
2 |
65% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.get_component_from_list_module |
32 |
7 |
0 |
22 |
7 |
74% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.get_generalized_component |
31 |
10 |
0 |
24 |
3 |
65% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.get_component |
23 |
10 |
0 |
18 |
5 |
54% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.translate_transformer_lens_path |
74 |
16 |
0 |
56 |
8 |
75% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._preprocess_parameter_path |
91 |
25 |
0 |
46 |
12 |
72% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.convert_hf_key_to_tl_key |
37 |
1 |
0 |
32 |
5 |
91% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.prepare_loading |
4 |
0 |
0 |
4 |
1 |
88% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.prepare_model |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.create_stateful_cache |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter.setup_component_testing |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/architecture_adapter.py |
ArchitectureAdapter._wire_rotary_for_testing |
26 |
1 |
0 |
20 |
1 |
96% |
| transformer_lens/model_bridge/architecture_adapter.py |
(no function) |
65 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
build_alias_to_canonical_map |
9 |
1 |
0 |
8 |
2 |
82% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.__init__ |
19 |
1 |
0 |
6 |
1 |
92% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.tokenizer |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.tokenizer |
18 |
3 |
0 |
12 |
1 |
80% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._initialize_hook_registry |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._scan_existing_hooks |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.clear_hook_registry |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.hook_dict |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.mod_dict |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._register_aliases |
20 |
0 |
0 |
12 |
1 |
97% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._collect_component_aliases |
20 |
0 |
0 |
16 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._compute_hook_aliases_cached |
27 |
1 |
0 |
18 |
2 |
93% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._collect_hook_aliases_from_registry |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._collect_block_instance_aliases |
36 |
0 |
0 |
26 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._add_aliases_to_hooks |
18 |
1 |
0 |
10 |
2 |
89% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._replay_captures |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.forward |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.to_tokens |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.close |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._input_device |
12 |
3 |
0 |
6 |
1 |
78% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.loss_fn |
7 |
2 |
0 |
6 |
2 |
69% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._causal_labels_loss |
13 |
3 |
0 |
8 |
3 |
71% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._prepare_loss_attention_mask |
22 |
6 |
0 |
16 |
6 |
68% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._finalize_return |
45 |
13 |
0 |
30 |
7 |
71% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._check_loss_supported |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._is_embedding_stage_hook |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._check_hook_fireable |
4 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._resolve_hook_point |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.get_hook_point |
12 |
12 |
0 |
6 |
0 |
0% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.check_hooks_to_add |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._add_fn_to_hook_point |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._gated_hook_reason |
9 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._warn_gated_skipped |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.add_hook |
42 |
14 |
0 |
24 |
2 |
64% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.add_perma_hook |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.hook_points |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.clear_contexts |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.remove_all_hook_fns |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.reset_hooks |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._normalize_names_filter |
9 |
1 |
0 |
8 |
1 |
88% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore._pos_slice_dim |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.get_caching_hooks |
17 |
1 |
0 |
10 |
1 |
93% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.get_caching_hooks.save_hook |
8 |
1 |
0 |
4 |
2 |
75% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.add_caching_hooks |
15 |
4 |
0 |
12 |
1 |
67% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.add_caching_hooks._point_reason |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.cache_all |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.cache_some |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.hooks |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.hooks._hooks_context |
16 |
1 |
0 |
6 |
1 |
91% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.hooks._hooks_context.add_hook_to_point |
9 |
5 |
0 |
4 |
1 |
38% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.hooks._hooks_context.apply_hooks |
21 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_hooks |
41 |
3 |
0 |
24 |
6 |
86% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_hooks.add_hook_to_point |
21 |
8 |
0 |
14 |
3 |
63% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_hooks.apply_hooks |
28 |
4 |
0 |
16 |
2 |
86% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_hooks.apply_hooks.wrapped_hook_fn |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_hooks.stop_hook |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_cache |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_cache |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_cache |
177 |
22 |
0 |
124 |
13 |
87% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_cache.create_names_filter_fn |
18 |
1 |
0 |
14 |
1 |
94% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_cache._store |
4 |
0 |
0 |
2 |
1 |
83% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_cache.make_cache_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_cache.make_cache_hook.cache_hook |
14 |
10 |
0 |
10 |
2 |
25% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_cache.make_grad_cache_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_cache.make_grad_cache_hook.grad_hook |
3 |
0 |
0 |
2 |
1 |
80% |
| transformer_lens/model_bridge/bridge_core.py |
BridgeCore.run_with_cache.stop_hook |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/bridge_core.py |
(no function) |
85 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/buffer_restore.py |
_is_valid_inv_freq |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/buffer_restore.py |
restore_rotary_inv_freq |
23 |
3 |
0 |
12 |
1 |
89% |
| transformer_lens/model_bridge/buffer_restore.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/compat.py |
patch_dynamic_cache_v5 |
14 |
2 |
0 |
6 |
2 |
80% |
| transformer_lens/model_bridge/compat.py |
patch_dynamic_cache_v5._from_legacy_cache |
5 |
5 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/compat.py |
patch_dynamic_cache_v5._get_usable_length |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/compat.py |
patch_dynamic_cache_v5._to_legacy_cache |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/compat.py |
(no function) |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
_ContainerStateOwner.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
_ContainerStateOwner._sync_original_container |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
_ContainerStateOwner._refresh_from_original_container |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
_ContainerStateOwner._apply |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
_ContainerStateOwner._load_from_state_dict |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
refresh_container_state_owners |
6 |
0 |
0 |
6 |
1 |
92% |
| transformer_lens/model_bridge/component_setup.py |
replace_remote_component |
10 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
set_original_components |
4 |
0 |
0 |
2 |
1 |
83% |
| transformer_lens/model_bridge/component_setup.py |
_register_unowned_container_state |
25 |
0 |
0 |
18 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
_register_unowned_container_state.get_owner |
11 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
_wire_symbolic_hooks |
9 |
1 |
0 |
6 |
3 |
73% |
| transformer_lens/model_bridge/component_setup.py |
setup_submodules |
57 |
11 |
0 |
38 |
4 |
80% |
| transformer_lens/model_bridge/component_setup.py |
_prune_hook_aliases_for_skipped |
28 |
8 |
0 |
20 |
2 |
67% |
| transformer_lens/model_bridge/component_setup.py |
_prune_hook_aliases_for_skipped._first_segment |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
_prune_hook_aliases_for_skipped._live_fallback |
19 |
1 |
0 |
12 |
1 |
94% |
| transformer_lens/model_bridge/component_setup.py |
setup_components |
15 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/component_setup.py |
setup_blocks_bridge |
15 |
1 |
0 |
6 |
1 |
90% |
| transformer_lens/model_bridge/component_setup.py |
(no function) |
26 |
0 |
2 |
0 |
0 |
100% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__torch_function__ |
5 |
0 |
0 |
2 |
1 |
86% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.shape |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.device |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.dtype |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__getitem__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__getattr__ |
5 |
2 |
0 |
0 |
0 |
60% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__gt__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__lt__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__ge__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__le__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__eq__ |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__ne__ |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/composition_scores.py |
CompositionScores.__repr__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/composition_scores.py |
(no function) |
21 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/driver_protocol.py |
TensorLike.shape |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/driver_protocol.py |
TensorLike.dtype |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/driver_protocol.py |
TensorLike.__array__ |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/driver_protocol.py |
Driver.forward |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/driver_protocol.py |
Driver.close |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/driver_protocol.py |
Driver.supports |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/driver_protocol.py |
to_torch |
11 |
0 |
0 |
4 |
1 |
93% |
| transformer_lens/model_bridge/driver_protocol.py |
validate_driver |
33 |
3 |
0 |
22 |
4 |
87% |
| transformer_lens/model_bridge/driver_protocol.py |
_expect_attr_type |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/driver_protocol.py |
(no function) |
32 |
0 |
24 |
0 |
0 |
100% |
| transformer_lens/model_bridge/exceptions.py |
StopAtLayerException.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/exceptions.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/__init__.py |
(no function) |
50 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/alibi_joint_qkv_attention.py |
ALiBiJointQKVAttentionBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/alibi_joint_qkv_attention.py |
ALiBiJointQKVAttentionBridge.forward |
14 |
4 |
0 |
6 |
2 |
60% |
| transformer_lens/model_bridge/generalized_components/alibi_joint_qkv_attention.py |
ALiBiJointQKVAttentionBridge._reconstruct_attention |
32 |
2 |
0 |
10 |
3 |
88% |
| transformer_lens/model_bridge/generalized_components/alibi_joint_qkv_attention.py |
ALiBiJointQKVAttentionBridge.get_random_inputs |
12 |
0 |
0 |
4 |
2 |
88% |
| transformer_lens/model_bridge/generalized_components/alibi_joint_qkv_attention.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/alibi_utils.py |
build_alibi_slopes |
10 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/alibi_utils.py |
build_alibi_tensor |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/alibi_utils.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/altup_block.py |
AltUpBlockBridge.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/altup_block.py |
AltUpBlockBridge.forward |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/altup_block.py |
AltUpBlockBridge._patch_active_stream |
9 |
4 |
0 |
4 |
2 |
54% |
| transformer_lens/model_bridge/generalized_components/altup_block.py |
AltUpBlockBridge._hook_input |
6 |
2 |
0 |
4 |
1 |
50% |
| transformer_lens/model_bridge/generalized_components/altup_block.py |
AltUpBlockBridge._hook_output |
6 |
1 |
0 |
4 |
2 |
70% |
| transformer_lens/model_bridge/generalized_components/altup_block.py |
AltUpBlockBridge._check_stop_at_layer |
11 |
9 |
0 |
8 |
1 |
16% |
| transformer_lens/model_bridge/generalized_components/altup_block.py |
(no function) |
16 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.__init__ |
33 |
1 |
0 |
10 |
2 |
93% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.set_original_component |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._apply_ln1_per_head |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._fork_and_norm_per_head |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.setup_hook_compatibility |
5 |
0 |
0 |
4 |
1 |
89% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.get_random_inputs |
31 |
10 |
0 |
22 |
5 |
64% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._setup_qkv_hook_reshaping |
53 |
6 |
0 |
34 |
5 |
85% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._setup_qkv_hook_reshaping.ReshapeForAttentionHeads.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._setup_qkv_hook_reshaping.ReshapeForAttentionHeads.handle_conversion |
5 |
0 |
0 |
4 |
1 |
89% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._setup_qkv_hook_reshaping.ReshapeForAttentionHeads.revert |
5 |
1 |
0 |
4 |
2 |
67% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._setup_qkv_hook_reshaping.conversion_dims |
7 |
1 |
0 |
4 |
1 |
82% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._setup_qkv_hook_reshaping.TransposeRotaryHeads.handle_conversion |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._setup_qkv_hook_reshaping.TransposeRotaryHeads.revert |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._update_kv_cache |
12 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._reshape_qkv_to_heads |
16 |
1 |
0 |
6 |
1 |
91% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._apply_attn_dropout |
7 |
0 |
0 |
6 |
1 |
92% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._apply_output_projection |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._softmax_dropout_pattern |
11 |
0 |
0 |
6 |
1 |
94% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._scrub_compatibility_pattern_nans |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._normalize_compatibility_mask_sentinel |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._reshape_attn_output |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._apply_reconstruct_attention_mask |
21 |
1 |
0 |
12 |
1 |
94% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._get_n_heads |
7 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._weight_layout_in_out |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._reshape_weight_to_3d |
11 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._check_head_split_width |
10 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._project_per_head_qkv |
14 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._compute_per_head_result |
13 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.forward |
29 |
2 |
0 |
18 |
3 |
89% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.W_Q |
19 |
5 |
0 |
12 |
4 |
65% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.W_K |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.W_V |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.W_O |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge._reshape_bias |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.b_Q |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.b_K |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.b_V |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
AttentionBridge.b_O |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/attention.py |
(no function) |
58 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/audio_feature_extractor.py |
AudioFeatureExtractorBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/audio_feature_extractor.py |
AudioFeatureExtractorBridge.forward |
8 |
2 |
0 |
4 |
2 |
67% |
| transformer_lens/model_bridge/generalized_components/audio_feature_extractor.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
align_offloaded_subtree |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
CloneOutputUnderGradMixin.forward |
4 |
0 |
0 |
2 |
1 |
83% |
| transformer_lens/model_bridge/generalized_components/base.py |
alias_generation |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
_bump_alias_generation |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.__init__ |
18 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.__call__ |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent._register_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent._register_aliases |
35 |
10 |
0 |
20 |
1 |
73% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.get_hooks |
8 |
0 |
0 |
8 |
2 |
88% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent._is_getattr_called_internally |
4 |
4 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.set_original_component |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.original_component |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.add_hook |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.remove_hooks |
8 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.set_processed_weights |
25 |
6 |
0 |
24 |
7 |
73% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.forward |
23 |
1 |
0 |
12 |
1 |
94% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.__getattr__ |
22 |
6 |
0 |
14 |
3 |
69% |
| transformer_lens/model_bridge/generalized_components/base.py |
GeneralizedComponent.__setattr__ |
23 |
2 |
0 |
14 |
0 |
95% |
| transformer_lens/model_bridge/generalized_components/base.py |
(no function) |
41 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge.__init__ |
19 |
1 |
0 |
8 |
1 |
93% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._wire_ln1_module |
10 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._maybe_wire_capture_hooks |
21 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._maybe_wire_capture_hooks._capture_pre_ln1 |
2 |
0 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._maybe_wire_capture_hooks._capture_mlp_in |
6 |
1 |
0 |
4 |
1 |
80% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._teardown_capture_hooks |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._read_use_hook_mlp_in |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._clear_attention_capture |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge.forward |
13 |
1 |
0 |
2 |
1 |
87% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._apply_output_hook |
14 |
0 |
0 |
10 |
1 |
96% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._is_standalone_hidden_state_call |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._extract_layer_idx |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._check_stop_at_layer |
11 |
3 |
0 |
8 |
1 |
68% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._maybe_inject_start_residual |
12 |
3 |
0 |
10 |
2 |
68% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._hook_input_hidden_states |
6 |
2 |
0 |
4 |
1 |
50% |
| transformer_lens/model_bridge/generalized_components/block.py |
BlockBridge._filter_kwargs_for_forward |
14 |
3 |
0 |
4 |
1 |
78% |
| transformer_lens/model_bridge/generalized_components/block.py |
MLABlockBridge.__init__ |
5 |
0 |
0 |
4 |
1 |
89% |
| transformer_lens/model_bridge/generalized_components/block.py |
ParallelBlockBridge.__init__ |
4 |
0 |
0 |
2 |
1 |
83% |
| transformer_lens/model_bridge/generalized_components/block.py |
ScaledResidualBlockBridge.__init__ |
14 |
0 |
0 |
8 |
1 |
95% |
| transformer_lens/model_bridge/generalized_components/block.py |
ScaledResidualBlockBridge.set_original_component |
13 |
0 |
0 |
8 |
1 |
95% |
| transformer_lens/model_bridge/generalized_components/block.py |
ScaledResidualBlockBridge._maybe_wire_capture_hooks |
13 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/block.py |
ScaledResidualBlockBridge._make_scaled_contribution_hook |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/block.py |
ScaledResidualBlockBridge._make_scaled_contribution_hook._hook |
12 |
1 |
0 |
6 |
1 |
89% |
| transformer_lens/model_bridge/generalized_components/block.py |
DelegatedAttentionBlockBridge.__init__ |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/block.py |
(no function) |
44 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/bloom_attention.py |
BloomAttentionBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/bloom_attention.py |
BloomAttentionBridge.forward |
24 |
4 |
0 |
12 |
3 |
75% |
| transformer_lens/model_bridge/generalized_components/bloom_attention.py |
BloomAttentionBridge._reconstruct_attention |
42 |
1 |
0 |
14 |
3 |
93% |
| transformer_lens/model_bridge/generalized_components/bloom_attention.py |
BloomAttentionBridge.set_processed_weights |
30 |
1 |
0 |
4 |
2 |
91% |
| transformer_lens/model_bridge/generalized_components/bloom_attention.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/bloom_block.py |
BloomBlockBridge.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/bloom_block.py |
BloomBlockBridge.build_alibi_tensor |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/bloom_block.py |
BloomBlockBridge.forward |
29 |
23 |
0 |
16 |
2 |
18% |
| transformer_lens/model_bridge/generalized_components/bloom_block.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/bloom_mlp.py |
BloomMLPBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/bloom_mlp.py |
BloomMLPBridge.set_original_component |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/bloom_mlp.py |
BloomMLPBridge.forward |
10 |
3 |
0 |
6 |
2 |
56% |
| transformer_lens/model_bridge/generalized_components/bloom_mlp.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/clip_vision_encoder.py |
CLIPVisionEncoderLayerBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/clip_vision_encoder.py |
CLIPVisionEncoderLayerBridge.forward |
8 |
8 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/clip_vision_encoder.py |
CLIPVisionEncoderBridge.__init__ |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/clip_vision_encoder.py |
CLIPVisionEncoderBridge.forward |
10 |
10 |
0 |
6 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/clip_vision_encoder.py |
CLIPVisionEncoderBridge.set_original_component |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/clip_vision_encoder.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/codegen_attention.py |
_rotate_every_two |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/codegen_attention.py |
_apply_rotary_pos_emb |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/codegen_attention.py |
CodeGenAttentionBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/codegen_attention.py |
CodeGenAttentionBridge.get_random_inputs |
11 |
11 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/codegen_attention.py |
CodeGenAttentionBridge.set_original_component |
3 |
0 |
0 |
2 |
1 |
80% |
| transformer_lens/model_bridge/generalized_components/codegen_attention.py |
CodeGenAttentionBridge.forward |
15 |
2 |
0 |
6 |
2 |
81% |
| transformer_lens/model_bridge/generalized_components/codegen_attention.py |
CodeGenAttentionBridge._reconstruct_attention |
35 |
3 |
0 |
6 |
3 |
85% |
| transformer_lens/model_bridge/generalized_components/codegen_attention.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/conv1d.py |
Conv1DBridge.forward |
9 |
9 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/conv1d.py |
Conv1DBridge.__repr__ |
8 |
8 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/conv1d.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/conv_pos_embed.py |
ConvPosEmbedBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/conv_pos_embed.py |
ConvPosEmbedBridge.forward |
8 |
2 |
0 |
4 |
2 |
67% |
| transformer_lens/model_bridge/generalized_components/conv_pos_embed.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/depthwise_conv1d.py |
DepthwiseConv1DBridge.forward |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/depthwise_conv1d.py |
DepthwiseConv1DBridge.get_random_inputs |
7 |
7 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/depthwise_conv1d.py |
DepthwiseConv1DBridge.__repr__ |
10 |
10 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/depthwise_conv1d.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/embedding.py |
EmbeddingBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/embedding.py |
EmbeddingBridge.W_E |
13 |
6 |
0 |
8 |
3 |
48% |
| transformer_lens/model_bridge/generalized_components/embedding.py |
EmbeddingBridge.forward |
20 |
6 |
0 |
8 |
4 |
64% |
| transformer_lens/model_bridge/generalized_components/embedding.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_delta_net.py |
GatedDeltaNetBridge.__init__ |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_delta_net.py |
GatedDeltaNetBridge.forward |
5 |
1 |
0 |
4 |
1 |
78% |
| transformer_lens/model_bridge/generalized_components/gated_delta_net.py |
GatedDeltaNetBridge._native_forward |
14 |
7 |
0 |
10 |
3 |
42% |
| transformer_lens/model_bridge/generalized_components/gated_delta_net.py |
GatedDeltaNetBridge._hooked_forward |
58 |
3 |
0 |
14 |
3 |
92% |
| transformer_lens/model_bridge/generalized_components/gated_delta_net.py |
GatedDeltaNetBridge._l2norm |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_delta_net.py |
GatedDeltaNetBridge._gated_delta_scan |
27 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_delta_net.py |
GatedDeltaNetBridge.compute_ssm_state |
23 |
3 |
0 |
8 |
1 |
87% |
| transformer_lens/model_bridge/generalized_components/gated_delta_net.py |
GatedDeltaNetBridge.compute_effective_attention |
27 |
3 |
0 |
6 |
1 |
88% |
| transformer_lens/model_bridge/generalized_components/gated_delta_net.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
_resolve_activation_fn_name |
7 |
1 |
0 |
6 |
1 |
85% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
identity_rule_supports_activation |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
resolve_activation_fn |
14 |
0 |
0 |
10 |
2 |
92% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
resolve_activation_fn.gelu_tanh |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
resolve_activation_fn.relu_squared |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
_IdentityRuleActivation.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
_IdentityRuleActivation.forward |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._is_gated_mlp_shaped |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._find_activation_attr |
7 |
1 |
0 |
6 |
1 |
85% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._activation_rule_installable |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._relevance_rule_kinds |
6 |
1 |
0 |
4 |
1 |
80% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._relevance_rule_unsupported_kinds |
5 |
2 |
0 |
4 |
2 |
56% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._install_activation_rule |
11 |
2 |
0 |
4 |
2 |
73% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._teardown_activation_rule |
10 |
2 |
0 |
6 |
3 |
69% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._install_gate_rule |
6 |
1 |
0 |
2 |
2 |
62% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._install_gate_rule._scale_product_gradient |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._teardown_gate_rule |
3 |
0 |
0 |
2 |
1 |
80% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._enable_relevance_rule |
6 |
0 |
0 |
4 |
1 |
90% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge._disable_relevance_rule |
6 |
0 |
0 |
4 |
1 |
90% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge.forward |
31 |
1 |
0 |
12 |
5 |
86% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
GatedMLPBridge.set_processed_weights |
45 |
13 |
0 |
24 |
7 |
62% |
| transformer_lens/model_bridge/generalized_components/gated_mlp.py |
(no function) |
31 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_rms_norm.py |
GatedRMSNormBridge.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/gated_rms_norm.py |
GatedRMSNormBridge.forward |
7 |
1 |
0 |
4 |
2 |
73% |
| transformer_lens/model_bridge/generalized_components/gated_rms_norm.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/glm_moe_dsa_attention.py |
_apply_rotary_pos_emb_single |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/glm_moe_dsa_attention.py |
_repeat_kv |
5 |
2 |
0 |
2 |
1 |
57% |
| transformer_lens/model_bridge/generalized_components/glm_moe_dsa_attention.py |
GlmMoeDsaAttentionBridge.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/glm_moe_dsa_attention.py |
GlmMoeDsaAttentionBridge.forward |
103 |
24 |
0 |
32 |
10 |
69% |
| transformer_lens/model_bridge/generalized_components/glm_moe_dsa_attention.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge.__init__ |
15 |
1 |
0 |
2 |
1 |
88% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge._filter_gate_up_state_dict |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge._restore_filtered_gate_up_state_dict |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge._default_split_gate_up |
19 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge.set_original_component |
10 |
0 |
0 |
4 |
1 |
93% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge._resolve_activation_fn |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge._activation_rule_installable |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge._install_activation_rule |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge._teardown_activation_rule |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge._install_gate_rule |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge._teardown_gate_rule |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
JointGateUpMLPBridge.forward |
12 |
1 |
0 |
4 |
1 |
88% |
| transformer_lens/model_bridge/generalized_components/joint_gate_up_mlp.py |
(no function) |
24 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge.__init__ |
25 |
0 |
0 |
8 |
1 |
97% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge.__deepcopy__ |
16 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._resolve_cloned_config |
7 |
1 |
0 |
4 |
1 |
82% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._filter_qkv_state_dict |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._restore_filtered_qkv_state_dict |
6 |
1 |
0 |
4 |
1 |
80% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._create_qkv_conversion_rule |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._create_qkv_conversion_rule.ConditionalRearrangeConversion.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._create_qkv_conversion_rule.ConditionalRearrangeConversion.handle_conversion |
5 |
2 |
0 |
4 |
2 |
56% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._create_qkv_conversion_rule.ConditionalRearrangeConversion.revert |
5 |
2 |
0 |
4 |
2 |
56% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._default_split_qkv_matrix |
33 |
4 |
0 |
14 |
7 |
77% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge.set_original_component |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge.forward |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._is_split_qkv_fork_active |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._split_forward_qkv |
22 |
2 |
0 |
4 |
1 |
88% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._process_output |
19 |
2 |
0 |
14 |
6 |
76% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._apply_attention_input_hook |
9 |
2 |
0 |
6 |
2 |
73% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
JointQKVAttentionBridge._reconstruct_attention |
33 |
3 |
0 |
8 |
2 |
88% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_attention.py |
(no function) |
28 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_position_embeddings_attention.py |
JointQKVPositionEmbeddingsAttentionBridge.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_position_embeddings_attention.py |
JointQKVPositionEmbeddingsAttentionBridge.get_random_inputs |
21 |
21 |
0 |
6 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_position_embeddings_attention.py |
JointQKVPositionEmbeddingsAttentionBridge._apply_rotary_pos_emb |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_position_embeddings_attention.py |
JointQKVPositionEmbeddingsAttentionBridge._apply_rotary_pos_emb.rotate_half |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_position_embeddings_attention.py |
JointQKVPositionEmbeddingsAttentionBridge._reconstruct_attention |
28 |
3 |
0 |
6 |
2 |
85% |
| transformer_lens/model_bridge/generalized_components/joint_qkv_position_embeddings_attention.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/lfm2_gated_short_conv.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/linear.py |
LinearBridge.forward |
9 |
1 |
0 |
4 |
1 |
85% |
| transformer_lens/model_bridge/generalized_components/linear.py |
LinearBridge.__repr__ |
9 |
3 |
0 |
2 |
1 |
64% |
| transformer_lens/model_bridge/generalized_components/linear.py |
LinearBridge.set_processed_weights |
24 |
11 |
0 |
18 |
5 |
52% |
| transformer_lens/model_bridge/generalized_components/linear.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mla_attention.py |
_rotate_half |
3 |
3 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/mla_attention.py |
_apply_rotary_pos_emb |
5 |
5 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/mla_attention.py |
_apply_rotary_pos_emb_interleave |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mla_attention.py |
_apply_rotary_complex |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mla_attention.py |
MLAAttentionBridge.__init__ |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mla_attention.py |
MLAAttentionBridge.forward |
100 |
19 |
0 |
38 |
4 |
76% |
| transformer_lens/model_bridge/generalized_components/mla_attention.py |
MLAAttentionBridge.get_random_inputs |
30 |
30 |
0 |
14 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/mla_attention.py |
MLAAttentionBridge.__getattr__ |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mla_attention.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mlp.py |
weight_layout_in_out |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mlp.py |
normalize_mlp_weight |
12 |
3 |
0 |
8 |
1 |
70% |
| transformer_lens/model_bridge/generalized_components/mlp.py |
MLPBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mlp.py |
MLPBridge.forward |
28 |
3 |
0 |
18 |
3 |
87% |
| transformer_lens/model_bridge/generalized_components/mlp.py |
MLPBridge._weight_layout_in_out |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mlp.py |
MLPBridge._normalize_mlp_weight |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mlp.py |
MLPBridge.W_in |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/mlp.py |
MLPBridge.W_gate |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mlp.py |
MLPBridge.W_out |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/mlp.py |
(no function) |
20 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
unwrap_bridge |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
has_batched_experts |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
_input_axis |
31 |
5 |
0 |
26 |
5 |
82% |
| transformer_lens/model_bridge/generalized_components/moe.py |
fold_scale_into_moe_block |
21 |
1 |
0 |
8 |
1 |
93% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge.__init__ |
10 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge._build_router_hook_aliases |
10 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge._binds_dense_projections |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge._binds_dense_gate |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge.set_original_component |
21 |
0 |
0 |
12 |
3 |
91% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge.validate_after_setup |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge.bound_dense |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge._dense_projection |
6 |
1 |
0 |
4 |
1 |
80% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge.W_gate |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge.W_in |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge.W_out |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge.get_random_inputs |
6 |
2 |
0 |
4 |
2 |
60% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoEBridge.forward |
25 |
1 |
0 |
18 |
3 |
91% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoERouterBridge.__init__ |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoERouterBridge.forward |
30 |
3 |
0 |
16 |
7 |
78% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoERouterBridge._expand_expert_weights |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoERouterBridge._collapse_expert_weights |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoERouterBridge._is_top_k_shaped |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/moe.py |
MoERouterBridge.set_processed_weights |
17 |
2 |
0 |
12 |
2 |
86% |
| transformer_lens/model_bridge/generalized_components/moe.py |
(no function) |
50 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mpt_alibi_attention.py |
_build_mpt_alibi_tensor |
10 |
1 |
0 |
2 |
1 |
83% |
| transformer_lens/model_bridge/generalized_components/mpt_alibi_attention.py |
MPTALiBiAttentionBridge.forward |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/generalized_components/mpt_alibi_attention.py |
MPTALiBiAttentionBridge.set_original_component |
7 |
0 |
0 |
2 |
1 |
89% |
| transformer_lens/model_bridge/generalized_components/mpt_alibi_attention.py |
MPTALiBiAttentionBridge._reconstruct_attention |
21 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/mpt_alibi_attention.py |
MPTALiBiAttentionBridge.get_random_inputs |
10 |
10 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/mpt_alibi_attention.py |
(no function) |
20 |
2 |
0 |
0 |
0 |
90% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
_NativeLNRuleForward.forward |
8 |
1 |
0 |
2 |
1 |
80% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
_NativeLNRuleForward.backward |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
NormalizationBridge.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
NormalizationBridge._relevance_rule_kinds |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
NormalizationBridge._enable_relevance_rule |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
NormalizationBridge._disable_relevance_rule |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
NormalizationBridge.uses_rms_norm |
9 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
NormalizationBridge.forward |
11 |
1 |
0 |
6 |
1 |
88% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
NormalizationBridge._python_norm_forward |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
NormalizationBridge._apply_weight_and_bias |
9 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
NormalizationBridge._hf_autograd_forward_with_hooks |
47 |
4 |
0 |
26 |
3 |
90% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
NormalizationBridge._native_forward_with_ln_rule |
22 |
1 |
0 |
6 |
1 |
93% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
LayerNormPreBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
LayerNormPreBridge._apply_weight_and_bias |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
RMSNormPreBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
RMSNormPreBridge._apply_weight_and_bias |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/normalization.py |
(no function) |
38 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/opaque_block.py |
OpaqueBlockBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/opaque_block.py |
OpaqueBlockBridge.forward |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/opaque_block.py |
OpaqueBlockBridge._apply_output_hook |
9 |
6 |
0 |
6 |
2 |
33% |
| transformer_lens/model_bridge/generalized_components/opaque_block.py |
OpaqueBlockBridge._hook_input_hidden_states |
6 |
2 |
0 |
4 |
1 |
50% |
| transformer_lens/model_bridge/generalized_components/opaque_block.py |
OpaqueBlockBridge._check_stop_at_layer |
17 |
5 |
0 |
12 |
3 |
66% |
| transformer_lens/model_bridge/generalized_components/opaque_block.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/pooler.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/pos_embed.py |
PosEmbedBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/pos_embed.py |
PosEmbedBridge.W_pos |
7 |
1 |
0 |
4 |
1 |
82% |
| transformer_lens/model_bridge/generalized_components/pos_embed.py |
PosEmbedBridge.forward |
13 |
1 |
0 |
8 |
1 |
90% |
| transformer_lens/model_bridge/generalized_components/pos_embed.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embedding_hooks_mixin.py |
PositionEmbeddingHooksMixin._init_position_embedding_hooks |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embedding_hooks_mixin.py |
PositionEmbeddingHooksMixin.set_rotary_emb |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embedding_hooks_mixin.py |
PositionEmbeddingHooksMixin._apply_position_embedding_hooks |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embedding_hooks_mixin.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
_apply_rotary_pos_emb_adjacent_pairs |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
_apply_rotary_pos_emb_adjacent_pairs._rotate |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
_setup_eager_attention_hook_wrapper |
11 |
2 |
0 |
2 |
0 |
85% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
_setup_eager_attention_hook_wrapper.hooked_eager_attention_forward |
8 |
4 |
0 |
6 |
1 |
36% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
PositionEmbeddingsAttentionBridge.__init__ |
11 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
PositionEmbeddingsAttentionBridge.set_original_component |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
PositionEmbeddingsAttentionBridge._own_scaled_hook_k |
6 |
0 |
0 |
4 |
1 |
90% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
PositionEmbeddingsAttentionBridge._fire_scaled_hook_k |
7 |
4 |
0 |
4 |
2 |
45% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
PositionEmbeddingsAttentionBridge._validate_submodule_declarations |
7 |
2 |
0 |
6 |
2 |
69% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
PositionEmbeddingsAttentionBridge._decide_qk_norm_phase |
23 |
4 |
0 |
16 |
2 |
79% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
PositionEmbeddingsAttentionBridge._apply_pre_reshape_qk_norm |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
PositionEmbeddingsAttentionBridge.forward |
169 |
8 |
0 |
72 |
11 |
92% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
PositionEmbeddingsAttentionBridge._pre_output_projection |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
PositionEmbeddingsAttentionBridge.get_random_inputs |
23 |
23 |
0 |
6 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/position_embeddings_attention.py |
(no function) |
30 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/qwen3_5_vision_encoder.py |
Qwen3_5VisionBlockBridge.__init__ |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/qwen3_5_vision_encoder.py |
Qwen3_5VisionEncoderBridge.__init__ |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/qwen3_5_vision_encoder.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/rms_normalization.py |
RMSNormalizationBridge.__init__ |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/generalized_components/rms_normalization.py |
(no function) |
6 |
0 |
2 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/rotary_embedding.py |
RotaryEmbeddingBridge.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/rotary_embedding.py |
RotaryEmbeddingBridge.get_random_inputs |
13 |
4 |
0 |
6 |
3 |
63% |
| transformer_lens/model_bridge/generalized_components/rotary_embedding.py |
RotaryEmbeddingBridge.forward |
18 |
5 |
0 |
12 |
4 |
63% |
| transformer_lens/model_bridge/generalized_components/rotary_embedding.py |
RotaryEmbeddingBridge.get_dummy_inputs |
5 |
5 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/rotary_embedding.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/siglip_vision_encoder.py |
_vision_attention_config |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/siglip_vision_encoder.py |
SiglipVisionEncoderLayerBridge.__init__ |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/generalized_components/siglip_vision_encoder.py |
SiglipVisionEncoderLayerBridge.forward |
8 |
2 |
0 |
4 |
2 |
67% |
| transformer_lens/model_bridge/generalized_components/siglip_vision_encoder.py |
SiglipVisionEncoderBridge.__init__ |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/siglip_vision_encoder.py |
SiglipVisionEncoderBridge.forward |
10 |
3 |
0 |
6 |
3 |
62% |
| transformer_lens/model_bridge/generalized_components/siglip_vision_encoder.py |
SiglipVisionEncoderBridge.set_original_component |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/siglip_vision_encoder.py |
(no function) |
20 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm2_mixer.py |
SSM2MixerBridge.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm2_mixer.py |
SSM2MixerBridge.forward |
20 |
3 |
0 |
14 |
4 |
79% |
| transformer_lens/model_bridge/generalized_components/ssm2_mixer.py |
SSM2MixerBridge._eager_scan_forward |
34 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm2_mixer.py |
SSM2MixerBridge._eager_scan_forward._mask_pad |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm2_mixer.py |
SSM2MixerBridge.compute_effective_attention |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm2_mixer.py |
SSM2MixerBridge.compute_ssm_state |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm2_mixer.py |
SSM2MixerBridge._ssd_terms |
32 |
1 |
0 |
4 |
1 |
94% |
| transformer_lens/model_bridge/generalized_components/ssm2_mixer.py |
SSM2MixerBridge._ssd_terms._mamba_dim |
5 |
1 |
0 |
4 |
2 |
67% |
| transformer_lens/model_bridge/generalized_components/ssm2_mixer.py |
(no function) |
20 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_block.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
_weightless_rms |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
SSMMixerBridge.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
SSMMixerBridge.forward |
20 |
6 |
0 |
14 |
4 |
65% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
SSMMixerBridge._eager_scan_forward |
36 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
SSMMixerBridge._eager_scan_forward._mask_cf |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
SSMMixerBridge.compute_effective_attention |
9 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
SSMMixerBridge.compute_ssm_state |
11 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
SSMMixerBridge._apply_selective_param_norms |
10 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
SSMMixerBridge._s6_terms |
33 |
1 |
0 |
10 |
1 |
95% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
SSMMixerBridge._s6_terms._mamba_dim |
5 |
1 |
0 |
4 |
2 |
67% |
| transformer_lens/model_bridge/generalized_components/ssm_mixer.py |
(no function) |
22 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_protocol.py |
SSMMixerProtocol.compute_effective_attention |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_protocol.py |
SSMStateHookMixin.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_protocol.py |
_is_realized_ssm_mixer |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_protocol.py |
find_ssm_mixer |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/ssm_protocol.py |
(no function) |
14 |
0 |
6 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/symbolic.py |
SymbolicBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/symbolic.py |
SymbolicBridge.forward |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/symbolic.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5_block.py |
_clamp_fp16_inf |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5_block.py |
T5BlockBridge.__init__ |
18 |
0 |
0 |
10 |
2 |
93% |
| transformer_lens/model_bridge/generalized_components/t5_block.py |
T5BlockBridge.set_original_component |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5_block.py |
T5BlockBridge._patch_t5_block_forward |
5 |
1 |
0 |
2 |
1 |
71% |
| transformer_lens/model_bridge/generalized_components/t5_block.py |
T5BlockBridge._patch_t5_block_forward.patched_forward |
51 |
14 |
0 |
26 |
9 |
65% |
| transformer_lens/model_bridge/generalized_components/t5_block.py |
T5BlockBridge.forward |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/generalized_components/t5_block.py |
T5BlockBridge.get_expected_parameter_names |
5 |
5 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/t5_block.py |
T5BlockBridge.get_list_size |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/t5_block.py |
(no function) |
16 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_decoder_block.py |
T5Gemma2DecoderBlockBridge.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_decoder_block.py |
T5Gemma2DecoderBlockBridge.set_original_component |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_decoder_block.py |
T5Gemma2DecoderBlockBridge._patch_decoder_layer_forward |
8 |
1 |
0 |
2 |
1 |
80% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_decoder_block.py |
T5Gemma2DecoderBlockBridge._patch_decoder_layer_forward.patched_forward |
14 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_decoder_block.py |
T5Gemma2DecoderBlockBridge.forward |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_decoder_block.py |
T5Gemma2DecoderBlockBridge.get_expected_parameter_names |
5 |
5 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_decoder_block.py |
T5Gemma2DecoderBlockBridge.get_list_size |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_decoder_block.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_merged_attention.py |
T5Gemma2MergedAttentionBridge.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_merged_attention.py |
T5Gemma2MergedAttentionBridge.get_random_inputs |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_merged_attention.py |
T5Gemma2MergedAttentionBridge.forward |
7 |
0 |
0 |
4 |
2 |
82% |
| transformer_lens/model_bridge/generalized_components/t5gemma2_merged_attention.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5gemma_decoder_block.py |
T5GemmaDecoderBlockBridge.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/t5gemma_decoder_block.py |
T5GemmaDecoderBlockBridge.set_original_component |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/t5gemma_decoder_block.py |
T5GemmaDecoderBlockBridge._patch_decoder_layer_forward |
10 |
10 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/t5gemma_decoder_block.py |
T5GemmaDecoderBlockBridge._patch_decoder_layer_forward.patched_forward |
20 |
20 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/t5gemma_decoder_block.py |
T5GemmaDecoderBlockBridge.forward |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/t5gemma_decoder_block.py |
T5GemmaDecoderBlockBridge.get_expected_parameter_names |
5 |
5 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/t5gemma_decoder_block.py |
T5GemmaDecoderBlockBridge.get_list_size |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/generalized_components/t5gemma_decoder_block.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/unembedding.py |
UnembeddingBridge.__init__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/unembedding.py |
UnembeddingBridge.set_original_component |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/unembedding.py |
UnembeddingBridge.W_U |
10 |
4 |
0 |
6 |
2 |
50% |
| transformer_lens/model_bridge/generalized_components/unembedding.py |
UnembeddingBridge.forward |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/unembedding.py |
UnembeddingBridge.b_U |
18 |
12 |
0 |
8 |
3 |
35% |
| transformer_lens/model_bridge/generalized_components/unembedding.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/vision_classifier_head.py |
VisionClassifierHeadBridge.forward |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/generalized_components/vision_classifier_head.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/vision_embeddings.py |
VisionEmbeddingsBridge.forward |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/vision_embeddings.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/vision_projection.py |
VisionProjectionBridge.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/generalized_components/vision_projection.py |
VisionProjectionBridge.forward |
8 |
2 |
0 |
4 |
2 |
67% |
| transformer_lens/model_bridge/generalized_components/vision_projection.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/get_params_util.py |
_tensor_attr |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/get_params_util.py |
get_bridge_params |
99 |
5 |
0 |
48 |
2 |
95% |
| transformer_lens/model_bridge/get_params_util.py |
get_bridge_params._get_device_dtype |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/get_params_util.py |
get_bridge_params._zeros |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/get_params_util.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge.__init__ |
10 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge.boot_vllm |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge.boot_inspect |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge._scan_existing_hooks |
0 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge.forward |
23 |
2 |
0 |
18 |
4 |
85% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge.run_with_hooks |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge._reject_stop_at_layer |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge._reject_start_at_layer |
2 |
1 |
0 |
2 |
1 |
50% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge.run_with_cache |
5 |
1 |
0 |
2 |
1 |
71% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge.to_tokens |
16 |
5 |
0 |
10 |
4 |
58% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge.__enter__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge.__exit__ |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/remote_bridge.py |
RemoteBridge.__del__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/remote_bridge.py |
(no function) |
24 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/__init__.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/_bridge_builder.py |
build_bridge_config_from_hf |
43 |
0 |
0 |
28 |
3 |
96% |
| transformer_lens/model_bridge/sources/_bridge_builder.py |
detect_tokenizer_bos_eos |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/_bridge_builder.py |
skip_tokenizer_for_modality |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/_bridge_builder.py |
configure_tokenizer |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/_bridge_builder.py |
build_bridge_from_module |
31 |
4 |
0 |
16 |
0 |
91% |
| transformer_lens/model_bridge/sources/_bridge_builder.py |
(no function) |
17 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/_driver_base.py |
DriverBase.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/_driver_base.py |
DriverBase.forward |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/_driver_base.py |
DriverBase.close |
0 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/_driver_base.py |
DriverBase.supports |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/_driver_base.py |
(no function) |
14 |
0 |
12 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/_hf_format.py |
get_effective_text_config |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/_hf_format.py |
map_default_transformer_lens_config |
184 |
23 |
0 |
140 |
17 |
87% |
| transformer_lens/model_bridge/sources/_hf_format.py |
map_default_transformer_lens_config.legacy_per_layer |
8 |
1 |
0 |
4 |
1 |
83% |
| transformer_lens/model_bridge/sources/_hf_format.py |
determine_architecture_from_hf_config |
14 |
1 |
0 |
12 |
1 |
92% |
| transformer_lens/model_bridge/sources/_hf_format.py |
setup_tokenizer |
21 |
3 |
0 |
16 |
3 |
84% |
| transformer_lens/model_bridge/sources/_hf_format.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/__init__.py |
__getattr__ |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/sources/inspect/__init__.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_message_text |
5 |
2 |
0 |
2 |
1 |
57% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_tool_schema |
3 |
3 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_parse_tool_calls |
16 |
3 |
0 |
6 |
1 |
82% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_warn_unsupported_config |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_require_served |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_require_interveneable |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_InspectModelAPIBase._generate_capture |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_InspectModelAPIBase._generate_eval |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_InspectModelAPIBase.supported_kinds |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_InspectModelAPIBase.capability_note |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_InspectModelAPIBase.generate |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_InspectModelAPIBase._parse_eval_capture |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_InspectModelAPIBase._messages_to_ids |
13 |
4 |
0 |
8 |
1 |
67% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
_InspectModelAPIBase._logprob_entry |
6 |
0 |
0 |
2 |
1 |
88% |
| transformer_lens/model_bridge/sources/inspect/_provider_base.py |
(no function) |
34 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/conftest.py |
(no function) |
3 |
3 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver.__init__ |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver.forward |
15 |
1 |
0 |
6 |
1 |
90% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver._generate |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver._assemble_captures |
16 |
1 |
0 |
6 |
1 |
91% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver._warn_missing |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver.close |
13 |
2 |
0 |
4 |
1 |
82% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver._ensure_loop |
9 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver._abandon_loop |
8 |
2 |
0 |
2 |
1 |
70% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver._run_coro |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver._wire_keys |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
InspectDriver._normalize_input_ids |
8 |
0 |
0 |
6 |
1 |
93% |
| transformer_lens/model_bridge/sources/inspect/driver.py |
(no function) |
28 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/eval.py |
_default_reduce |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/eval.py |
capture_activations |
11 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/eval.py |
capture_activations.solve |
13 |
1 |
0 |
2 |
1 |
87% |
| transformer_lens/model_bridge/sources/inspect/eval.py |
turn_activations |
16 |
1 |
0 |
10 |
2 |
88% |
| transformer_lens/model_bridge/sources/inspect/eval.py |
activations_column |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/eval.py |
(no function) |
14 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/hooks.py |
supported_hook_points |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/hooks.py |
all_hook_points |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/hooks.py |
nonfireable_hook_points |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/hooks.py |
resolve |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/hooks.py |
wire_key |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/hooks.py |
name_from_wire_key |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/hooks.py |
(no function) |
16 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/intervention.py |
build_interventions |
24 |
4 |
0 |
18 |
4 |
81% |
| transformer_lens/model_bridge/sources/inspect/intervention.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
TLBridgeProfile.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
TLBridgeProfile.supported_hooks |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
TLBridgeProfile.translate_interventions |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
TLBridgeProfile.build_request |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
TLBridgeProfile.decode_logits |
5 |
2 |
0 |
2 |
1 |
57% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
VLLMLensProfile.supported_hooks |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
VLLMLensProfile.translate_interventions |
18 |
10 |
0 |
12 |
3 |
43% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
VLLMLensProfile.build_request |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
VLLMLensProfile.decode_logits |
6 |
0 |
0 |
2 |
1 |
88% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
for_provider |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/profiles.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/source.py |
boot_inspect |
35 |
1 |
0 |
14 |
3 |
92% |
| transformer_lens/model_bridge/sources/inspect/source.py |
_provider_model_kwargs |
5 |
0 |
0 |
2 |
1 |
86% |
| transformer_lens/model_bridge/sources/inspect/source.py |
_log_hook_summary |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/source.py |
(no function) |
17 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
transformer_lens_provider |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
TransformerLensTransformersModelAPI.__init__ |
21 |
1 |
0 |
4 |
1 |
92% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
TransformerLensTransformersModelAPI._generate_capture |
38 |
0 |
0 |
14 |
1 |
98% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
TransformerLensTransformersModelAPI._generate_eval |
47 |
13 |
0 |
22 |
6 |
64% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
TransformerLensTransformersModelAPI._eval_capture_scope |
16 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
TransformerLensTransformersModelAPI._install_hooks |
20 |
0 |
0 |
14 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
TransformerLensTransformersModelAPI._install_head_hooks |
21 |
5 |
0 |
16 |
4 |
70% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_require_layer_modules |
7 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_plan |
12 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_assemble |
11 |
0 |
0 |
8 |
2 |
89% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_pre_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_pre_hook.hook |
13 |
1 |
0 |
10 |
2 |
87% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_out_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_out_hook.hook |
12 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_proj_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_proj_hook.hook |
9 |
1 |
0 |
6 |
2 |
80% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_zin_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_zin_hook.hook |
11 |
3 |
0 |
8 |
4 |
63% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_affine_op |
9 |
2 |
0 |
6 |
2 |
73% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_apply_affine |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_detect_capabilities |
19 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_detect_head_capabilities |
22 |
1 |
0 |
8 |
1 |
93% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_out_width |
5 |
2 |
0 |
2 |
1 |
57% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_in_width |
6 |
1 |
0 |
4 |
1 |
80% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_attn_geometry |
9 |
1 |
0 |
4 |
1 |
85% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_resid_mid_derivable |
37 |
6 |
0 |
14 |
4 |
80% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_resid_mid_derivable.grab |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_resid_mid_derivable.grab.hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_resid_mid_derivable.grab_in |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_resid_mid_derivable.grab_in.hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_resid_mid_derivable.perturb_attn |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_locate_layers |
9 |
1 |
0 |
8 |
1 |
88% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_locate_mlp |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_projection_host |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
_first_attr |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/transformers_provider.py |
(no function) |
51 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/vllm_provider.py |
_kinds_from_specs |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/vllm_provider.py |
transformer_lens_vllm_provider |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/inspect/vllm_provider.py |
TransformerLensVLLMModelAPI.__init__ |
38 |
4 |
0 |
4 |
0 |
90% |
| transformer_lens/model_bridge/sources/inspect/vllm_provider.py |
TransformerLensVLLMModelAPI._generate_capture |
44 |
1 |
0 |
20 |
2 |
95% |
| transformer_lens/model_bridge/sources/inspect/vllm_provider.py |
TransformerLensVLLMModelAPI._generate_eval |
53 |
4 |
0 |
30 |
6 |
88% |
| transformer_lens/model_bridge/sources/inspect/vllm_provider.py |
TransformerLensVLLMModelAPI._logprob_from_dict |
9 |
0 |
0 |
4 |
1 |
92% |
| transformer_lens/model_bridge/sources/inspect/vllm_provider.py |
TransformerLensVLLMModelAPI.close |
13 |
13 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/sources/inspect/vllm_provider.py |
_synthesize_logits |
10 |
3 |
0 |
8 |
2 |
61% |
| transformer_lens/model_bridge/sources/inspect/vllm_provider.py |
(no function) |
22 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/wire.py |
encode_array |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/wire.py |
decode_array |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/sources/inspect/wire.py |
encode_activations |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/inspect/wire.py |
decode_activations |
13 |
0 |
0 |
8 |
1 |
95% |
| transformer_lens/model_bridge/sources/inspect/wire.py |
_safe_int |
4 |
2 |
0 |
0 |
0 |
50% |
| transformer_lens/model_bridge/sources/inspect/wire.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/__init__.py |
boot |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/__init__.py |
boot |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/__init__.py |
boot |
20 |
0 |
0 |
14 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/__init__.py |
(no function) |
12 |
0 |
20 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/init.py |
_unwrap_component |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/init.py |
initialize_native_model |
34 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/init.py |
initialize_native_model._staged |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/init.py |
initialize_native_model._staged.apply |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/init.py |
_init_norm |
11 |
1 |
0 |
8 |
1 |
89% |
| transformer_lens/model_bridge/sources/native/init.py |
_init_block |
9 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/init.py |
_init_attention |
9 |
0 |
0 |
6 |
2 |
87% |
| transformer_lens/model_bridge/sources/native/init.py |
_init_mlp |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/init.py |
_init_gated_mlp |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/init.py |
(no function) |
17 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
_normalization_type |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
_uses_rms_norm |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
_uses_no_norm |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
_positional_kind |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeRMSNorm.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeRMSNorm.forward |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeRMSNormPre.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeRMSNormPre.forward |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeLayerNormPre.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeLayerNormPre.forward |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
_uses_param_free_norm |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
_make_norm |
10 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
_uses_causal_attention |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
_resolve_rope_scaling |
26 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeRotary.__init__ |
14 |
1 |
0 |
2 |
1 |
88% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeRotary._rotate_half |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeRotary.apply_rope |
9 |
2 |
0 |
2 |
1 |
73% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeRotary.apply_rope._rope |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeAttention.__init__ |
26 |
1 |
0 |
6 |
1 |
94% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeAttention.forward |
24 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeAttention._combine_attention_mask |
9 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeMLP.__init__ |
10 |
1 |
0 |
2 |
1 |
83% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeMLP.forward |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeGatedMLP.__init__ |
10 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeGatedMLP.forward |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeBlock.__init__ |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeBlock.forward |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeModel.__init__ |
18 |
2 |
0 |
6 |
2 |
83% |
| transformer_lens/model_bridge/sources/native/model.py |
NativeModel.forward |
23 |
2 |
0 |
14 |
2 |
89% |
| transformer_lens/model_bridge/sources/native/model.py |
(no function) |
58 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/tl_legacy.py |
_fetch_json |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/tl_legacy.py |
derive_tl_legacy_config |
8 |
1 |
0 |
2 |
1 |
80% |
| transformer_lens/model_bridge/sources/tl_legacy.py |
_convert_neel_solu_old_weights |
15 |
15 |
0 |
12 |
0 |
0% |
| transformer_lens/model_bridge/sources/tl_legacy.py |
_convert_mingpt_weights |
21 |
21 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/sources/tl_legacy.py |
fetch_tl_legacy_state_dict |
14 |
3 |
0 |
6 |
3 |
70% |
| transformer_lens/model_bridge/sources/tl_legacy.py |
boot |
29 |
2 |
0 |
14 |
3 |
84% |
| transformer_lens/model_bridge/sources/tl_legacy.py |
(no function) |
16 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/transformers/__init__.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/transformers/helpers.py |
get_hf_model_class_for_architecture |
27 |
4 |
0 |
18 |
2 |
87% |
| transformer_lens/model_bridge/sources/transformers/helpers.py |
_ensure_torchvision |
17 |
17 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/sources/transformers/helpers.py |
load_modality_processor |
13 |
2 |
0 |
6 |
0 |
89% |
| transformer_lens/model_bridge/sources/transformers/helpers.py |
_resolve_checkpoint_to_revision |
18 |
0 |
0 |
14 |
0 |
100% |
| transformer_lens/model_bridge/sources/transformers/helpers.py |
list_supported_models |
8 |
8 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/transformers/helpers.py |
check_model_support |
12 |
12 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/sources/transformers/helpers.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/transformers/source.py |
boot |
152 |
14 |
0 |
100 |
14 |
89% |
| transformer_lens/model_bridge/sources/transformers/source.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/transformers_driver.py |
TransformersDriver.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/transformers_driver.py |
TransformersDriver.forward |
19 |
0 |
0 |
16 |
1 |
97% |
| transformer_lens/model_bridge/sources/transformers_driver.py |
TransformersDriver.parameters |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/transformers_driver.py |
TransformersDriver.named_parameters |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/transformers_driver.py |
TransformersDriver.underlying_model |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/transformers_driver.py |
TransformersDriver.set_underlying_model |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/transformers_driver.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/__init__.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver.__init__ |
27 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver.forward |
36 |
24 |
0 |
16 |
3 |
37% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._sampler_logprobs |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._read_names |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._expose_captured |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._forward_batched |
30 |
30 |
0 |
14 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._assemble_padded |
18 |
13 |
0 |
8 |
0 |
19% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._assemble_padded._captures_for |
6 |
2 |
0 |
4 |
2 |
60% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._synthesize_logits_batched |
14 |
14 |
0 |
10 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver.get_param |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._rpc_tensor |
5 |
1 |
0 |
4 |
1 |
78% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._rpc_captures |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._gather_param |
10 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._merge_rank_captures |
9 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._verify_rank_layout |
20 |
0 |
0 |
16 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver.probe_logit_reconstruction |
14 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._reconstruct_logits |
18 |
4 |
0 |
10 |
3 |
75% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._unfold_ln_final |
14 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver.close |
26 |
5 |
0 |
6 |
1 |
81% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._synthesize_logits |
10 |
10 |
0 |
8 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._validate_interventions |
12 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._reject_pos_beyond_seq |
8 |
1 |
0 |
6 |
1 |
86% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._normalize_input_ids |
8 |
1 |
0 |
6 |
2 |
79% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
VLLMDriver._normalize_input_ids_batched |
8 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/driver.py |
(no function) |
48 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/internals.py |
extract_hf_config |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/internals.py |
verify_hook_coverage |
11 |
1 |
0 |
8 |
1 |
89% |
| transformer_lens/model_bridge/sources/vllm/internals.py |
_to_cpu_offsets |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/internals.py |
segment_by_request |
18 |
11 |
0 |
14 |
2 |
28% |
| transformer_lens/model_bridge/sources/vllm/internals.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/intervention_specs.py |
validate_spec |
38 |
3 |
0 |
28 |
1 |
94% |
| transformer_lens/model_bridge/sources/vllm/intervention_specs.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/overlays/__init__.py |
get_overlay |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/overlays/__init__.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/overlays/base.py |
AdapterOverlay.capture_specs |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/overlays/base.py |
AdapterOverlay.nonfiring_hooks |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/overlays/base.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/overlays/decoder_only.py |
DecoderOnlyOverlay.capture_specs |
8 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/overlays/decoder_only.py |
DecoderOnlyOverlay.nonfiring_hooks |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/overlays/decoder_only.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
configure |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
clear_config |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
_serialize_config |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
_deserialize_config |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
_active_config |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
register |
7 |
7 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
register.patched_load_model |
40 |
40 |
0 |
10 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
_make_capture_hook |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
_make_capture_hook._affine |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
_make_capture_hook.hook |
21 |
21 |
0 |
10 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
_gated_capture |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
_make_batched_hook |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
_make_batched_hook.hook |
35 |
13 |
0 |
20 |
7 |
60% |
| transformer_lens/model_bridge/sources/vllm/plugin.py |
(no function) |
21 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/source.py |
construct_instrumented_llm |
17 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/source.py |
boot_vllm |
32 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/source.py |
_reject_locked_overrides |
3 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/source.py |
_dtype_from_hf_config |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/source.py |
_log_hook_summary |
6 |
0 |
0 |
2 |
1 |
88% |
| transformer_lens/model_bridge/sources/vllm/source.py |
(no function) |
26 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
dtype_name |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
is_pp_missing_layer |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
resolve_dot_path |
11 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
encode_tensor |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
decode_tensor |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_absent_hooks |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_read_captures |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_set_interventions |
13 |
2 |
0 |
8 |
1 |
86% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_get_param |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_reset_counter |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_reset_capture_flags |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_read_counter |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_reset_accumulators |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_read_batched_captures |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_set_batched_interventions |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
TLWorkerExtension.tl_remove_hooks |
10 |
1 |
0 |
2 |
1 |
83% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
_apply_op |
13 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
_write_rows |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
_apply_intervention |
31 |
2 |
0 |
18 |
2 |
92% |
| transformer_lens/model_bridge/sources/vllm/worker_extension.py |
(no function) |
34 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/__init__.py |
(no function) |
157 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/_remote_code_compat.py |
iter_remote_modeling_modules |
10 |
0 |
0 |
8 |
1 |
94% |
| transformer_lens/model_bridge/supported_architectures/_remote_code_compat.py |
force_import_remote_class |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/_remote_code_compat.py |
patch_init_weights_skip_loaded |
10 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/_remote_code_compat.py |
patch_init_weights_skip_loaded.safe_init_weights |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/_remote_code_compat.py |
retie_weights_keys_v5 |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/_remote_code_compat.py |
disable_tied_weights_lookup |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/_remote_code_compat.py |
compute_default_rope_inv_freq |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/_remote_code_compat.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/afmoe.py |
AfmoeArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/afmoe.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/apertus.py |
ApertusArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/apertus.py |
ApertusArchitectureAdapter.prepare_loading |
16 |
3 |
0 |
4 |
1 |
80% |
| transformer_lens/model_bridge/supported_architectures/apertus.py |
ApertusArchitectureAdapter.prepare_loading._patched_init |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/apertus.py |
ApertusArchitectureAdapter.prepare_loading._patched_forward |
4 |
0 |
0 |
2 |
1 |
83% |
| transformer_lens/model_bridge/supported_architectures/apertus.py |
ApertusArchitectureAdapter._xielu_needs_patch |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/apertus.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/arcee.py |
ArceeArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/arcee.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/ast.py |
ASTArchitectureAdapter.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/ast.py |
ASTArchitectureAdapter._build_component_mapping |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/ast.py |
ASTArchitectureAdapter.prepare_model |
12 |
1 |
0 |
6 |
3 |
78% |
| transformer_lens/model_bridge/supported_architectures/ast.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/audio_flamingo3.py |
(no function) |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/baichuan.py |
_BaichuanAttentionBridge._reconstruct_attention |
46 |
6 |
0 |
14 |
4 |
83% |
| transformer_lens/model_bridge/supported_architectures/baichuan.py |
_patch_init_weights_for_baichuan |
9 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/baichuan.py |
BaichuanArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/baichuan.py |
BaichuanArchitectureAdapter._split_baichuan_w_pack |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/baichuan.py |
BaichuanArchitectureAdapter._split_baichuan_w_pack._make_linear |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/baichuan.py |
BaichuanArchitectureAdapter.setup_component_testing |
10 |
10 |
0 |
6 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/baichuan.py |
BaichuanArchitectureAdapter.prepare_loading |
20 |
4 |
0 |
6 |
1 |
81% |
| transformer_lens/model_bridge/supported_architectures/baichuan.py |
BaichuanArchitectureAdapter.prepare_model |
27 |
5 |
0 |
14 |
2 |
73% |
| transformer_lens/model_bridge/supported_architectures/baichuan.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bamba.py |
BambaArchitectureAdapter.__init__ |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bamba.py |
BambaArchitectureAdapter.create_stateful_cache |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bamba.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bart.py |
BartFamilyArchitectureAdapter.__init__ |
29 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bart.py |
BartFamilyArchitectureAdapter._norm |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bart.py |
BartFamilyArchitectureAdapter._attention |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bart.py |
BartFamilyArchitectureAdapter._encoder_attention |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bart.py |
BartFamilyArchitectureAdapter._mlp |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bart.py |
BartFamilyArchitectureAdapter._encoder_block |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bart.py |
BartFamilyArchitectureAdapter._decoder_block |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bart.py |
BartFamilyArchitectureAdapter._build_component_mapping |
15 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bart.py |
BartFamilyArchitectureAdapter.setup_hook_compatibility |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bart.py |
(no function) |
22 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bd3lm.py |
BD3LMArchitectureAdapter.__init__ |
21 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bd3lm.py |
BD3LMArchitectureAdapter.prepare_loading |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/bd3lm.py |
BD3LMArchitectureAdapter.prepare_model |
18 |
2 |
0 |
12 |
3 |
83% |
| transformer_lens/model_bridge/supported_architectures/bd3lm.py |
BD3LMArchitectureAdapter.prepare_model.patched_forward |
6 |
6 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/bd3lm.py |
BD3LMArchitectureAdapter.convert_weights |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/bd3lm.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bert.py |
BertArchitectureAdapter.__init__ |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bert.py |
BertArchitectureAdapter.prepare_model |
10 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bert.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bitnet.py |
_BitNetAttentionBridge._pre_output_projection |
5 |
0 |
0 |
2 |
1 |
86% |
| transformer_lens/model_bridge/supported_architectures/bitnet.py |
BitNetArchitectureAdapter.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bitnet.py |
BitNetArchitectureAdapter.prepare_model |
7 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bitnet.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/blenderbot.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bloom.py |
BloomArchitectureAdapter.__init__ |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bloom.py |
BloomArchitectureAdapter.split_qkv_matrix |
21 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/bloom.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/codegen.py |
CodeGenArchitectureAdapter.__init__ |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/codegen.py |
CodeGenArchitectureAdapter.split_qkv_matrix |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/codegen.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/cohere.py |
CohereArchitectureAdapter.__init__ |
23 |
1 |
0 |
4 |
2 |
89% |
| transformer_lens/model_bridge/supported_architectures/cohere.py |
CohereArchitectureAdapter.preprocess_weights |
11 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/cohere.py |
CohereArchitectureAdapter.postprocess_weights |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/cohere.py |
CohereArchitectureAdapter.apply_output_logits_transform |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/cohere.py |
_Cohere2AttentionBridge.forward |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/cohere.py |
_Cohere2AttentionBridge._is_nope_layer |
10 |
2 |
0 |
6 |
2 |
75% |
| transformer_lens/model_bridge/supported_architectures/cohere.py |
_cohere2_layer_types |
19 |
1 |
0 |
12 |
1 |
94% |
| transformer_lens/model_bridge/supported_architectures/cohere.py |
Cohere2ArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/cohere.py |
(no function) |
17 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v2.py |
DeepSeekMLAFamilyArchitectureAdapter.__init__ |
9 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v2.py |
DeepSeekMLAFamilyArchitectureAdapter._build_component_mapping |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v2.py |
DeepSeekMLAFamilyArchitectureAdapter._build_attention_submodules |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v2.py |
DeepSeekMLAFamilyArchitectureAdapter._build_q_a_layernorm |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v2.py |
DeepSeekMLAFamilyArchitectureAdapter._build_router |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v2.py |
DeepSeekMLAFamilyArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v2.py |
(no function) |
18 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v3.py |
DeepSeekV3ArchitectureAdapter._build_q_a_layernorm |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v3.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v4.py |
DeepseekV4HyperConnectionBridge.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v4.py |
DeepseekV4HyperConnectionBridge.forward |
11 |
4 |
0 |
8 |
3 |
53% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v4.py |
DeepseekV4CompressorBridge.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v4.py |
DeepseekV4CompressorBridge.forward |
14 |
4 |
0 |
10 |
4 |
58% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v4.py |
_compressor_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v4.py |
DeepSeekV4ArchitectureAdapter.__init__ |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v4.py |
DeepSeekV4ArchitectureAdapter.__init__.hyper_connection |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v4.py |
DeepSeekV4ArchitectureAdapter.prepare_loading |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v4.py |
DeepSeekV4ArchitectureAdapter.prepare_model |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/deepseek_v4.py |
(no function) |
23 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/dream.py |
_patch_eager_attention_mask |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/supported_architectures/dream.py |
_patch_eager_attention_mask.forward |
7 |
7 |
0 |
6 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/dream.py |
_patch_from_model_config |
5 |
1 |
0 |
2 |
2 |
57% |
| transformer_lens/model_bridge/supported_architectures/dream.py |
_patch_from_model_config.from_model_config |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/dream.py |
_register_default_rope_init |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/dream.py |
DreamArchitectureAdapter.native_sampler_kwargs |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/dream.py |
DreamArchitectureAdapter._build_attention_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/dream.py |
DreamArchitectureAdapter.prepare_loading |
9 |
0 |
0 |
4 |
2 |
85% |
| transformer_lens/model_bridge/supported_architectures/dream.py |
(no function) |
18 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/emu3.py |
Emu3ArchitectureAdapter.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/emu3.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/ernie4_5.py |
Ernie4_5ArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/ernie4_5.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/ernie4_5_moe.py |
Ernie4_5_MoeArchitectureAdapter.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/ernie4_5_moe.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/exaone.py |
ExaoneArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/exaone.py |
ExaoneArchitectureAdapter.prepare_model |
9 |
9 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/exaone.py |
ExaoneArchitectureAdapter.prepare_model._shim |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/exaone.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/exaone4.py |
_Exaone4AttentionBridge.forward |
5 |
1 |
0 |
4 |
1 |
78% |
| transformer_lens/model_bridge/supported_architectures/exaone4.py |
_Exaone4AttentionBridge._is_nope_layer |
6 |
1 |
0 |
4 |
1 |
80% |
| transformer_lens/model_bridge/supported_architectures/exaone4.py |
Exaone4ArchitectureAdapter.__init__ |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/exaone4.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/falcon.py |
_patch_decoder_inplace_add |
7 |
1 |
0 |
2 |
1 |
78% |
| transformer_lens/model_bridge/supported_architectures/falcon.py |
_patch_decoder_inplace_add.patched_forward |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/falcon.py |
FalconArchitectureAdapter.__init__ |
26 |
1 |
0 |
10 |
1 |
94% |
| transformer_lens/model_bridge/supported_architectures/falcon.py |
FalconArchitectureAdapter.prepare_model |
7 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/falcon.py |
FalconArchitectureAdapter.prepare_model._make_patched_linear |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/falcon.py |
FalconArchitectureAdapter.prepare_model._make_patched_linear.patched_forward |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/supported_architectures/falcon.py |
FalconArchitectureAdapter._split_falcon_qkv |
32 |
19 |
0 |
10 |
3 |
38% |
| transformer_lens/model_bridge/supported_architectures/falcon.py |
FalconArchitectureAdapter._split_falcon_qkv.build_linear |
5 |
1 |
0 |
2 |
1 |
71% |
| transformer_lens/model_bridge/supported_architectures/falcon.py |
FalconArchitectureAdapter.setup_component_testing |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/falcon.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/falcon_h1.py |
_make_optional |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/falcon_h1.py |
FalconH1ArchitectureAdapter.__init__ |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/falcon_h1.py |
FalconH1ArchitectureAdapter.apply_output_logits_transform |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/falcon_h1.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/falcon_mamba.py |
(no function) |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/flex_olmo.py |
FlexOlmoArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/flex_olmo.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/florence2.py |
Florence2ArchitectureAdapter.__init__ |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/florence2.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma1.py |
Gemma1ArchitectureAdapter.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma1.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma2.py |
Gemma2ArchitectureAdapter.__init__ |
10 |
2 |
0 |
4 |
2 |
71% |
| transformer_lens/model_bridge/supported_architectures/gemma2.py |
Gemma2ArchitectureAdapter._block_norms |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma2.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma3.py |
_enable_native_qk_norm_autograd |
10 |
10 |
0 |
10 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/gemma3.py |
Gemma3ArchitectureAdapter.__init__ |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma3.py |
Gemma3ArchitectureAdapter.setup_component_testing |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/gemma3.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma3_multimodal.py |
Gemma3MultimodalArchitectureAdapter.__init__ |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma3_multimodal.py |
Gemma3MultimodalArchitectureAdapter.setup_component_testing |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/gemma3_multimodal.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma3n.py |
_SparsityPreservingGatedMLPBridge.set_processed_weights |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma3n.py |
Gemma3nArchitectureAdapter.__init__ |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma3n.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma4.py |
Gemma4ArchitectureAdapter.__init__ |
19 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma4.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma4_text.py |
Gemma4TextArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gemma4_text.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gidd.py |
restore_frequencies |
13 |
1 |
0 |
4 |
1 |
88% |
| transformer_lens/model_bridge/supported_architectures/gidd.py |
GiddArchitectureAdapter.native_sampler_kwargs |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gidd.py |
GiddArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gidd.py |
GiddArchitectureAdapter.prepare_loading |
8 |
8 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/gidd.py |
GiddArchitectureAdapter.prepare_model |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/gidd.py |
(no function) |
17 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm.py |
GlmArchitectureAdapter.__init__ |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm.py |
GlmArchitectureAdapter._block_extra_norms |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm4.py |
Glm4ArchitectureAdapter._block_extra_norms |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm4.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm4_moe.py |
Glm4MoeArchitectureAdapter.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm4_moe.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm4_moe_lite.py |
Glm4MoeLiteArchitectureAdapter._build_router |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm4_moe_lite.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm4v.py |
Glm4vArchitectureAdapter.__init__ |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm4v.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm_asr.py |
(no function) |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/glm_moe_dsa.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt2.py |
QKVSplitRearrangeConversion.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt2.py |
QKVSplitRearrangeConversion._is_combined_qkv |
7 |
4 |
0 |
4 |
1 |
36% |
| transformer_lens/model_bridge/supported_architectures/gpt2.py |
QKVSplitRearrangeConversion.handle_conversion |
10 |
3 |
0 |
6 |
1 |
62% |
| transformer_lens/model_bridge/supported_architectures/gpt2.py |
QKVSplitRearrangeConversion.revert |
5 |
3 |
0 |
4 |
1 |
33% |
| transformer_lens/model_bridge/supported_architectures/gpt2.py |
GPT2ArchitectureAdapter.__init__ |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt2.py |
(no function) |
14 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt2_lm_head_custom.py |
Gpt2LmHeadCustomArchitectureAdapter.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt2_lm_head_custom.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt_bigcode.py |
MQAQKVConversionRule.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt_bigcode.py |
MQAQKVConversionRule.handle_conversion |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt_bigcode.py |
MQAQKVConversionRule.revert |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt_bigcode.py |
GPTBigCodeArchitectureAdapter.__init__ |
16 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt_bigcode.py |
GPTBigCodeArchitectureAdapter._split_qkv_matrix |
13 |
0 |
0 |
2 |
1 |
93% |
| transformer_lens/model_bridge/supported_architectures/gpt_bigcode.py |
GPTBigCodeArchitectureAdapter._split_qkv_matrix._make_linear |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt_bigcode.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt_oss.py |
GPTOSSArchitectureAdapter.__init__ |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt_oss.py |
GPTOSSArchitectureAdapter.setup_hook_compatibility |
7 |
0 |
0 |
8 |
1 |
93% |
| transformer_lens/model_bridge/supported_architectures/gpt_oss.py |
GPTOSSArchitectureAdapter.setup_no_processing_hooks |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gpt_oss.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gptj.py |
GptjArchitectureAdapter.__init__ |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/gptj.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite.py |
GraniteArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite.py |
GraniteArchitectureAdapter._setup_common_config |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite.py |
GraniteArchitectureAdapter._build_attention_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite.py |
GraniteArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite.py |
GraniteArchitectureAdapter._build_moe_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite.py |
GraniteArchitectureAdapter._build_component_mapping |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite.py |
GraniteArchitectureAdapter.apply_output_logits_transform |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite.py |
(no function) |
14 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite_moe.py |
GraniteMoeArchitectureAdapter._build_component_mapping |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite_moe.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite_moe_hybrid.py |
GraniteMoeHybridArchitectureAdapter.__init__ |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite_moe_hybrid.py |
GraniteMoeHybridArchitectureAdapter._build_mamba_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite_moe_hybrid.py |
GraniteMoeHybridArchitectureAdapter._build_component_mapping |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/granite_moe_hybrid.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hrm_text.py |
HrmTextArchitectureAdapter.__init__ |
13 |
2 |
0 |
8 |
2 |
81% |
| transformer_lens/model_bridge/supported_architectures/hrm_text.py |
HrmTextArchitectureAdapter.__init__._make_block_submodules |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hrm_text.py |
HrmTextArchitectureAdapter._build_weight_conversions |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hrm_text.py |
HrmTextArchitectureAdapter.setup_component_testing |
15 |
15 |
0 |
12 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/hrm_text.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hubert.py |
HubertArchitectureAdapter.__init__ |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hubert.py |
HubertArchitectureAdapter._build_component_mapping |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hubert.py |
HubertArchitectureAdapter.prepare_loading |
8 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hubert.py |
HubertArchitectureAdapter.prepare_model |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hubert.py |
(no function) |
14 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hunyuan_v1_dense.py |
HunYuanDenseV1ArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hunyuan_v1_dense.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hyenadna.py |
_HyenaBlockBridge._is_standalone_hidden_state_call |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hyenadna.py |
HyenaDNAArchitectureAdapter.__init__ |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/hyenadna.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/idefics3.py |
Idefics3ArchitectureAdapter.__init__ |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/idefics3.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/internlm2.py |
_InternLM2AttentionBridge.forward |
11 |
1 |
0 |
8 |
3 |
79% |
| transformer_lens/model_bridge/supported_architectures/internlm2.py |
_InternLM2AttentionBridge._reconstruct_attention |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/internlm2.py |
InternLM2ArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/internlm2.py |
InternLM2ArchitectureAdapter._split_internlm2_wqkv |
23 |
0 |
0 |
2 |
1 |
96% |
| transformer_lens/model_bridge/supported_architectures/internlm2.py |
InternLM2ArchitectureAdapter._split_internlm2_wqkv._make_linear |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/internlm2.py |
InternLM2ArchitectureAdapter.setup_component_testing |
10 |
10 |
0 |
6 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/internlm2.py |
InternLM2ArchitectureAdapter.prepare_model |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/internlm2.py |
InternLM2ArchitectureAdapter.prepare_loading |
11 |
11 |
0 |
8 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/internlm2.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/jais2.py |
Jais2ArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/jais2.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/jamba.py |
_make_optional |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/jamba.py |
JambaArchitectureAdapter.__init__ |
31 |
0 |
0 |
2 |
1 |
97% |
| transformer_lens/model_bridge/supported_architectures/jamba.py |
JambaArchitectureAdapter._build_mamba_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/jamba.py |
JambaArchitectureAdapter._build_attention_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/jamba.py |
JambaArchitectureAdapter._build_ffn_bridge |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/jamba.py |
JambaArchitectureAdapter._build_component_mapping |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/jamba.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/jetmoe.py |
JetMoeArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/jetmoe.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/laguna.py |
LagunaArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/laguna.py |
LagunaArchitectureAdapter.prepare_loading |
9 |
2 |
0 |
4 |
2 |
69% |
| transformer_lens/model_bridge/supported_architectures/laguna.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/led.py |
LEDArchitectureAdapter.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/led.py |
LEDArchitectureAdapter._encoder_attention |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/led.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/lfm2.py |
Lfm2ArchitectureAdapter.__init__ |
8 |
0 |
0 |
2 |
1 |
90% |
| transformer_lens/model_bridge/supported_architectures/lfm2.py |
Lfm2ArchitectureAdapter.setup_component_testing |
11 |
0 |
0 |
8 |
1 |
95% |
| transformer_lens/model_bridge/supported_architectures/lfm2.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/lfm2_moe.py |
Lfm2MoeGateBridge.get_random_inputs |
9 |
9 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/lfm2_moe.py |
Lfm2MoeArchitectureAdapter.__init__ |
10 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/lfm2_moe.py |
Lfm2MoeArchitectureAdapter.setup_component_testing |
16 |
0 |
0 |
16 |
4 |
88% |
| transformer_lens/model_bridge/supported_architectures/lfm2_moe.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDAAttentionBridge.set_original_component |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDAAttentionBridge.original_component |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDAAttentionBridge.forward |
57 |
9 |
0 |
24 |
10 |
77% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDABlockBridge.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDABlockBridge._route_attn_input |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDABlockBridge._route_mlp_input |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDABlockBridge._route_mlp_output |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDABlockBridge._route_pre_mlp_norm |
5 |
3 |
0 |
4 |
1 |
33% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDABlockBridge._maybe_wire_capture_hooks |
5 |
0 |
0 |
4 |
1 |
89% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDABlockBridge._wire_llada_container_hooks |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDABlockBridge.forward |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDAGatedMLPBridge.set_original_component |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
_LLaDAGatedMLPBridge.forward |
12 |
12 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
LLaDAArchitectureAdapter.__init__ |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
LLaDAArchitectureAdapter._enum_value |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
LLaDAArchitectureAdapter._require_value |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
LLaDAArchitectureAdapter._validate_config |
18 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
LLaDAArchitectureAdapter.prepare_loading |
5 |
0 |
0 |
2 |
1 |
86% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
LLaDAArchitectureAdapter.prepare_model |
4 |
0 |
0 |
4 |
1 |
88% |
| transformer_lens/model_bridge/supported_architectures/llada.py |
(no function) |
39 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada2_moe.py |
LLaDA2MoeArchitectureAdapter.native_sampler_kwargs |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada2_moe.py |
LLaDA2MoeArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada2_moe.py |
LLaDA2MoeArchitectureAdapter.prepare_loading |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llada2_moe.py |
LLaDA2MoeArchitectureAdapter.setup_hook_compatibility |
6 |
1 |
0 |
2 |
2 |
62% |
| transformer_lens/model_bridge/supported_architectures/llada2_moe.py |
LLaDA2MoeArchitectureAdapter.setup_hook_compatibility._mask_guard |
14 |
0 |
0 |
8 |
1 |
95% |
| transformer_lens/model_bridge/supported_architectures/llada2_moe.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llama.py |
LlamaArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llama.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llama4.py |
_Llama4MoEBridge.forward |
16 |
3 |
0 |
8 |
3 |
75% |
| transformer_lens/model_bridge/supported_architectures/llama4.py |
Llama4ArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llama4.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llama4_multimodal.py |
Llama4MultimodalArchitectureAdapter.__init__ |
8 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llama4_multimodal.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llava.py |
LlavaArchitectureAdapter.__init__ |
12 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llava.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llava_next.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/llava_onevision.py |
LlavaOnevisionArchitectureAdapter.prepare_model |
12 |
2 |
0 |
10 |
2 |
82% |
| transformer_lens/model_bridge/supported_architectures/llava_onevision.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/longt5.py |
LongT5ArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/longt5.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/m2m100.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mamba.py |
MambaArchitectureAdapter.__init__ |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mamba.py |
MambaArchitectureAdapter.create_stateful_cache |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/supported_architectures/mamba.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mamba2.py |
Mamba2ArchitectureAdapter.__init__ |
21 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mamba2.py |
Mamba2ArchitectureAdapter.create_stateful_cache |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/supported_architectures/mamba2.py |
compute_effective_attention |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mamba2.py |
compute_ssm_state |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mamba2.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/marian.py |
(no function) |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mbart.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mingpt.py |
MingptArchitectureAdapter.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mingpt.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/minimax_m2.py |
MiniMaxM2ArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/minimax_m2.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/ministral3.py |
(no function) |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mistral.py |
MistralArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mistral.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mistral3.py |
Mistral3ArchitectureAdapter.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mistral3.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mixtral.py |
MixtralArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mixtral.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/modernbert_decoder.py |
ModernBertDecoderArchitectureAdapter.__init__ |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/modernbert_decoder.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mpt.py |
MPTArchitectureAdapter.__init__ |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mpt.py |
MPTArchitectureAdapter.validate_output_logits_transform |
3 |
0 |
0 |
2 |
1 |
80% |
| transformer_lens/model_bridge/supported_architectures/mpt.py |
MPTArchitectureAdapter._split_mpt_qkv |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mpt.py |
MPTArchitectureAdapter._split_mpt_qkv.make_linear |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/mpt.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/music_flamingo.py |
(no function) |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/nanochat.py |
NanoChatArchitectureAdapter.__init__ |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/nanochat.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/nanogpt.py |
NanogptArchitectureAdapter.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/nanogpt.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/native.py |
_norm_type |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/native.py |
_uses_rms |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/native.py |
_uses_param_free_norm |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/native.py |
_uses_no_norm |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/native.py |
_is_rotary |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/native.py |
_make_norm_bridge |
11 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/native.py |
_make_mlp_bridge |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/native.py |
_make_block_submodules |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/native.py |
NativeArchitectureAdapter.__init__ |
17 |
0 |
0 |
6 |
1 |
96% |
| transformer_lens/model_bridge/supported_architectures/native.py |
NativeArchitectureAdapter.prepare_model |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/native.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neel_solu_old.py |
NeelSoluOldArchitectureAdapter.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neel_solu_old.py |
convert_neel_solu_old_weights |
15 |
15 |
0 |
12 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/neel_solu_old.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/nemotron.py |
NemotronArchitectureAdapter.__init__ |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/nemotron.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/nemotron_h.py |
_make_optional |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/nemotron_h.py |
NemotronHArchitectureAdapter.__init__ |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/nemotron_h.py |
NemotronHArchitectureAdapter.create_stateful_cache |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/nemotron_h.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neo.py |
NeoLinearTransposeConversion.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neo.py |
NeoLinearTransposeConversion.handle_conversion |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neo.py |
NeoLinearTransposeConversion.revert |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neo.py |
NeoArchitectureAdapter.__init__ |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neo.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neox.py |
NeoxArchitectureAdapter.__init__ |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neox.py |
NeoxArchitectureAdapter.prepare_model |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neox.py |
NeoxArchitectureAdapter.split_qkv_matrix |
27 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/neox.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmo.py |
OlmoArchitectureAdapter.__init__ |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmo.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmo2.py |
Olmo2ArchitectureAdapter.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmo2.py |
Olmo2ArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmo2.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmo3.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmo_hybrid.py |
_OlmoHybridBlockBridge.__init__ |
5 |
1 |
0 |
2 |
1 |
71% |
| transformer_lens/model_bridge/supported_architectures/olmo_hybrid.py |
_OlmoHybridBlockBridge.set_original_component |
10 |
1 |
0 |
4 |
1 |
86% |
| transformer_lens/model_bridge/supported_architectures/olmo_hybrid.py |
OlmoHybridArchitectureAdapter.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmo_hybrid.py |
OlmoHybridArchitectureAdapter.create_stateful_cache |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmo_hybrid.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmoe.py |
OlmoeArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/olmoe.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/openai_gpt.py |
_OpenAIGPTJointQKVAttentionBridge._process_output |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/supported_architectures/openai_gpt.py |
OpenAIGPTArchitectureAdapter.__init__ |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/openai_gpt.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/openelm.py |
OpenElmArchitectureAdapter.__init__ |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/openelm.py |
OpenElmArchitectureAdapter.prepare_loading |
15 |
1 |
0 |
8 |
3 |
83% |
| transformer_lens/model_bridge/supported_architectures/openelm.py |
OpenElmArchitectureAdapter.prepare_loading.safe_compute |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/openelm.py |
OpenElmArchitectureAdapter.prepare_model |
26 |
23 |
0 |
18 |
1 |
14% |
| transformer_lens/model_bridge/supported_architectures/openelm.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/opt.py |
_UnflattenTokens.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/opt.py |
_UnflattenTokens.handle_conversion |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/opt.py |
_UnflattenTokens.revert |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/opt.py |
_OptBlockBridge.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/opt.py |
_OptBlockBridge.forward |
6 |
0 |
0 |
4 |
1 |
90% |
| transformer_lens/model_bridge/supported_architectures/opt.py |
_OptBlockBridge._unflatten_conversions |
14 |
1 |
0 |
12 |
3 |
85% |
| transformer_lens/model_bridge/supported_architectures/opt.py |
OptArchitectureAdapter._with_unflatten |
7 |
0 |
0 |
6 |
1 |
92% |
| transformer_lens/model_bridge/supported_architectures/opt.py |
OptArchitectureAdapter.__init__ |
19 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/opt.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/ouro.py |
OuroArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/ouro.py |
OuroArchitectureAdapter.prepare_loading |
9 |
9 |
0 |
8 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/ouro.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/pegasus.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/phi.py |
PhiArchitectureAdapter.__init__ |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/phi.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/phi3.py |
_SizedSplitConversion.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/phi3.py |
_SizedSplitConversion.handle_conversion |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/phi3.py |
Phi3ArchitectureAdapter.__init__ |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/phi3.py |
Phi3ArchitectureAdapter._split_gate_up |
17 |
3 |
0 |
6 |
3 |
74% |
| transformer_lens/model_bridge/supported_architectures/phi3.py |
Phi3ArchitectureAdapter._split_phi3_qkv |
24 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/phi3.py |
Phi3ArchitectureAdapter.prepare_loading |
9 |
9 |
0 |
6 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/phi3.py |
Phi3ArchitectureAdapter.preprocess_weights |
33 |
2 |
0 |
22 |
6 |
82% |
| transformer_lens/model_bridge/supported_architectures/phi3.py |
(no function) |
20 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/phimoe.py |
PhiMoEArchitectureAdapter.__init__ |
29 |
0 |
0 |
16 |
2 |
96% |
| transformer_lens/model_bridge/supported_architectures/phimoe.py |
PhiMoEArchitectureAdapter.prepare_loading |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/phimoe.py |
PhiMoEArchitectureAdapter.prepare_model |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/phimoe.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/pretrain.py |
DenseOrMoEFeedForwardBridge.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/pretrain.py |
DenseOrMoEFeedForwardBridge.set_original_component |
17 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/pretrain.py |
DenseOrMoEFeedForwardBridge.forward |
17 |
5 |
0 |
12 |
4 |
62% |
| transformer_lens/model_bridge/supported_architectures/pretrain.py |
_LogitsAttrDict.__getattr__ |
4 |
2 |
0 |
0 |
0 |
50% |
| transformer_lens/model_bridge/supported_architectures/pretrain.py |
PretrainModelContainer.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/pretrain.py |
PretrainModelContainer.forward |
28 |
1 |
0 |
20 |
1 |
96% |
| transformer_lens/model_bridge/supported_architectures/pretrain.py |
PretrainArchitectureAdapter.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/pretrain.py |
build_pretrain_bridge |
9 |
2 |
0 |
6 |
2 |
73% |
| transformer_lens/model_bridge/supported_architectures/pretrain.py |
(no function) |
26 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen.py |
QwenArchitectureAdapter.__init__ |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen.py |
QwenArchitectureAdapter._split_qkv_matrix |
27 |
1 |
0 |
8 |
1 |
94% |
| transformer_lens/model_bridge/supported_architectures/qwen.py |
QwenArchitectureAdapter._split_qkv_matrix.build_linear |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen2.py |
Qwen2ArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen2.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen2_5_vl.py |
Qwen2_5_VLArchitectureAdapter.__init__ |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen2_5_vl.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen2_audio.py |
Qwen2AudioArchitectureAdapter.__init__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen2_audio.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen2_moe.py |
Qwen2MoeArchitectureAdapter.__init__ |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/model_bridge/supported_architectures/qwen2_moe.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3.py |
Qwen3ArchitectureAdapter.__init__ |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3.py |
Qwen3ArchitectureAdapter._setup_qwen3_config |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3.py |
Qwen3ArchitectureAdapter._build_attention_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3.py |
Qwen3ArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3.py |
Qwen3ArchitectureAdapter._build_linear_attn_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3.py |
Qwen3ArchitectureAdapter._build_component_mapping |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_5.py |
Qwen3_5ArchitectureAdapter.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_5.py |
Qwen3_5ArchitectureAdapter.prepare_loading |
3 |
0 |
0 |
2 |
1 |
80% |
| transformer_lens/model_bridge/supported_architectures/qwen3_5.py |
Qwen3_5ArchitectureAdapter.prepare_model |
10 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_5.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_5_moe.py |
_sparse_moe_mlp |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_5_moe.py |
Qwen3_5MoeArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_5_moe.py |
Qwen3_5MoeMultimodalArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_5_moe.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_5_multimodal.py |
Qwen3_5MultimodalArchitectureAdapter.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_5_multimodal.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_moe.py |
Qwen3MoeArchitectureAdapter.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_moe.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_next.py |
Qwen3NextArchitectureAdapter.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_next.py |
Qwen3NextArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_next.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_vl.py |
Qwen3VLArchitectureAdapter.__init__ |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_vl.py |
Qwen3VLArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_vl.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_vl_moe.py |
Qwen3VLMoeArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/qwen3_vl_moe.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/raven.py |
RavenArchitectureAdapter.__init__ |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/raven.py |
RavenArchitectureAdapter._sandwich_submodules |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/raven.py |
RavenArchitectureAdapter.prepare_loading |
7 |
1 |
0 |
6 |
2 |
77% |
| transformer_lens/model_bridge/supported_architectures/raven.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/recurrent_gemma.py |
RecurrentGemmaArchitectureAdapter.__init__ |
13 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/recurrent_gemma.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/rwkv.py |
RwkvArchitectureAdapter.__init__ |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/rwkv.py |
RwkvArchitectureAdapter.prepare_loading |
4 |
0 |
0 |
2 |
1 |
83% |
| transformer_lens/model_bridge/supported_architectures/rwkv.py |
RwkvArchitectureAdapter.prepare_model |
4 |
0 |
0 |
2 |
1 |
83% |
| transformer_lens/model_bridge/supported_architectures/rwkv.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/rwkv7.py |
RWKV7ArchitectureAdapter.__init__ |
25 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/rwkv7.py |
RWKV7ArchitectureAdapter.prepare_loading |
10 |
1 |
0 |
6 |
2 |
81% |
| transformer_lens/model_bridge/supported_architectures/rwkv7.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/seed_oss.py |
SeedOssArchitectureAdapter.__init__ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/seed_oss.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/smollm3.py |
_SmolLM3AttentionBridge.forward |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/smollm3.py |
SmolLM3ArchitectureAdapter.__init__ |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/smollm3.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/stablelm.py |
StableLmArchitectureAdapter.__init__ |
18 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/stablelm.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/starcoder2.py |
Starcoder2ArchitectureAdapter.__init__ |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/starcoder2.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/switch_transformers.py |
_SwitchBlockBridge._is_standalone_hidden_state_call |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/switch_transformers.py |
SwitchTransformersArchitectureAdapter.__init__ |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/switch_transformers.py |
SwitchTransformersArchitectureAdapter.__init__.attention |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/switch_transformers.py |
SwitchTransformersArchitectureAdapter.prepare_loading |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/switch_transformers.py |
SwitchTransformersArchitectureAdapter._build_ff_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/switch_transformers.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/t5.py |
T5ArchitectureAdapter.__init__ |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/t5.py |
T5ArchitectureAdapter._build_ff_bridge |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/t5.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/t5gemma.py |
T5GemmaArchitectureAdapter.__init__ |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/t5gemma.py |
T5GemmaArchitectureAdapter.setup_component_testing |
14 |
14 |
0 |
10 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/t5gemma.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/t5gemma2.py |
T5Gemma2ArchitectureAdapter.__init__ |
16 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/t5gemma2.py |
T5Gemma2ArchitectureAdapter.setup_component_testing |
12 |
12 |
0 |
8 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/t5gemma2.py |
T5Gemma2ArchitectureAdapter.setup_component_testing._enable_qk_native_autograd |
8 |
8 |
0 |
8 |
0 |
0% |
| transformer_lens/model_bridge/supported_architectures/t5gemma2.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/vaultgemma.py |
VaultGemmaArchitectureAdapter.__init__ |
4 |
0 |
0 |
4 |
1 |
88% |
| transformer_lens/model_bridge/supported_architectures/vaultgemma.py |
VaultGemmaArchitectureAdapter._block_norms |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/vaultgemma.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/vit.py |
ViTArchitectureAdapter.__init__ |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/vit.py |
ViTArchitectureAdapter._build_component_mapping |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/vit.py |
ViTArchitectureAdapter.prepare_model |
11 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/vit.py |
(no function) |
13 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/wav2vec2.py |
Wav2Vec2ArchitectureAdapter.prepare_model |
6 |
0 |
0 |
6 |
1 |
92% |
| transformer_lens/model_bridge/supported_architectures/wav2vec2.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/xglm.py |
XGLMArchitectureAdapter.__init__ |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/xglm.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/youtu.py |
YoutuArchitectureAdapter._build_mlp_bridge |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/youtu.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/zamba2.py |
_make_optional |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/zamba2.py |
Zamba2ArchitectureAdapter.__init__ |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/supported_architectures/zamba2.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
_resolve_attr_path |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
_storage_group_keys |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.__init__ |
26 |
4 |
0 |
6 |
1 |
84% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.__setstate__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.original_model |
10 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.original_model |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.init_weights |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._set_processed_weight_attributes |
37 |
37 |
0 |
18 |
0 |
0% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._register_all_aliases_recursive |
5 |
0 |
0 |
6 |
1 |
91% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.__setattr__ |
13 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._scan_existing_hooks |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._scan_existing_hooks.scan_module |
38 |
0 |
0 |
26 |
1 |
98% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.n_params_total |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._has_registered_blocks |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.__getattr__ |
27 |
5 |
0 |
18 |
2 |
80% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.__str__ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.__str__._describe |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.enable_compatibility_mode |
24 |
0 |
0 |
16 |
3 |
92% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.enable_compatibility_mode.set_compatibility_mode |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._setup_hook_compatibility |
20 |
2 |
0 |
20 |
2 |
90% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.process_weights |
51 |
10 |
0 |
36 |
11 |
76% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._fold_layer_norms_into_batched_experts |
28 |
8 |
0 |
18 |
5 |
72% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.to_tokens |
22 |
0 |
0 |
18 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.encoder_output |
20 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._audio_encoder_is_stable_layer_norm |
9 |
2 |
0 |
6 |
3 |
67% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._require_frame_entry_support |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.to_sentence_pair_tokens |
9 |
1 |
0 |
4 |
2 |
77% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.predict_next_sentence |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.to_string |
7 |
2 |
0 |
6 |
2 |
69% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.to_str_tokens |
19 |
12 |
0 |
12 |
1 |
32% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.to_single_token |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.get_token_position |
17 |
2 |
0 |
12 |
2 |
86% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.to_single_str_token |
5 |
5 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._enumerate_blocks |
8 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._resolve_submodule_name |
4 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._rewrite_submodule_path |
5 |
1 |
0 |
4 |
1 |
78% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.blocks_with |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.stack_params_for |
15 |
1 |
0 |
8 |
1 |
91% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._stack_block_params |
27 |
0 |
0 |
14 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._reshape_qkv |
4 |
2 |
0 |
2 |
1 |
50% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._reshape_o |
4 |
2 |
0 |
2 |
1 |
50% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._expand_kv_heads |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_K |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_Q |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_V |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_O |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_in |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_gate |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_out |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.b_K |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.b_Q |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.b_V |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.b_O |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.b_in |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.b_out |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_U |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.b_U |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_E |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_pos |
10 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.W_E_pos |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.QK |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.OV |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.QK_for_attn_layers |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.OV_for_attn_layers |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.tokens_to_residual_directions |
13 |
1 |
0 |
8 |
1 |
90% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._get_block_variant_bias |
14 |
2 |
0 |
10 |
2 |
83% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.accumulated_bias |
17 |
0 |
0 |
12 |
1 |
97% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.all_composition_scores |
28 |
1 |
0 |
8 |
1 |
94% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.all_composition_scores._stack |
12 |
3 |
0 |
8 |
3 |
70% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._reject_encoder_decoder_composition |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.composition_layer_indices |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.block_hooks |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.block_submodules |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.layer_types |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.all_head_labels |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.attn_head_labels |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.parameters |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.named_parameters |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.tl_parameters |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.tl_named_parameters |
1 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.input_to_embed |
11 |
3 |
0 |
6 |
2 |
71% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._accepts_derived_position_ids |
10 |
0 |
0 |
4 |
1 |
93% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._accepts_derived_position_ids.verdict |
19 |
0 |
0 |
18 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._seq2seq_loss |
8 |
2 |
0 |
6 |
2 |
71% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._finalize_seq2seq_return |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.forward |
122 |
15 |
0 |
96 |
11 |
85% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._setup_start_at_layer |
22 |
1 |
0 |
18 |
1 |
95% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._teardown_start_at_layer |
4 |
0 |
0 |
4 |
1 |
88% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._resolve_stopping_criteria |
20 |
0 |
0 |
16 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._encdec_ngram_processor |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._generate_tokens |
110 |
2 |
0 |
70 |
2 |
98% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._resolve_generation_caching |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._ensure_generation_supported |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.generate |
192 |
11 |
0 |
130 |
12 |
93% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.generate._logits_to_tuple |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.diffusion_generate |
26 |
5 |
0 |
14 |
6 |
72% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.generate_stream |
90 |
7 |
0 |
50 |
5 |
91% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.generate_stream._decode_delta |
9 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.generate_stream._maybe_decode |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.generate_stream._flush_held |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.hf_generate |
69 |
17 |
0 |
46 |
9 |
70% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.prepare_multimodal_inputs |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.to |
28 |
1 |
0 |
22 |
3 |
92% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.cuda |
5 |
5 |
0 |
4 |
0 |
0% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.cpu |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.mps |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.train |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.set_use_attn_result |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.set_use_split_qkv_input |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.set_use_attn_in |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.set_use_hook_mlp_in |
11 |
3 |
0 |
6 |
1 |
76% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._write_propagated_hook_flag |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._propagate_attention_flag |
12 |
4 |
0 |
8 |
2 |
70% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._validate_attention_fork_supported |
27 |
5 |
0 |
16 |
4 |
79% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._is_valid_bridge_path |
32 |
2 |
0 |
22 |
3 |
91% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._normalize_bridge_key_to_hf |
23 |
1 |
0 |
22 |
1 |
96% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.state_dict |
15 |
1 |
0 |
10 |
1 |
92% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge._tl_key_to_actual_keys |
11 |
1 |
0 |
6 |
1 |
88% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.load_state_dict |
57 |
1 |
0 |
40 |
2 |
97% |
| transformer_lens/model_bridge/transformer_bridge.py |
TransformerBridge.get_params |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_bridge/transformer_bridge.py |
(no function) |
175 |
0 |
2 |
0 |
0 |
100% |
| transformer_lens/model_bridge/types.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TransformerLensModel.cfg |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TransformerLensModel.run_with_cache |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TransformerLensModel.run_with_hooks |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TransformerLensModel.to_tokens |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TrainableTransformerLensModel.parameters |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TrainableTransformerLensModel.train |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TrainableTransformerLensModel.to |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TrainableTransformerLensModel.state_dict |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TrainableTransformerLensModel.__call__ |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TransformerLensModelWithWeights.blocks |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TransformerLensModelWithWeights.ln_final |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TransformerLensModelWithWeights.accumulated_bias |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TransformerLensModelWithWeights.to_single_token |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
TransformerLensModelWithWeights.tokens_to_residual_directions |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/model_protocol.py |
(no function) |
8 |
0 |
35 |
0 |
0 |
100% |
| transformer_lens/patching.py |
make_df_from_ranges |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/patching.py |
generic_activation_patch |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/patching.py |
generic_activation_patch |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/patching.py |
generic_activation_patch |
30 |
7 |
0 |
12 |
5 |
71% |
| transformer_lens/patching.py |
generic_activation_patch.patching_hook |
3 |
0 |
0 |
2 |
1 |
80% |
| transformer_lens/patching.py |
layer_pos_patch_setter |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/patching.py |
layer_pos_head_vector_patch_setter |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/patching.py |
layer_head_vector_patch_setter |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/patching.py |
layer_head_pattern_patch_setter |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/patching.py |
layer_head_pos_pattern_patch_setter |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/patching.py |
layer_head_dest_src_pos_pattern_patch_setter |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/patching.py |
get_act_patch_attn_head_all_pos_every |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/patching.py |
get_act_patch_attn_head_by_pos_every |
10 |
10 |
0 |
0 |
0 |
0% |
| transformer_lens/patching.py |
get_act_patch_block_every |
5 |
5 |
0 |
0 |
0 |
0% |
| transformer_lens/patching.py |
(no function) |
58 |
0 |
32 |
0 |
0 |
100% |
| transformer_lens/supported_models.py |
_model_alias_map |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/supported_models.py |
get_official_model_name |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/supported_models.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/__init__.py |
__getattr__ |
3 |
1 |
0 |
2 |
1 |
60% |
| transformer_lens/tools/__init__.py |
__dir__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/__init__.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/__init__.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/_model_state.py |
require_eval_mode |
17 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/tools/analysis/_model_state.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
Node.__post_init__ |
25 |
5 |
0 |
30 |
5 |
82% |
| transformer_lens/tools/analysis/attribution_patching.py |
Node.hook_name |
11 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
EdgeAttributionConfig.__post_init__ |
4 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
AttributionResult.top_nodes |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
AttributionResult.top_edges |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
_required_hook_names |
10 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
_ensure_edge_hook_flags |
6 |
0 |
0 |
6 |
3 |
75% |
| transformer_lens/tools/analysis/attribution_patching.py |
_edge_hook_flags |
20 |
0 |
0 |
10 |
1 |
97% |
| transformer_lens/tools/analysis/attribution_patching.py |
_check_required_hooks |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
enumerate_nodes |
11 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
_assert_edges_unique |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
_required_edge_reader_hook_names |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
_edge_hook_names |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
enumerate_edges |
25 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
_as_predicate |
9 |
4 |
0 |
6 |
3 |
53% |
| transformer_lens/tools/analysis/attribution_patching.py |
cache_activation_and_gradient |
33 |
4 |
0 |
16 |
4 |
84% |
| transformer_lens/tools/analysis/attribution_patching.py |
cache_activation_and_gradient.make_fwd_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
cache_activation_and_gradient.make_fwd_hook.hook |
5 |
0 |
0 |
2 |
1 |
86% |
| transformer_lens/tools/analysis/attribution_patching.py |
cache_activation_and_gradient.make_bwd_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
cache_activation_and_gradient.make_bwd_hook.hook |
4 |
0 |
0 |
2 |
1 |
83% |
| transformer_lens/tools/analysis/attribution_patching.py |
_node_effects |
13 |
1 |
0 |
6 |
1 |
89% |
| transformer_lens/tools/analysis/attribution_patching.py |
_writer_hook_name |
7 |
1 |
0 |
6 |
1 |
85% |
| transformer_lens/tools/analysis/attribution_patching.py |
_edge_effects |
25 |
1 |
0 |
12 |
1 |
95% |
| transformer_lens/tools/analysis/attribution_patching.py |
_edge_effects.writer_delta_vec |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
_aggregate_edge_scores_to_writer_nodes |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/attribution_patching.py |
attribution_patch |
32 |
1 |
0 |
16 |
1 |
96% |
| transformer_lens/tools/analysis/attribution_patching.py |
(no function) |
52 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLensMatrixResult.logits |
9 |
1 |
0 |
6 |
1 |
87% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLensMatrixResult._retained_ranking |
8 |
2 |
0 |
6 |
2 |
71% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLensMatrixResult.top |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLensMatrixResult.bottom |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLensMatrixResult.top_tokens |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLensMatrixResult.bottom_tokens |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLensMatrixResult.target_ranks |
16 |
4 |
0 |
14 |
4 |
73% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLensMatrixResult.gradient_descent_target_ranks |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLensResult.layer |
4 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_validate_floating_matrix |
11 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_reconstruction_errors |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_to_detached_float32 |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_build_linear_gradient_factors |
22 |
1 |
0 |
10 |
1 |
94% |
| transformer_lens/tools/analysis/backward_lens.py |
_rank_vocabulary_logits |
15 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_slice_vocabulary_ranking |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_target_vocabulary_ranks |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_decode_vocabulary_ranking |
6 |
1 |
0 |
4 |
1 |
80% |
| transformer_lens/tools/analysis/backward_lens.py |
_project_residual_factors |
14 |
4 |
0 |
8 |
4 |
64% |
| transformer_lens/tools/analysis/backward_lens.py |
_factor_norms_and_normalized_rows |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_build_matrix_result |
24 |
3 |
0 |
8 |
3 |
81% |
| transformer_lens/tools/analysis/backward_lens.py |
_validate_requested_layers |
15 |
1 |
0 |
12 |
1 |
93% |
| transformer_lens/tools/analysis/backward_lens.py |
_require_raw_dense_mlp_bridge |
16 |
1 |
0 |
16 |
1 |
94% |
| transformer_lens/tools/analysis/backward_lens.py |
_get_dense_mlp_projections |
34 |
3 |
0 |
18 |
3 |
88% |
| transformer_lens/tools/analysis/backward_lens.py |
_capture_once |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_capture_once.capture |
5 |
2 |
0 |
4 |
2 |
56% |
| transformer_lens/tools/analysis/backward_lens.py |
_capture_projection_tensors |
13 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_single_batch_matrix |
8 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
_preserve_model_rng |
9 |
1 |
0 |
2 |
1 |
82% |
| transformer_lens/tools/analysis/backward_lens.py |
_capture_dense_mlp_gradient_factors |
47 |
5 |
0 |
20 |
5 |
85% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLens.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/backward_lens.py |
BackwardLens.analyze |
20 |
2 |
0 |
10 |
2 |
87% |
| transformer_lens/tools/analysis/backward_lens.py |
(no function) |
108 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/direct_logit_attribution.py |
DirectLogitAttribution.top |
5 |
0 |
0 |
2 |
1 |
86% |
| transformer_lens/tools/analysis/direct_logit_attribution.py |
_residual_stack_and_labels |
7 |
1 |
0 |
6 |
1 |
85% |
| transformer_lens/tools/analysis/direct_logit_attribution.py |
_validate_bridge_compatibility |
9 |
1 |
0 |
6 |
1 |
87% |
| transformer_lens/tools/analysis/direct_logit_attribution.py |
direct_logit_attribution |
12 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/tools/analysis/direct_logit_attribution.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/direct_path_patching.py |
_check_fold_ln |
9 |
1 |
0 |
4 |
1 |
85% |
| transformer_lens/tools/analysis/direct_path_patching.py |
_make_direct_path_hook |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/direct_path_patching.py |
_make_direct_path_hook.hook_fn |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/tools/analysis/direct_path_patching.py |
get_act_patch_direct_path |
21 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/direct_path_patching.py |
get_act_patch_direct_path._head_result |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/direct_path_patching.py |
get_act_patch_direct_path_all_sources |
8 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/direct_path_patching.py |
(no function) |
12 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_load_registry |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_resolve_registry_entry |
8 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLensReadout.top_tokens |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.__init__ |
12 |
2 |
0 |
6 |
2 |
78% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.source_layers |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.__repr__ |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.save |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.load |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens._from_checkpoint_payload |
31 |
2 |
0 |
18 |
0 |
96% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.from_pretrained |
15 |
2 |
0 |
8 |
2 |
83% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.merge |
21 |
0 |
0 |
14 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.validate_model |
21 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.clear_device_cache |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens._matrix_on |
9 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.transport |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.readout |
45 |
4 |
0 |
20 |
4 |
88% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.lens_vectors |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.lens_vector_dictionary |
17 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.decompose |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.coordinate_patch |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens._resolve_activation |
20 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.occupancy |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.fraction_of_variance |
42 |
1 |
0 |
20 |
1 |
97% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.steering_hooks |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.steering_hooks.transform |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.ablation_hooks |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.ablation_hooks.transform |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.swap_hooks |
16 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.swap_hooks.transform |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.swap_clamp_hooks |
25 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.swap_clamp_hooks.transform |
8 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.coordinate_patch_hooks |
15 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.coordinate_patch_hooks.hook_fn |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
JacobianLens.fit |
30 |
5 |
0 |
16 |
5 |
78% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_resid_post_hook_name |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_get_model_revision |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_require_raw_bridge |
30 |
0 |
0 |
20 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_validate_metadata |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_validate_metadata.validate |
13 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_normalize_positions |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_cached_on_device |
6 |
0 |
0 |
2 |
1 |
88% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_unit_rows |
5 |
1 |
0 |
2 |
1 |
71% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_make_intervention_hook |
5 |
1 |
0 |
2 |
2 |
57% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_make_intervention_hook.hook_fn |
13 |
1 |
0 |
4 |
1 |
88% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_unembed |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_to_token_ids |
16 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_validate_residual_activation |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_normalize_layer |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_frozen_parameters.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_frozen_parameters.__enter__ |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_frozen_parameters.__exit__ |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_ordinary_vjp |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_jacobian_for_prompt |
31 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_jacobian_for_prompt.capture_fn |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
_fit_transport_matrices |
19 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens.py |
(no function) |
92 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
_validate_inputs |
15 |
1 |
0 |
12 |
1 |
93% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
_validate_atom_rows |
6 |
2 |
0 |
4 |
2 |
60% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
_validate_index |
6 |
1 |
0 |
4 |
1 |
80% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
_close |
5 |
1 |
0 |
2 |
1 |
71% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
_condition_number |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
_validate_decomposition |
55 |
8 |
0 |
28 |
7 |
82% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
_basis_diagnostics |
12 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
_warn_if_near_parallel |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
_warn_if_alpha_extrapolates |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
solve_coordinate_patch |
51 |
1 |
0 |
16 |
1 |
97% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
solve_coordinate_patch_positions |
20 |
1 |
0 |
12 |
1 |
94% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
_postcondition_witnesses |
17 |
2 |
0 |
4 |
2 |
81% |
| transformer_lens/tools/analysis/jacobian_lens_coordinate_patch.py |
(no function) |
43 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens_decomposition.py |
_linalg_on_cpu_if_mps |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/tools/analysis/jacobian_lens_decomposition.py |
_diagnose_intervention_pair |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens_decomposition.py |
_nnls_tolerances |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens_decomposition.py |
_validate_nnls_kkt |
14 |
1 |
0 |
10 |
1 |
92% |
| transformer_lens/tools/analysis/jacobian_lens_decomposition.py |
_nonnegative_least_squares |
68 |
5 |
0 |
22 |
5 |
89% |
| transformer_lens/tools/analysis/jacobian_lens_decomposition.py |
_gradient_pursuit_step |
17 |
2 |
0 |
6 |
2 |
83% |
| transformer_lens/tools/analysis/jacobian_lens_decomposition.py |
get_sparse_decomposition |
51 |
0 |
0 |
26 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens_decomposition.py |
_greedy_captured_variance_gains |
17 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/jacobian_lens_decomposition.py |
estimate_occupancy |
38 |
1 |
0 |
22 |
1 |
97% |
| transformer_lens/tools/analysis/jacobian_lens_decomposition.py |
(no function) |
43 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
SubspaceBasis.ambient_dim |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
AttentionHeadRef.label |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/analysis/projection_kernel.py |
HeadAffinityResult.top_pairs |
11 |
0 |
0 |
4 |
1 |
93% |
| transformer_lens/tools/analysis/projection_kernel.py |
HeadAffinityResult.top_pairs.sort_key |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
_compute_dtype |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
_rank_tolerance_dtype |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
_validate_rtol |
10 |
1 |
0 |
6 |
1 |
88% |
| transformer_lens/tools/analysis/projection_kernel.py |
_validate_rank |
7 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
orthonormal_subspace |
24 |
3 |
0 |
14 |
3 |
84% |
| transformer_lens/tools/analysis/projection_kernel.py |
_validate_subspace |
26 |
10 |
0 |
24 |
10 |
60% |
| transformer_lens/tools/analysis/projection_kernel.py |
_clamp_projection_scores |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
_clamp_principal_cosines |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
_singular_values |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
projection_kernel |
27 |
3 |
0 |
12 |
3 |
85% |
| transformer_lens/tools/analysis/projection_kernel.py |
random_projection_kernel_moments |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
_pairwise_projection_kernel |
22 |
4 |
0 |
12 |
4 |
76% |
| transformer_lens/tools/analysis/projection_kernel.py |
_architecture_name |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
_read_role_matrix |
21 |
3 |
0 |
12 |
3 |
82% |
| transformer_lens/tools/analysis/projection_kernel.py |
_validate_role_shapes |
7 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
_extract_bases |
15 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/projection_kernel.py |
attention_head_subspace_affinity |
52 |
2 |
0 |
16 |
2 |
94% |
| transformer_lens/tools/analysis/projection_kernel.py |
(no function) |
87 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/svd_circuits.py |
HeadSVD.is_degenerate |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/svd_circuits.py |
HeadSVD.block_of |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/svd_circuits.py |
HeadSVD.degenerate_blocks |
16 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/tools/analysis/svd_circuits.py |
HeadSVD.require_isolated |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/svd_circuits.py |
_read_weight |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/svd_circuits.py |
_head_weights |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/svd_circuits.py |
_degeneracy_blocks |
17 |
1 |
0 |
6 |
1 |
91% |
| transformer_lens/tools/analysis/svd_circuits.py |
_build_rank_report |
11 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/analysis/svd_circuits.py |
_factored_head_svd |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/analysis/svd_circuits.py |
decompose_head |
22 |
0 |
0 |
14 |
0 |
100% |
| transformer_lens/tools/analysis/svd_circuits.py |
(no function) |
45 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/__init__.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/alias_drift.py |
DriftReport.has_drift |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/alias_drift.py |
DriftReport.to_dict |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/alias_drift.py |
check_drift |
16 |
1 |
0 |
4 |
1 |
90% |
| transformer_lens/tools/model_registry/alias_drift.py |
print_report |
15 |
15 |
0 |
10 |
0 |
0% |
| transformer_lens/tools/model_registry/alias_drift.py |
main |
11 |
11 |
0 |
4 |
0 |
0% |
| transformer_lens/tools/model_registry/alias_drift.py |
(no function) |
17 |
1 |
0 |
2 |
1 |
89% |
| transformer_lens/tools/model_registry/api.py |
_load_json |
8 |
3 |
0 |
2 |
1 |
60% |
| transformer_lens/tools/model_registry/api.py |
_get_supported_models_report |
8 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/model_registry/api.py |
_get_architecture_gaps_report |
8 |
0 |
0 |
2 |
1 |
90% |
| transformer_lens/tools/model_registry/api.py |
_get_verification_history |
8 |
8 |
0 |
2 |
0 |
0% |
| transformer_lens/tools/model_registry/api.py |
clear_cache |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/api.py |
get_supported_models |
7 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/model_registry/api.py |
get_unsupported_architectures |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/tools/model_registry/api.py |
is_model_supported |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/api.py |
get_model_architecture |
5 |
5 |
0 |
4 |
0 |
0% |
| transformer_lens/tools/model_registry/api.py |
get_architecture_models |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/api.py |
suggest_similar_model |
12 |
2 |
0 |
4 |
2 |
75% |
| transformer_lens/tools/model_registry/api.py |
suggest_similar_model.score_model |
13 |
3 |
0 |
12 |
4 |
72% |
| transformer_lens/tools/model_registry/api.py |
get_model_info |
6 |
1 |
0 |
4 |
1 |
80% |
| transformer_lens/tools/model_registry/api.py |
get_supported_architectures |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/api.py |
get_all_architectures_with_stats |
18 |
18 |
0 |
12 |
0 |
0% |
| transformer_lens/tools/model_registry/api.py |
is_architecture_supported |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/api.py |
get_registry_stats |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/api.py |
(no function) |
29 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/checkpoints.py |
get_checkpoint_labels |
16 |
1 |
0 |
10 |
1 |
92% |
| transformer_lens/tools/model_registry/checkpoints.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/discover_architectures.py |
discover_architectures |
79 |
79 |
0 |
28 |
0 |
0% |
| transformer_lens/tools/model_registry/discover_architectures.py |
main |
5 |
5 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/model_registry/discover_architectures.py |
(no function) |
16 |
16 |
0 |
2 |
0 |
0% |
| transformer_lens/tools/model_registry/exceptions.py |
ModelNotFoundError.__init__ |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/tools/model_registry/exceptions.py |
ArchitectureNotSupportedError.__init__ |
6 |
6 |
0 |
2 |
0 |
0% |
| transformer_lens/tools/model_registry/exceptions.py |
DataNotLoadedError.__init__ |
6 |
6 |
0 |
2 |
0 |
0% |
| transformer_lens/tools/model_registry/exceptions.py |
DataValidationError.__init__ |
8 |
8 |
0 |
4 |
0 |
0% |
| transformer_lens/tools/model_registry/exceptions.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/generate_report.py |
get_architecture_description |
18 |
18 |
0 |
12 |
0 |
0% |
| transformer_lens/tools/model_registry/generate_report.py |
generate_report |
76 |
76 |
0 |
18 |
0 |
0% |
| transformer_lens/tools/model_registry/generate_report.py |
main |
9 |
9 |
0 |
2 |
0 |
0% |
| transformer_lens/tools/model_registry/generate_report.py |
(no function) |
10 |
1 |
0 |
2 |
1 |
83% |
| transformer_lens/tools/model_registry/hf_scraper.py |
_extract_architecture |
6 |
6 |
0 |
4 |
0 |
0% |
| transformer_lens/tools/model_registry/hf_scraper.py |
_extract_param_count |
26 |
26 |
0 |
14 |
0 |
0% |
| transformer_lens/tools/model_registry/hf_scraper.py |
_load_existing_models |
15 |
15 |
0 |
6 |
0 |
0% |
| transformer_lens/tools/model_registry/hf_scraper.py |
_load_existing_gaps |
15 |
15 |
0 |
8 |
0 |
0% |
| transformer_lens/tools/model_registry/hf_scraper.py |
_extract_profile_signals |
8 |
8 |
0 |
2 |
0 |
0% |
| transformer_lens/tools/model_registry/hf_scraper.py |
_build_model_entry |
13 |
13 |
0 |
6 |
0 |
0% |
| transformer_lens/tools/model_registry/hf_scraper.py |
_canonical_author_sweep |
37 |
37 |
6 |
26 |
0 |
0% |
| transformer_lens/tools/model_registry/hf_scraper.py |
scrape_all_models |
222 |
222 |
0 |
88 |
0 |
0% |
| transformer_lens/tools/model_registry/hf_scraper.py |
_save_checkpoint |
3 |
3 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/model_registry/hf_scraper.py |
main |
13 |
13 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/model_registry/hf_scraper.py |
(no function) |
24 |
24 |
0 |
2 |
0 |
0% |
| transformer_lens/tools/model_registry/registry_io.py |
is_incompatible_quantized |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
is_hf_loadable_quantized |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
required_quant_library_for_model |
4 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
is_quantized_model |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
load_model_aliases |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
resolve_model_alias |
4 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
load_supported_models_raw |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
save_supported_models_raw |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
load_verification_history |
5 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
save_verification_history |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
_get_tl_version |
9 |
3 |
0 |
2 |
1 |
64% |
| transformer_lens/tools/model_registry/registry_io.py |
pass_status |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
extract_phase_scores |
15 |
0 |
0 |
14 |
1 |
97% |
| transformer_lens/tools/model_registry/registry_io.py |
recompute_registry_totals |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
update_model_status |
58 |
5 |
0 |
42 |
2 |
89% |
| transformer_lens/tools/model_registry/registry_io.py |
registry_prompt_profile |
9 |
9 |
0 |
4 |
0 |
0% |
| transformer_lens/tools/model_registry/registry_io.py |
add_verification_record |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/model_registry/registry_io.py |
(no function) |
43 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/relevancy.py |
_normalize_demand |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/model_registry/relevancy.py |
_normalize_usage |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/model_registry/relevancy.py |
_score_benchmarkability |
13 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/tools/model_registry/relevancy.py |
compute_relevancy_score |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/relevancy.py |
compute_scores_for_gaps |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/model_registry/relevancy.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/schemas.py |
ModelMetadata.to_dict |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/schemas.py |
ModelMetadata.from_dict |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/model_registry/schemas.py |
ModelEntry.to_dict |
13 |
7 |
0 |
8 |
3 |
43% |
| transformer_lens/tools/model_registry/schemas.py |
ModelEntry.from_dict |
12 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/tools/model_registry/schemas.py |
ArchitectureGap.to_dict |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/schemas.py |
ArchitectureGap.from_dict |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/schemas.py |
ScanInfo.to_dict |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/tools/model_registry/schemas.py |
ScanInfo.from_dict |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/schemas.py |
SupportedModelsReport.to_dict |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/schemas.py |
SupportedModelsReport.from_dict |
4 |
0 |
0 |
2 |
1 |
83% |
| transformer_lens/tools/model_registry/schemas.py |
ArchitectureGapsReport.to_dict |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/schemas.py |
ArchitectureGapsReport.from_dict |
5 |
0 |
0 |
2 |
1 |
86% |
| transformer_lens/tools/model_registry/schemas.py |
ArchitectureStats.to_dict |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/model_registry/schemas.py |
ArchitectureAnalysis.to_dict |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/model_registry/schemas.py |
(no function) |
90 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/validate.py |
ValidationError.__str__ |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/tools/model_registry/validate.py |
ValidationResult.error_count |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/validate.py |
_validate_string |
9 |
3 |
0 |
8 |
2 |
59% |
| transformer_lens/tools/model_registry/validate.py |
_validate_int |
9 |
2 |
0 |
8 |
3 |
71% |
| transformer_lens/tools/model_registry/validate.py |
_validate_bool |
7 |
3 |
0 |
6 |
2 |
46% |
| transformer_lens/tools/model_registry/validate.py |
_validate_date_string |
11 |
3 |
0 |
6 |
2 |
71% |
| transformer_lens/tools/model_registry/validate.py |
_validate_datetime_string |
11 |
5 |
0 |
6 |
2 |
47% |
| transformer_lens/tools/model_registry/validate.py |
_validate_list |
7 |
3 |
0 |
6 |
2 |
46% |
| transformer_lens/tools/model_registry/validate.py |
_validate_model_metadata |
17 |
17 |
0 |
16 |
0 |
0% |
| transformer_lens/tools/model_registry/validate.py |
_validate_model_entry |
36 |
15 |
0 |
30 |
6 |
59% |
| transformer_lens/tools/model_registry/validate.py |
_validate_architecture_gap |
6 |
1 |
0 |
2 |
1 |
75% |
| transformer_lens/tools/model_registry/validate.py |
_validate_verification_record |
18 |
2 |
0 |
14 |
7 |
72% |
| transformer_lens/tools/model_registry/validate.py |
validate_supported_models_report |
15 |
2 |
0 |
8 |
3 |
78% |
| transformer_lens/tools/model_registry/validate.py |
validate_architecture_gaps_report |
12 |
1 |
0 |
6 |
2 |
83% |
| transformer_lens/tools/model_registry/validate.py |
validate_verification_history |
11 |
1 |
0 |
8 |
3 |
79% |
| transformer_lens/tools/model_registry/validate.py |
validate_json_schema |
19 |
9 |
0 |
14 |
2 |
48% |
| transformer_lens/tools/model_registry/validate.py |
validate_data_directory |
23 |
23 |
0 |
8 |
0 |
0% |
| transformer_lens/tools/model_registry/validate.py |
(no function) |
35 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/verification.py |
VerificationRecord.to_dict |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/verification.py |
VerificationRecord.from_dict |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/verification.py |
VerificationHistory.to_dict |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/verification.py |
VerificationHistory.from_dict |
4 |
0 |
0 |
2 |
1 |
83% |
| transformer_lens/tools/model_registry/verification.py |
VerificationHistory.get_record |
4 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/model_registry/verification.py |
VerificationHistory.is_verified |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/verification.py |
VerificationHistory.add_record |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/verification.py |
VerificationHistory.invalidate |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/tools/model_registry/verification.py |
(no function) |
29 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/verify_models.py |
_handle_sigint |
6 |
6 |
0 |
2 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
_sanitize_note |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/model_registry/verify_models.py |
_phases_to_run |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/verify_models.py |
_full_and_core_phases |
11 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/tools/model_registry/verify_models.py |
_default_phases_for_architecture |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/verify_models.py |
_get_current_model_status |
7 |
7 |
0 |
6 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
VerificationProgress.to_dict |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
VerificationProgress.from_dict |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
published_param_count |
8 |
2 |
0 |
0 |
0 |
75% |
| transformer_lens/tools/model_registry/verify_models.py |
estimate_model_params |
50 |
19 |
0 |
24 |
8 |
55% |
| transformer_lens/tools/model_registry/verify_models.py |
estimate_benchmark_memory_gb |
19 |
3 |
0 |
12 |
4 |
77% |
| transformer_lens/tools/model_registry/verify_models.py |
get_available_memory_gb |
17 |
17 |
0 |
6 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
select_models_for_verification |
36 |
36 |
0 |
24 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
_extract_prompt_profile |
6 |
0 |
0 |
6 |
1 |
92% |
| transformer_lens/tools/model_registry/verify_models.py |
_measured_nothing |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/tools/model_registry/verify_models.py |
_check_phase_scores |
27 |
5 |
0 |
20 |
3 |
79% |
| transformer_lens/tools/model_registry/verify_models.py |
_build_verified_note |
29 |
11 |
0 |
20 |
7 |
59% |
| transformer_lens/tools/model_registry/verify_models.py |
_preserved_issue_suffix |
17 |
3 |
0 |
10 |
1 |
85% |
| transformer_lens/tools/model_registry/verify_models.py |
_p1_only_core_note |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/tools/model_registry/verify_models.py |
_clear_hf_cache |
19 |
4 |
0 |
10 |
2 |
79% |
| transformer_lens/tools/model_registry/verify_models.py |
_save_checkpoint |
3 |
3 |
0 |
0 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
_skip_model |
8 |
8 |
0 |
6 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
_load_checkpoint |
8 |
8 |
0 |
2 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
verify_models |
242 |
242 |
0 |
128 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
_print_dry_run |
27 |
27 |
0 |
8 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
_print_summary |
27 |
27 |
0 |
18 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
main |
71 |
71 |
0 |
32 |
0 |
0% |
| transformer_lens/tools/model_registry/verify_models.py |
(no function) |
70 |
1 |
0 |
2 |
1 |
97% |
| transformer_lens/tools/training.py |
train |
47 |
17 |
0 |
32 |
9 |
59% |
| transformer_lens/tools/training.py |
(no function) |
30 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/__init__.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/activation_functions.py |
gelu_new |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/activation_functions.py |
gelu_fast |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/activation_functions.py |
gelu_pytorch_tanh |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/activation_functions.py |
solu |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/activation_functions.py |
XIELU.__init__ |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/activation_functions.py |
XIELU.forward |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/activation_functions.py |
xielu |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/activation_functions.py |
relu2 |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/activation_functions.py |
softcap_enabled |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/activation_functions.py |
apply_softcap |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/activation_functions.py |
(no function) |
20 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/addmm.py |
vanilla_addmm |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/addmm.py |
batch_addmm |
5 |
5 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/addmm.py |
(no function) |
4 |
4 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/aliases.py |
resolve_alias |
15 |
7 |
0 |
8 |
1 |
57% |
| transformer_lens/utilities/aliases.py |
resolve_alias._resolve_single_target |
14 |
1 |
0 |
10 |
1 |
92% |
| transformer_lens/utilities/aliases.py |
_collect_aliases_from_module |
21 |
0 |
0 |
18 |
0 |
100% |
| transformer_lens/utilities/aliases.py |
collect_aliases_recursive |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/aliases.py |
(no function) |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/architectures.py |
classify_architecture |
11 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/utilities/architectures.py |
get_architectures_for_config |
6 |
1 |
0 |
4 |
2 |
70% |
| transformer_lens/utilities/architectures.py |
classify_model_config |
7 |
2 |
0 |
6 |
3 |
62% |
| transformer_lens/utilities/architectures.py |
classify_model_name |
9 |
2 |
0 |
2 |
1 |
73% |
| transformer_lens/utilities/architectures.py |
is_masked_lm_model |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/architectures.py |
is_encoder_decoder_model |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/architectures.py |
is_multimodal_model |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/architectures.py |
is_audio_model |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/architectures.py |
(no function) |
20 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/attention.py |
simple_attn_linear |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/utilities/attention.py |
complex_attn_linear |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/attention.py |
clamp_qkv |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/attention.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/attn_implementation.py |
force_eager_attention |
18 |
0 |
0 |
14 |
0 |
100% |
| transformer_lens/utilities/attn_implementation.py |
_layer_candidates |
9 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/attn_implementation.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/attribute_utils.py |
get_nested_attr |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/utilities/attribute_utils.py |
set_nested_attr |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/utilities/attribute_utils.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/bridge_components.py |
collect_all_submodules_of_component |
8 |
1 |
0 |
8 |
1 |
88% |
| transformer_lens/utilities/bridge_components.py |
collect_components_of_block_bridge |
16 |
4 |
0 |
8 |
3 |
71% |
| transformer_lens/utilities/bridge_components.py |
collect_all_components |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/bridge_components.py |
apply_fn_to_all_components |
6 |
0 |
0 |
4 |
1 |
90% |
| transformer_lens/utilities/bridge_components.py |
(no function) |
8 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/cache.py |
get_pos_offset |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/cache.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/components_utils.py |
get_act_name |
25 |
4 |
0 |
18 |
4 |
81% |
| transformer_lens/utilities/components_utils.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/defaults_utils.py |
override_or_use_default_value |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/defaults_utils.py |
LocallyOverridenDefaults.__init__ |
5 |
5 |
0 |
2 |
0 |
0% |
| transformer_lens/utilities/defaults_utils.py |
LocallyOverridenDefaults.__enter__ |
11 |
11 |
0 |
4 |
0 |
0% |
| transformer_lens/utilities/defaults_utils.py |
LocallyOverridenDefaults.__exit__ |
7 |
7 |
0 |
4 |
0 |
0% |
| transformer_lens/utilities/defaults_utils.py |
(no function) |
10 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/devices.py |
_torch_version_tuple |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/devices.py |
_torch_mps_has_known_broken_bug |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/devices.py |
get_device |
7 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/utilities/devices.py |
warn_if_mps |
14 |
0 |
0 |
12 |
0 |
100% |
| transformer_lens/utilities/devices.py |
ModelWithCfg.state_dict |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/utilities/devices.py |
ModelWithCfg.to |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/utilities/devices.py |
move_to_and_update_config |
19 |
0 |
0 |
16 |
3 |
91% |
| transformer_lens/utilities/devices.py |
(no function) |
21 |
0 |
3 |
0 |
0 |
100% |
| transformer_lens/utilities/exploratory_utils.py |
test_prompt |
38 |
38 |
0 |
16 |
0 |
0% |
| transformer_lens/utilities/exploratory_utils.py |
(no function) |
10 |
2 |
0 |
0 |
0 |
80% |
| transformer_lens/utilities/gpu_utils.py |
print_gpu_mem |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/gpu_utils.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/heterogeneous_config.py |
per_layer_attr_names |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/heterogeneous_config.py |
per_layer_values |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/heterogeneous_config.py |
majority_value |
11 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/utilities/heterogeneous_config.py |
safe_config_get |
4 |
2 |
0 |
2 |
1 |
50% |
| transformer_lens/utilities/heterogeneous_config.py |
HetSafeConfigView.__init__ |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/heterogeneous_config.py |
HetSafeConfigView.__getattr__ |
7 |
1 |
0 |
4 |
1 |
82% |
| transformer_lens/utilities/heterogeneous_config.py |
het_safe_view |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/heterogeneous_config.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/hf_utils.py |
_is_hf_rate_limit_error |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/hf_utils.py |
_retry_after_seconds |
11 |
1 |
0 |
4 |
1 |
87% |
| transformer_lens/utilities/hf_utils.py |
enable_hf_retry |
9 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/hf_utils.py |
enable_hf_retry._wrapped |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/hf_utils.py |
call_hf_with_retry |
13 |
1 |
0 |
6 |
1 |
89% |
| transformer_lens/utilities/hf_utils.py |
get_hf_token |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/hf_utils.py |
get_rotary_pct_from_config |
11 |
1 |
0 |
8 |
1 |
89% |
| transformer_lens/utilities/hf_utils.py |
select_compatible_kwargs |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/hf_utils.py |
download_file_from_hf |
7 |
3 |
0 |
4 |
2 |
55% |
| transformer_lens/utilities/hf_utils.py |
clear_huggingface_cache |
11 |
11 |
0 |
4 |
0 |
0% |
| transformer_lens/utilities/hf_utils.py |
clear_huggingface_cache.handle_remove_readonly |
14 |
14 |
0 |
8 |
0 |
0% |
| transformer_lens/utilities/hf_utils.py |
keep_single_column |
4 |
1 |
0 |
4 |
1 |
75% |
| transformer_lens/utilities/hf_utils.py |
get_dataset |
5 |
5 |
0 |
2 |
0 |
0% |
| transformer_lens/utilities/hf_utils.py |
autoconfig_with_remote_post_init_compat |
16 |
8 |
0 |
8 |
1 |
46% |
| transformer_lens/utilities/hf_utils.py |
_strict_dataclass_error |
3 |
0 |
4 |
0 |
0 |
100% |
| transformer_lens/utilities/hf_utils.py |
autotokenizer_with_special_token_compat |
12 |
1 |
0 |
6 |
1 |
89% |
| transformer_lens/utilities/hf_utils.py |
_fast_tokenizer_from_file |
23 |
1 |
0 |
12 |
2 |
91% |
| transformer_lens/utilities/hf_utils.py |
enable_hf_numeric_tower |
9 |
0 |
2 |
2 |
0 |
100% |
| transformer_lens/utilities/hf_utils.py |
enable_hf_numeric_tower.tolerant |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/hf_utils.py |
_resolve_remote_config_class |
8 |
8 |
0 |
2 |
0 |
0% |
| transformer_lens/utilities/hf_utils.py |
make_post_init_kwarg_tolerant |
7 |
0 |
0 |
2 |
1 |
89% |
| transformer_lens/utilities/hf_utils.py |
make_post_init_kwarg_tolerant.tolerant_post_init |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/hf_utils.py |
(no function) |
43 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/initialization_utils.py |
calc_fan_in_and_fan_out |
14 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/utilities/initialization_utils.py |
init_xavier_uniform_ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/initialization_utils.py |
init_xavier_normal_ |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/initialization_utils.py |
init_kaiming_uniform_ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/initialization_utils.py |
init_kaiming_normal_ |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/initialization_utils.py |
(no function) |
11 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/library_utils.py |
is_library_available |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/library_utils.py |
(no function) |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/lm_utils.py |
lm_cross_entropy_loss |
11 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/lm_utils.py |
lm_accuracy |
5 |
1 |
0 |
2 |
1 |
71% |
| transformer_lens/utilities/lm_utils.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/logits_utils.py |
logits_to_df |
14 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/logits_utils.py |
_apply_repetition_penalty |
6 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/logits_utils.py |
sample_logits |
29 |
1 |
0 |
14 |
1 |
95% |
| transformer_lens/utilities/logits_utils.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/matrix.py |
composition_scores |
14 |
0 |
0 |
6 |
1 |
95% |
| transformer_lens/utilities/matrix.py |
get_matrix_corner |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/matrix.py |
(no function) |
7 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/multi_gpu.py |
calculate_available_device_cuda_memory |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/multi_gpu.py |
_get_available_cuda_memory_for_device |
5 |
2 |
0 |
0 |
0 |
60% |
| transformer_lens/utilities/multi_gpu.py |
determine_available_memory_for_available_devices |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/multi_gpu.py |
sort_devices_based_on_available_memory |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/multi_gpu.py |
get_best_available_cuda_device |
6 |
6 |
0 |
2 |
0 |
0% |
| transformer_lens/utilities/multi_gpu.py |
get_best_available_device |
5 |
5 |
0 |
2 |
0 |
0% |
| transformer_lens/utilities/multi_gpu.py |
get_device_for_block_index |
9 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/multi_gpu.py |
resolve_device_map |
13 |
1 |
0 |
10 |
1 |
91% |
| transformer_lens/utilities/multi_gpu.py |
_validate_device_map_values |
4 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/multi_gpu.py |
is_mixed_offload_gpu |
11 |
0 |
0 |
10 |
1 |
95% |
| transformer_lens/utilities/multi_gpu.py |
cast_floating_params_to_dtype |
11 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/utilities/multi_gpu.py |
maybe_cast_floating_params |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/multi_gpu.py |
find_embedding_device |
19 |
4 |
0 |
8 |
0 |
85% |
| transformer_lens/utilities/multi_gpu.py |
count_unique_devices |
4 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/multi_gpu.py |
find_misplaced_modules |
19 |
2 |
0 |
8 |
0 |
93% |
| transformer_lens/utilities/multi_gpu.py |
(no function) |
25 |
0 |
3 |
0 |
0 |
100% |
| transformer_lens/utilities/parameter_swap.py |
temporarily_swap_parameter |
15 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/utilities/parameter_swap.py |
(no function) |
6 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/quantization.py |
unreadable_weight_reason |
10 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/utilities/quantization.py |
_is_meta |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/quantization.py |
quantization_method |
12 |
2 |
0 |
8 |
2 |
80% |
| transformer_lens/utilities/quantization.py |
describe_quantization |
10 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/utilities/quantization.py |
require_readable_weight |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/quantization.py |
(no function) |
9 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/slice.py |
Slice.__init__ |
16 |
3 |
0 |
10 |
2 |
81% |
| transformer_lens/utilities/slice.py |
Slice.apply |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/slice.py |
Slice.indices |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/slice.py |
Slice.__repr__ |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/utilities/slice.py |
Slice.unwrap |
5 |
0 |
0 |
4 |
1 |
89% |
| transformer_lens/utilities/slice.py |
(no function) |
15 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tensors.py |
to_numpy |
13 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/utilities/tensors.py |
get_corner |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tensors.py |
remove_batch_dim |
3 |
3 |
0 |
2 |
0 |
0% |
| transformer_lens/utilities/tensors.py |
transpose |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tensors.py |
is_square |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tensors.py |
is_lower_triangular |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/tensors.py |
check_structure |
33 |
33 |
0 |
22 |
0 |
0% |
| transformer_lens/utilities/tensors.py |
get_offset_position_ids |
3 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tensors.py |
get_cumsum_along_dim |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/tensors.py |
repeat_along_head_dimension |
4 |
4 |
0 |
2 |
0 |
0% |
| transformer_lens/utilities/tensors.py |
filter_dict_by_prefix |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tensors.py |
(no function) |
17 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
_validate_shape |
2 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
_kv_heads |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
_convert_w_q |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
_convert_w_kv |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
_convert_w_o |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
_convert_b_q |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
_convert_b_kv |
2 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
_identity |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
_transpose |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
_convert_key_and_tensor |
5 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
convert_tl_checkpoint |
10 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/utilities/tl_checkpoint_conversion.py |
(no function) |
19 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tokenize_utils.py |
tokenize_and_concatenate |
18 |
0 |
0 |
8 |
2 |
92% |
| transformer_lens/utilities/tokenize_utils.py |
tokenize_and_concatenate.tokenize_function |
28 |
2 |
0 |
14 |
3 |
88% |
| transformer_lens/utilities/tokenize_utils.py |
get_tokenizer_with_bos |
16 |
0 |
0 |
6 |
0 |
100% |
| transformer_lens/utilities/tokenize_utils.py |
get_input_with_manually_prepended_bos |
6 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/tokenize_utils.py |
get_tokens_with_bos_removed |
15 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/utilities/tokenize_utils.py |
get_attention_mask |
15 |
1 |
0 |
8 |
1 |
91% |
| transformer_lens/utilities/tokenize_utils.py |
(no function) |
17 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tracr.py |
infer_tracr_output_label |
12 |
2 |
0 |
6 |
2 |
78% |
| transformer_lens/utilities/tracr.py |
make_tracr_categorical_unembed |
10 |
1 |
0 |
4 |
1 |
86% |
| transformer_lens/utilities/tracr.py |
make_tracr_transformer_bridge_config |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tracr.py |
make_tracr_transformer_bridge_state_dict |
8 |
0 |
0 |
4 |
0 |
100% |
| transformer_lens/utilities/tracr.py |
_categorical_output_encoding_map |
5 |
1 |
0 |
2 |
1 |
71% |
| transformer_lens/utilities/tracr.py |
_residual_labels |
4 |
1 |
0 |
2 |
1 |
67% |
| transformer_lens/utilities/tracr.py |
_param_array |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tracr.py |
_tensor |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/tracr.py |
(no function) |
14 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/utilities/typed_module_list.py |
(no function) |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights._get_param_key |
3 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights._prepare_component_path |
8 |
2 |
0 |
4 |
2 |
67% |
| transformer_lens/weight_processing.py |
ProcessWeights._resolve_state_dict_key |
21 |
2 |
0 |
18 |
4 |
85% |
| transformer_lens/weight_processing.py |
ProcessWeights._safe_get_tensor |
2 |
2 |
0 |
0 |
0 |
0% |
| transformer_lens/weight_processing.py |
ProcessWeights.fold_layer_norm_bias_single |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.fold_layer_norm_weight_single |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.center_weight_single |
1 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.fold_layer_norm_biases |
5 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.fold_layer_norm_biases._zero_bias |
1 |
1 |
0 |
0 |
0 |
0% |
| transformer_lens/weight_processing.py |
ProcessWeights.fold_layer_norm_weights |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.center_attention_weights |
4 |
0 |
0 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.extract_attention_tensors_for_folding |
40 |
0 |
0 |
14 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.extract_attention_tensors_for_folding._reshape_bias_if_needed |
7 |
0 |
0 |
6 |
1 |
92% |
| transformer_lens/weight_processing.py |
ProcessWeights._fold_short_conv_layer_norm |
23 |
10 |
0 |
10 |
3 |
48% |
| transformer_lens/weight_processing.py |
ProcessWeights._fold_layer |
64 |
8 |
0 |
34 |
9 |
81% |
| transformer_lens/weight_processing.py |
ProcessWeights._fold_mlp_layer_norm |
123 |
13 |
0 |
68 |
18 |
84% |
| transformer_lens/weight_processing.py |
ProcessWeights._store_processed_attention_tensors |
20 |
2 |
0 |
10 |
2 |
87% |
| transformer_lens/weight_processing.py |
ProcessWeights._fold_unembed_layer_norm |
37 |
4 |
0 |
18 |
5 |
84% |
| transformer_lens/weight_processing.py |
ProcessWeights._fold_final_rms_bias |
26 |
3 |
0 |
12 |
4 |
82% |
| transformer_lens/weight_processing.py |
ProcessWeights.fold_layer_norm |
7 |
0 |
0 |
2 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.center_writing_weights |
94 |
30 |
0 |
44 |
11 |
70% |
| transformer_lens/weight_processing.py |
ProcessWeights.center_unembed |
23 |
0 |
0 |
10 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.fold_value_biases |
92 |
53 |
0 |
42 |
7 |
43% |
| transformer_lens/weight_processing.py |
ProcessWeights.process_weights |
39 |
7 |
0 |
34 |
5 |
81% |
| transformer_lens/weight_processing.py |
ProcessWeights.refactor_factored_attn_matrices |
51 |
0 |
0 |
8 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.convert_tensor_to_tl_format |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.convert_tensor_to_tl_format |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.convert_tensor_to_tl_format |
36 |
8 |
0 |
28 |
4 |
75% |
| transformer_lens/weight_processing.py |
ProcessWeights.convert_tensor_to_hf_format |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.convert_tensor_to_hf_format |
0 |
0 |
1 |
0 |
0 |
100% |
| transformer_lens/weight_processing.py |
ProcessWeights.convert_tensor_to_hf_format |
32 |
9 |
0 |
24 |
4 |
70% |
| transformer_lens/weight_processing.py |
ProcessWeights.distribute_weights_to_components |
43 |
22 |
0 |
30 |
8 |
51% |
| transformer_lens/weight_processing.py |
(no function) |
63 |
0 |
42 |
0 |
0 |
100% |