lingvo.core.self_attention_layer module