digraph G { e1[label="softmax\n"] e2[label="layer_norm\n"] c[label="b"] e[label="a"] subgraph cluster_0 { label="in-placed" a -> e1 e1 -> c } c -> e2 e2 -> e }