Fix QK norm reshape to match RMSNorm's expected 3D input shape#1221

brendanlong wants to merge 1 commit intoTransformerLensOrg:devTransformerLensOrg/TransformerLens:devfrom

brendanlong:brendanlong/gemma3-attention-shapebrendanlong/TransformerLens:brendanlong/gemma3-attention-shapeCopy head branch name to clipboard