Fix: Pass einsum_str instead of module name to get_axis_to_reduce_from_einsum_str#480

markknoffler wants to merge 3 commits intogoogle-deepmind:maingoogle-deepmind/gemma:mainfrom

markknoffler:einsum_bug_fixmarkknoffler/gemma:einsum_bug_fixCopy head branch name to clipboard