transformer_lens.conversion_utils.conversion_steps.ternary_tensor_conversion module

Weight conversion that performs ternary operations on weights.

class transformer_lens.conversion_utils.conversion_steps.ternary_tensor_conversion.TernaryTensorConversion(fallback_conversion: Any, primary_conversion: Tensor | BaseTensorConversion | None = None, input_filter: Callable | None = None, output_filter: Callable | None = None)

Bases: BaseTensorConversion

find_context_field(field_key: str, *full_context)

handle_conversion(input_value: Tensor | None, *full_context) → Tensor | None

handle_fallback_conversion(*full_context) → Tensor | None

handle_primary_conversion(input_value: Tensor, *full_context) → Tensor