Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | W _ __add__() (tdhook.hooks.MultiHookHandle method) __call__() (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper method) (tdhook.attribution.lrp_helpers.rules.ParamModifier method) (tdhook.metrics.InfidelityMetric method) (tdhook.metrics.SensitivityMetric method) __enter__() (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle method) (tdhook.contexts.HookingContext method) (tdhook.contexts.HookingContextWithCache method) (tdhook.hooks.MultiHookHandle method) (tdhook.modules.HookedModuleRun method) (tdhook.weights.pruning.PruningContext method) __exit__() (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle method) (tdhook.contexts.HookingContext method) (tdhook.hooks.MultiHookHandle method) (tdhook.modules.HookedModuleRun method) (tdhook.weights.pruning.PruningContext method) __instancecheck__() (tdhook._types.UnraveledMeta method) (tdhook.attribution.lrp_helpers.types.SubclassMeta method) __repr__() (tdhook.modules.FunctionModule method) (tdhook.modules.HookedModule method) (tdhook.modules.IntermediateKeysCleaner method) (tdhook.modules.ModuleCall method) (tdhook.modules.ModuleCallWithCache method) (tdhook.modules.PGDModule method) __subclass__ (tdhook.attribution.lrp_helpers.types.Activation attribute) (tdhook.attribution.lrp_helpers.types.AvgPool attribute) (tdhook.attribution.lrp_helpers.types.BatchNorm attribute) (tdhook.attribution.lrp_helpers.types.Convolution attribute) (tdhook.attribution.lrp_helpers.types.ConvolutionStandard attribute) (tdhook.attribution.lrp_helpers.types.ConvolutionTranspose attribute) (tdhook.attribution.lrp_helpers.types.Linear attribute) (tdhook.attribution.lrp_helpers.types.MaxPool attribute) __subclasscheck__() (tdhook.attribution.lrp_helpers.types.SubclassMeta method) _absolute (tdhook.attribution.grad_cam.GradCAM attribute) (tdhook.attribution.GradCAM attribute) (tdhook.attribution.Saliency attribute) (tdhook.attribution.saliency.Saliency attribute) _adapters (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _additional_init_keys (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _additional_keys (tdhook.latent.Probing attribute) (tdhook.latent.probing.Probing attribute) _allow_overwrite (tdhook.latent.probing.SklearnProbeManager attribute) _alpha (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.modules.PGDModule attribute) _alphas (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) _amount_to_prune (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _apply_kwargs (tdhook.attribution.lrp_helpers.rules.Rule attribute) _ascent (tdhook.modules.PGDModule attribute) _attr_key (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _attribution_key (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) _attributor_fn() (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) _baseline_key (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline attribute) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline attribute) _cache (tdhook.contexts.HookingContextWithCache attribute) (tdhook.hooks.CacheProxy attribute) (tdhook.modules.HookedModuleRun attribute) _cache_as_output (tdhook.modules.ModuleCallWithCache attribute) _cache_callback (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationAddition attribute) (tdhook.latent.ActivationPatching attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _cache_key (tdhook.modules.ModuleCallWithCache attribute) _cache_ref (tdhook.modules.ModuleCallWithCache attribute) _call() (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper method) (tdhook.attribution.lrp_helpers.rules.EpsilonPlus method) _callback (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) _check_callback_signature() (tdhook.hooks.HookFactory static method) _check_hook_signature() (in module tdhook.hooks) _classes_to_hook (tdhook.hooks.MultiHookManager attribute) _classes_to_skip (tdhook.hooks.MultiHookManager attribute) _clean_intermediate_keys (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationAddition attribute) (tdhook.latent.ActivationPatching attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _clear_cache (tdhook.contexts.HookingContextWithCache attribute) _coef (tdhook.latent.probing.MeanDifferenceClassifier attribute) _compute_convergence_delta (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline attribute) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline attribute) _compute_convergence_delta_fn() (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) _compute_metrics (tdhook.latent.probing.SklearnProbeManager attribute) _compute_steering_vectors() (tdhook.latent.ActivationAddition method) (tdhook.latent.steering_vectors.ActivationAddition method) _contexts (tdhook.contexts.CompositeHookingContextFactory attribute) _data (tdhook.modules.HookedModuleRun attribute) _data_preprocess_callback (tdhook.latent.probing.SklearnProbe attribute) (tdhook.latent.probing.SklearnProbeManager attribute) _decision_function() (tdhook.latent.probing.MeanDifferenceClassifier method) _default_data_preprocess_callback() (tdhook.latent.probing.SklearnProbe method) _directions (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.Probing attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _ensure_in_context() (tdhook.modules.HookedModuleRun method) _fit_callback (tdhook.latent.probing.SklearnProbe attribute) _fit_metrics (tdhook.latent.probing.SklearnProbeManager attribute) _flatten (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) _grad_attr() (tdhook.attribution.grad_cam.GradCAM method) (tdhook.attribution.GradCAM method) (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) (tdhook.attribution.guided_backpropagation.GuidedBackpropagation method) (tdhook.attribution.GuidedBackpropagation method) (tdhook.attribution.integrated_gradients.IntegratedGradients method) (tdhook.attribution.IntegratedGradients method) (tdhook.attribution.LRP method) (tdhook.attribution.lrp.LRP method) (tdhook.attribution.Saliency method) (tdhook.attribution.saliency.Saliency method) _grad_enabled (tdhook.modules.HookedModuleRun attribute) _grad_key (tdhook.modules.PGDModule attribute) _handle (tdhook.contexts.HookingContext attribute) _handles (tdhook.hooks.MultiHookHandle attribute) (tdhook.modules.HookedModuleRun attribute) _has_torchvision (in module tdhook.attribution.lrp_helpers.types) _hook (tdhook.contexts.HookingContext attribute) _hook_manager (tdhook.attribution.guided_backpropagation.GuidedBackpropagation attribute) (tdhook.attribution.GuidedBackpropagation attribute) (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.Probing attribute) _hook_module() (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) (tdhook.attribution.guided_backpropagation.GuidedBackpropagation method) (tdhook.attribution.GuidedBackpropagation method) (tdhook.contexts.CompositeHookingContextFactory method) (tdhook.contexts.HookingContextFactory method) (tdhook.latent.activation_caching.ActivationCaching method) (tdhook.latent.activation_patching.ActivationPatching method) (tdhook.latent.ActivationAddition method) (tdhook.latent.ActivationCaching method) (tdhook.latent.ActivationPatching method) (tdhook.latent.Probing method) (tdhook.latent.probing.Probing method) (tdhook.latent.steering_vectors.ActivationAddition method) (tdhook.latent.steering_vectors.SteeringVectors method) (tdhook.latent.SteeringVectors method) (tdhook.weights.Adapters method) (tdhook.weights.adapters.Adapters method) _hooked_module (tdhook.contexts.HookingContext attribute) _hooked_module_class (tdhook.contexts.HookingContextFactory attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) (tdhook.weights.task_vectors.TaskVectors attribute) (tdhook.weights.TaskVectors attribute) _hooked_module_kwargs (tdhook.contexts.HookingContextFactory attribute) _hooking_context (tdhook.modules.HookedModule attribute) _hooking_context_class (tdhook.contexts.HookingContextFactory attribute) (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) (tdhook.weights.task_vectors.TaskVectors attribute) (tdhook.weights.TaskVectors attribute) _hooking_context_kwargs (tdhook.contexts.HookingContextFactory attribute) _importance_callback (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _in_context (tdhook.contexts.HookingContext attribute) (tdhook.modules.HookedModuleRun attribute) _in_key (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) _in_keys (tdhook.contexts.HookingContext attribute) _init_attr_cache_in (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _init_attr_grads (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _init_attr_inputs (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _init_attr_targets (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _input_modules (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _intercept (tdhook.latent.probing.MeanDifferenceClassifier attribute) _intermediate_keys (tdhook.modules.IntermediateKeysCleaner attribute) _key (tdhook.hooks.CacheProxy attribute) (tdhook.hooks.EarlyStoppingException attribute) _key_pattern (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.Probing attribute) _max_value (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.modules.PGDModule attribute) _method (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) _min_value (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.modules.PGDModule attribute) _modify_fn (tdhook.attribution.lrp_helpers.rules.ParamModifier attribute) _module (tdhook.modules.HookedModuleRun attribute) _module_ref (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle attribute) _modules_to_attribute (tdhook.attribution.grad_cam.GradCAM attribute) (tdhook.attribution.GradCAM attribute) _modules_to_maximise (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) _modules_to_patch (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationPatching attribute) _modules_to_prune (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _modules_to_steer (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) (tdhook.latent.steering_vectors.SteeringVectors attribute) (tdhook.latent.SteeringVectors attribute) _multiply_by_inputs (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline attribute) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline attribute) (tdhook.attribution.guided_backpropagation.GuidedBackpropagation attribute) (tdhook.attribution.GuidedBackpropagation attribute) (tdhook.attribution.Saliency attribute) (tdhook.attribution.saliency.Saliency attribute) _multiply_by_inputs_fn() (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) _n_steps (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) (tdhook.modules.PGDModule attribute) _name (tdhook.modules.HookedModuleRun attribute) _negative_key (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _normalize (tdhook.latent.probing.MeanDifferenceClassifier attribute) _old_weights (tdhook.weights.pruning.PruningContext attribute) _out_key (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) _out_keys (tdhook.contexts.HookingContext attribute) _outer_cache (tdhook.modules.HookedModuleRun attribute) _output_grad_callbacks (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _patch_fn (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationPatching attribute) _patch_key (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationPatching attribute) _pattern (tdhook.hooks.MultiHookManager attribute) _perturb_data() (tdhook.metrics.InfidelityMetric method) (tdhook.metrics.SensitivityMetric method) _pgd_step() (tdhook.modules.PGDModule method) _positive_key (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _pre_factories (tdhook.contexts.HookingContext attribute) _predict_callback (tdhook.latent.probing.SklearnProbe attribute) _predict_metrics (tdhook.latent.probing.SklearnProbeManager attribute) _prepare (tdhook.contexts.HookingContext attribute) _prepare_module() (tdhook.attribution.activation_maximisation.ActivationMaximisation method) (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) (tdhook.attribution.LRP method) (tdhook.attribution.lrp.LRP method) (tdhook.contexts.CompositeHookingContextFactory method) (tdhook.contexts.HookingContextFactory method) (tdhook.latent.activation_patching.ActivationPatching method) (tdhook.latent.ActivationAddition method) (tdhook.latent.ActivationPatching method) (tdhook.latent.steering_vectors.ActivationAddition method) (tdhook.weights.Pruning method) (tdhook.weights.pruning.Pruning method) _probe (tdhook.latent.probing.SklearnProbe attribute) _probe_class (tdhook.latent.probing.SklearnProbeManager attribute) _probe_factory (tdhook.latent.Probing attribute) (tdhook.latent.probing.Probing attribute) _probe_kwargs (tdhook.latent.probing.SklearnProbeManager attribute) _probes (tdhook.latent.probing.SklearnProbeManager attribute) _reduce_baselines_fn() (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) (tdhook.attribution.integrated_gradients.IntegratedGradients method) (tdhook.attribution.IntegratedGradients method) _ref (tdhook.hooks.MutableWeakRef attribute) _reg_exp (tdhook.hooks.MultiHookManager attribute) _register_inputs_fn() (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) _relative (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.Probing attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _relative_path (tdhook.modules.HookedModule attribute) (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _restore (tdhook.contexts.HookingContext attribute) _restore_module() (tdhook.attribution.LRP method) (tdhook.attribution.lrp.LRP method) (tdhook.contexts.CompositeHookingContextFactory method) (tdhook.contexts.HookingContextFactory method) _rule (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle attribute) _rule_mapper (tdhook.attribution.LRP attribute) (tdhook.attribution.lrp.LRP attribute) (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper attribute) _rules (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper attribute) _run_callback (tdhook.modules.HookedModuleRun attribute) _save_cache (tdhook.modules.HookedModuleRun attribute) _select_fn (tdhook.attribution.lrp_helpers.rules.ParamModifier attribute) _sep (tdhook.modules.HookedModuleRun attribute) _skip_modules (tdhook.attribution.LRP attribute) (tdhook.attribution.lrp.LRP attribute) (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _spawn (tdhook.contexts.HookingContext attribute) _spawn_hooked_module() (tdhook.contexts.HookingContextFactory method) _stabilizer (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper attribute) _stack (tdhook.contexts.HookingContext attribute) _steer_fn (tdhook.latent.steering_vectors.SteeringVectors attribute) (tdhook.latent.SteeringVectors attribute) _steer_key (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _step_sizes (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) _target_modules (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _td (tdhook.hooks.TensorDictRef attribute) _td_fn (tdhook.modules.FunctionModule attribute) _td_module (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) (tdhook.modules.PGDModule attribute) _use_inputs (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _use_nested_keys (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) _use_outputs (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _use_sign (tdhook.modules.PGDModule attribute) _warn_on_missing_rule (tdhook.attribution.LRP attribute) (tdhook.attribution.lrp.LRP attribute) _weights (tdhook.weights.task_vectors.TaskVectorsModule attribute) _working_key (tdhook.modules.PGDModule attribute) A AbstractFunctionMeta (class in tdhook.attribution.lrp_helpers.rules) Activation (class in tdhook.attribution.lrp_helpers.types) ActivationAddition (class in tdhook.latent) (class in tdhook.latent.steering_vectors) ActivationCaching (class in tdhook.latent) (class in tdhook.latent.activation_caching) ActivationMaximisation (class in tdhook.attribution.activation_maximisation) ActivationPatching (class in tdhook.latent) (class in tdhook.latent.activation_patching) Adapters (class in tdhook.weights) (class in tdhook.weights.adapters) adapters (tdhook.weights.adapters.HookedModuleWithAdapters attribute) AHQKVRule (class in tdhook.attribution.lrp_helpers.rules) AlphaBetaRule (class in tdhook.attribution.lrp_helpers.rules) alphas (tdhook.weights.task_vectors.TaskVectorsContext attribute) approximation_parameters() (in module tdhook.attribution.gradient_helpers.helpers) AvgPool (class in tdhook.attribution.lrp_helpers.types) B backward() (tdhook.attribution.lrp_helpers.rules.AHQKVRule static method) (tdhook.attribution.lrp_helpers.rules.AlphaBetaRule static method) (tdhook.attribution.lrp_helpers.rules.EpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.IgnoreRule static method) (tdhook.attribution.lrp_helpers.rules.LayerNormRule static method) (tdhook.attribution.lrp_helpers.rules.PassRule static method) (tdhook.attribution.lrp_helpers.rules.PseudoIdentityRule static method) (tdhook.attribution.lrp_helpers.rules.Rule static method) (tdhook.attribution.lrp_helpers.rules.SoftmaxEpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.StopRule static method) (tdhook.attribution.lrp_helpers.rules.UniformEpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.UniformRule static method) (tdhook.attribution.lrp_helpers.rules.WSquareRule static method) BaseRuleMapper (class in tdhook.attribution.lrp_helpers.rules) BatchNorm (class in tdhook.attribution.lrp_helpers.types) C cache (tdhook.contexts.HookingContextWithCache property) (tdhook.modules.HookedModuleRun property) cache_ref (tdhook.modules.ModuleCallWithCache property) CacheProxy (class in tdhook.hooks) clear() (tdhook.contexts.HookingContextWithCache method) coef_ (tdhook.latent.probing.MeanDifferenceClassifier property) CompositeHookingContextFactory (class in tdhook.contexts) compute_alpha() (tdhook.weights.task_vectors.TaskVectorsContext method) Convolution (class in tdhook.attribution.lrp_helpers.types) ConvolutionStandard (class in tdhook.attribution.lrp_helpers.types) ConvolutionTranspose (class in tdhook.attribution.lrp_helpers.types) D default_classes_to_hook (tdhook.latent.Probing attribute) (tdhook.latent.probing.Probing attribute) default_classes_to_skip (tdhook.latent.Probing attribute) (tdhook.latent.probing.Probing attribute) default_skip() (tdhook.attribution.LRP static method) (tdhook.attribution.lrp.LRP static method) (tdhook.weights.Pruning static method) (tdhook.weights.pruning.Pruning static method) dim (tdhook.attribution.lrp_helpers.layers.Sum attribute) DimsConfig (class in tdhook.attribution.grad_cam) DIRECTION_TO_PARAMS (in module tdhook.hooks) DIRECTION_TO_RETURN (in module tdhook.hooks) DIRECTION_TO_RETURN_INDEX (in module tdhook.hooks) DIRECTION_TO_TYPE (in module tdhook.hooks) disable() (tdhook.contexts.HookingContext method) disable_context() (tdhook.modules.HookedModule method) disable_context_hooks() (tdhook.modules.HookedModule method) disable_hooks() (tdhook.contexts.HookingContext method) E EarlyStoppingException epsilon (tdhook.attribution.lrp_helpers.rules.EpsilonRule property) EpsilonPlus (class in tdhook.attribution.lrp_helpers.rules) EpsilonRule (class in tdhook.attribution.lrp_helpers.rules) F feature_sum_dims (tdhook.attribution.grad_cam.DimsConfig attribute) fit() (tdhook.latent.probing.MeanDifferenceClassifier method) fit_metrics (tdhook.latent.probing.SklearnProbeManager property) FlatRule (class in tdhook.attribution.lrp_helpers.rules) flatten_select_reshape_call() (in module tdhook.modules) forward() (tdhook.attribution.lrp_helpers.layers.Sum method) (tdhook.attribution.lrp_helpers.rules.AHQKVRule static method) (tdhook.attribution.lrp_helpers.rules.AlphaBetaRule static method) (tdhook.attribution.lrp_helpers.rules.EpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.IgnoreRule static method) (tdhook.attribution.lrp_helpers.rules.LayerNormRule static method) (tdhook.attribution.lrp_helpers.rules.PassRule static method) (tdhook.attribution.lrp_helpers.rules.PseudoIdentityRule static method) (tdhook.attribution.lrp_helpers.rules.Rule static method) (tdhook.attribution.lrp_helpers.rules.StopRule static method) (tdhook.attribution.lrp_helpers.rules.UniformRule static method) (tdhook.attribution.lrp_helpers.rules.WSquareRule static method) (tdhook.modules.FunctionModule method) (tdhook.modules.HookedModule method) (tdhook.modules.IntermediateKeysCleaner method) (tdhook.modules.ModuleCall method) (tdhook.modules.ModuleCallWithCache method) (tdhook.modules.PGDModule method) from_modifiers() (tdhook.attribution.lrp_helpers.rules.ParamModifier static method) from_module() (tdhook.modules.HookedModule class method) FunctionModule (class in tdhook.modules) G gauss_legendre_builders() (in module tdhook.attribution.gradient_helpers.helpers) get() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_best_device() (in module tdhook.modules) get_control_adequacy (tdhook.weights.task_vectors.TaskVectorsContext attribute) get_forget_vector() (tdhook.weights.task_vectors.TaskVectorsModule method) get_grad() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_grad_output() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_input() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_task_vector() (tdhook.weights.task_vectors.TaskVectorsModule method) get_test_accuracy (tdhook.weights.task_vectors.TaskVectorsContext attribute) get_weights() (tdhook.weights.task_vectors.TaskVectorsModule method) GradCAM (class in tdhook.attribution) (class in tdhook.attribution.grad_cam) GradientAttribution (class in tdhook.attribution.gradient_helpers) (class in tdhook.attribution.gradient_helpers.common) GradientAttributionWithBaseline (class in tdhook.attribution.gradient_helpers) (class in tdhook.attribution.gradient_helpers.common) GuidedBackpropagation (class in tdhook.attribution) (class in tdhook.attribution.guided_backpropagation) H HookDirection (in module tdhook.hooks) HookedModule (class in tdhook.modules) HookedModuleRun (class in tdhook.modules) HookedModuleWithAdapters (class in tdhook.weights.adapters) HookFactory (class in tdhook.hooks) hooking_context (tdhook.modules.HookedModule property) HookingContext (class in tdhook.contexts) HookingContextFactory (class in tdhook.contexts) HookingContextWithCache (class in tdhook.contexts) I IgnoreRule (class in tdhook.attribution.lrp_helpers.rules) in_keys (tdhook.modules.FunctionModule attribute) (tdhook.modules.IntermediateKeysCleaner attribute) (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) (tdhook.modules.PGDModule attribute) InfidelityMetric (class in tdhook.metrics) init_attr_targets_with_labels() (tdhook.attribution.integrated_gradients.IntegratedGradients static method) (tdhook.attribution.IntegratedGradients static method) IntegratedGradients (class in tdhook.attribution) (class in tdhook.attribution.integrated_gradients) intercept_ (tdhook.latent.probing.MeanDifferenceClassifier property) IntermediateKeysCleaner (class in tdhook.modules) K key_pattern (tdhook.latent.activation_caching.ActivationCaching property) (tdhook.latent.ActivationCaching property) (tdhook.latent.Probing property) (tdhook.latent.probing.Probing property) L LayerNormRule (class in tdhook.attribution.lrp_helpers.rules) left (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) Linear (class in tdhook.attribution.lrp_helpers.types) LRP (class in tdhook.attribution) (class in tdhook.attribution.lrp) M make_caching_hook() (tdhook.hooks.HookFactory static method) make_reading_hook() (tdhook.hooks.HookFactory static method) make_setting_hook() (tdhook.hooks.HookFactory static method) make_stopping_hook() (tdhook.hooks.HookFactory static method) MaxPool (class in tdhook.attribution.lrp_helpers.types) MeanDifferenceClassifier (class in tdhook.latent.probing) merge_paths() (in module tdhook.hooks) middle (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) module tdhook tdhook._types tdhook.attribution tdhook.attribution.activation_maximisation tdhook.attribution.grad_cam tdhook.attribution.gradient_helpers tdhook.attribution.gradient_helpers.common tdhook.attribution.gradient_helpers.helpers tdhook.attribution.guided_backpropagation tdhook.attribution.integrated_gradients tdhook.attribution.lrp tdhook.attribution.lrp_helpers tdhook.attribution.lrp_helpers.layers tdhook.attribution.lrp_helpers.rules tdhook.attribution.lrp_helpers.types tdhook.attribution.saliency tdhook.auto tdhook.contexts tdhook.hooks tdhook.latent tdhook.latent.activation_caching tdhook.latent.activation_patching tdhook.latent.probing tdhook.latent.representation_similarity tdhook.latent.steering_vectors tdhook.metrics tdhook.modules tdhook.sources tdhook.weights tdhook.weights.adapters tdhook.weights.pruning tdhook.weights.task_vectors ModuleCall (class in tdhook.modules) ModuleCallWithCache (class in tdhook.modules) MultiHookHandle (class in tdhook.hooks) MultiHookManager (class in tdhook.hooks) MutableWeakRef (class in tdhook.hooks) N n_perturb_samples (tdhook.metrics.InfidelityMetric attribute) O out_keys (tdhook.modules.FunctionModule attribute) (tdhook.modules.IntermediateKeysCleaner attribute) (tdhook.modules.ModuleCall attribute) (tdhook.modules.PGDModule attribute) P ParamModifier (class in tdhook.attribution.lrp_helpers.rules) PassRule (class in tdhook.attribution.lrp_helpers.rules) pattern (tdhook.hooks.MultiHookManager property) perturb_radius (tdhook.metrics.SensitivityMetric attribute) PGDModule (class in tdhook.modules) predict() (tdhook.latent.probing.MeanDifferenceClassifier method) predict_metrics (tdhook.latent.probing.SklearnProbeManager property) predict_proba() (tdhook.latent.probing.MeanDifferenceClassifier method) prepare() (tdhook.contexts.HookingContextFactory method) Probe (class in tdhook.latent.probing) probe_factory() (tdhook.latent.probing.SklearnProbeManager method) probes (tdhook.latent.probing.SklearnProbeManager property) Probing (class in tdhook.latent) (class in tdhook.latent.probing) Pruning (class in tdhook.weights) (class in tdhook.weights.pruning) PruningContext (class in tdhook.weights.pruning) PseudoIdentityRule (class in tdhook.attribution.lrp_helpers.rules) R raise_for_unconserved_rel_factory() (in module tdhook.attribution.lrp_helpers.rules) register() (tdhook.attribution.lrp_helpers.rules.IgnoreRule method) (tdhook.attribution.lrp_helpers.rules.Rule method) register_hook() (tdhook.hooks.MultiHookManager method) register_hook_to_module() (in module tdhook.hooks) register_submodule_hook() (tdhook.modules.HookedModule method) relative_path (tdhook.modules.HookedModule property) RemovableHandleProtocol (class in tdhook.hooks) RemovableRuleHandle (class in tdhook.attribution.lrp_helpers.rules) remove() (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle method) (tdhook.hooks.MultiHookHandle method) (tdhook.hooks.RemovableHandleProtocol method) reset_metrics() (tdhook.latent.probing.SklearnProbeManager method) reset_probes() (tdhook.latent.probing.SklearnProbeManager method) resolve() (tdhook.hooks.CacheProxy method) (tdhook.hooks.MutableWeakRef method) (tdhook.hooks.TensorDictRef method) resolve_submodule_path() (in module tdhook.hooks) Riemann (class in tdhook.attribution.gradient_helpers.helpers) riemann_builders() (in module tdhook.attribution.gradient_helpers.helpers) right (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) Rule (class in tdhook.attribution.lrp_helpers.rules) run() (tdhook.modules.HookedModule method) S Saliency (class in tdhook.attribution) (class in tdhook.attribution.saliency) save() (tdhook.modules.HookedModuleRun method) save_grad() (tdhook.modules.HookedModuleRun method) save_grad_output() (tdhook.modules.HookedModuleRun method) save_input() (tdhook.modules.HookedModuleRun method) select_all() (tdhook.attribution.lrp_helpers.rules.ParamModifier static method) SensitivityMetric (class in tdhook.metrics) set() (tdhook.hooks.MutableWeakRef method) (tdhook.hooks.TensorDictRef method) (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) set_grad() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) set_grad_output() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) set_input() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) SklearnProbe (class in tdhook.latent.probing) SklearnProbeManager (class in tdhook.latent.probing) SoftmaxEpsilonRule (class in tdhook.attribution.lrp_helpers.rules) stabilize() (in module tdhook.attribution.lrp_helpers.rules) state_dicts() (tdhook.attribution.lrp_helpers.rules.ParamModifier method) SteeringVectors (class in tdhook.latent) (class in tdhook.latent.steering_vectors) step() (tdhook.latent.probing.Probe method) (tdhook.latent.probing.SklearnProbe method) stop() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) StopRule (class in tdhook.attribution.lrp_helpers.rules) SubclassMeta (class in tdhook.attribution.lrp_helpers.types) submodule_path_to_name() (in module tdhook.hooks) Sum (class in tdhook.attribution.lrp_helpers.layers) SUPPORTED_METHODS (in module tdhook.attribution.gradient_helpers.helpers) SUPPORTED_RIEMANN_METHODS (in module tdhook.attribution.gradient_helpers.helpers) T T (in module tdhook.hooks) TaskVectors (class in tdhook.weights) (class in tdhook.weights.task_vectors) TaskVectorsContext (class in tdhook.weights.task_vectors) TaskVectorsModule (class in tdhook.weights.task_vectors) tdhook module tdhook._types module tdhook.attribution module tdhook.attribution.activation_maximisation module tdhook.attribution.grad_cam module tdhook.attribution.gradient_helpers module tdhook.attribution.gradient_helpers.common module tdhook.attribution.gradient_helpers.helpers module tdhook.attribution.guided_backpropagation module tdhook.attribution.integrated_gradients module tdhook.attribution.lrp module tdhook.attribution.lrp_helpers module tdhook.attribution.lrp_helpers.layers module tdhook.attribution.lrp_helpers.rules module tdhook.attribution.lrp_helpers.types module tdhook.attribution.saliency module tdhook.auto module tdhook.contexts module tdhook.hooks module tdhook.latent module tdhook.latent.activation_caching module tdhook.latent.activation_patching module tdhook.latent.probing module tdhook.latent.representation_similarity module tdhook.latent.steering_vectors module tdhook.metrics module tdhook.modules module tdhook.sources module tdhook.weights module tdhook.weights.adapters module tdhook.weights.pruning module tdhook.weights.task_vectors module TensorDictRef (class in tdhook.hooks) torchvision (in module tdhook.attribution.lrp_helpers.types) trapezoid (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) U UniformEpsilonRule (class in tdhook.attribution.lrp_helpers.rules) UniformRule (class in tdhook.attribution.lrp_helpers.rules) UnraveledKey (class in tdhook._types) UnraveledMeta (class in tdhook._types) unregister() (tdhook.attribution.lrp_helpers.rules.IgnoreRule method) (tdhook.attribution.lrp_helpers.rules.Rule method) W weight_pooling_dims (tdhook.attribution.grad_cam.DimsConfig attribute) with_applied_vectors() (tdhook.weights.task_vectors.TaskVectorsModule method) WSquareRule (class in tdhook.attribution.lrp_helpers.rules)
