[keras] init recursive find_layer function

keras.py

@@ -1583,3 +1583,14 @@ def write_summary(model, target):
     model.summary(print_fn=lambda x: summary.append(x))
     summary = "\n".join(summary)
     target.dump(summary)
+
+
+def find_layer(layer, name="LBNLayer"):
+    if layer.name == name:
+        return layer
+    if hasattr(layer, "layers"):
+        sub_layers = layer.layers
+        for sub_layer in sub_layers:
+            layer = find_layer(sub_layer, name=name)
+            if layer is not None:
+                return layer