"embeddings_2","layer_norm()","(1,8,128)","256","0","(embeddings_1, (128,), l_embeddings_layer_norm_weight_, l_embeddings_layer_norm_bias_, 1e-12)","{}"],["embeddings_3","dropout()","(1,8,128)","0","0","(embeddings_2, 0.1, False, False)","{}"],["getitem_1","getitem()","(1,1,1,8)","0","0","(l_attention_mask_, (slice(None, None, None), None, None, slice(None, None, None)))","{}"],["expand",".expand()","(1,1,8,8)","0","0","(getitem_1, 1, 1, 8, 8)","{}"],["expanded_mask",".to()","(1,1,8,8)","0","0","(expand, to