lingvo.core.layers_with_attention module