lingvo.core.batch_major_attention module