Broadcast src0 into src1 across dimensions 2 and 3 when needed. This is required for models that use GQA.