diff --git a/python/dllib/test/dev/diff.py b/python/dllib/test/dev/diff.py
index <HASH>..<HASH> 100755
--- a/python/dllib/test/dev/diff.py
+++ b/python/dllib/test/dev/diff.py
@@ -29,7 +29,9 @@ scala_to_python = {"Graph": "Model"}
 
 def extract_scala_class(class_path):
     exclude_key_words = set(["*", "abstract", "Const", "Fill", "Shape",
-                             "SplitAndSelect", "StrideSlice", "Scheduler", "StaticGraph", "DynamicGraph", "DynamicContainer"])   # noqa
+                             "SplitAndSelect", "StrideSlice", "Scheduler",
+                             "StaticGraph", "DynamicGraph", "DynamicContainer",
+                             "SplitHeads", "CombineHeads"])   # noqa
     include_key_words = set(["Module", "Criterion", "Container", "Cell", "TensorNumeric"])  # noqa
     content = "\n".join([line for line in open(class_path).readlines() if all([key not in line for key in exclude_key_words])])  # noqa
     match = re.search(r"class ([\w]+)[^{]+", content)
[New feature] Add attention layer and ffn layer (#<I>)

* add attention layer



* add ffn layer and more unit tests



* refactor according to pr comments



* add SerializationTest



* fix unit tests



* add python api