respect dims_per_head in the case without value projection
respect dims_per_head in the case without value projection

PiperOrigin-RevId: 393396847

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo