fix(0306): MoE prefill reductions for subfunction export#1028

quic-rishinr merged 2 commits into

release/v1.22.0_tmpquic/efficient-transformers:release/v1.22.0_tmpfrom

fix/subfunction-reducesum-einsumquic/efficient-transformers:fix/subfunction-reducesum-einsumCopy head branch name to clipboard