neuron_explainer/activations/derived_scalars/reconstituted.py [291:302]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                            detach_layer_norm_scale=detach_layer_norm_scale,
                        )

                case PreOrPostAct.POST:

                    def act_fn(
                        resid: torch.Tensor,
                        layer_index: LayerIndex,
                        pass_type: PassType,
                    ) -> torch.Tensor:
                        assert pass_type == PassType.FORWARD
                        assert layer_index is not None
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



neuron_explainer/activations/derived_scalars/reconstituted.py [332:343]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                            detach_layer_norm_scale=detach_layer_norm_scale,
                        )

                case PreOrPostAct.POST:

                    def act_fn(
                        resid: torch.Tensor,
                        layer_index: LayerIndex,
                        pass_type: PassType,
                    ) -> torch.Tensor:
                        assert pass_type == PassType.FORWARD
                        assert layer_index is not None
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -