Internal changes only.

PiperOrigin-RevId: 371633732
dgrahn · May 3, 2021 · 2b00f0f · 2b00f0f
1 parent bf6859b
commit 2b00f0f
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 1 deletion.
diff --git a/keras/api/golden/v1/tensorflow.keras.__internal__.legacy.layers.-input-spec.pbtxt b/keras/api/golden/v1/tensorflow.keras.__internal__.legacy.layers.-input-spec.pbtxt
@@ -0,0 +1,17 @@
+path: "tensorflow.keras.__internal__.legacy.layers.InputSpec"
+tf_class {
+  is_instance: "<class \'keras.engine.input_spec.InputSpec\'>"
+  is_instance: "<type \'object\'>"
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'dtype\', \'shape\', \'ndim\', \'max_ndim\', \'min_ndim\', \'axes\', \'allow_last_axis_squeeze\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'None\', \'False\', \'None\'], "
+  }
+  member_method {
+    name: "from_config"
+    argspec: "args=[\'cls\', \'config\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "get_config"
+    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
+  }
+}
diff --git a/keras/api/golden/v1/tensorflow.keras.__internal__.legacy.layers.pbtxt b/keras/api/golden/v1/tensorflow.keras.__internal__.legacy.layers.pbtxt
@@ -48,6 +48,10 @@ tf_module {
     name: "Flatten"
     mtype: "<type \'type\'>"
   }
+  member {
+    name: "InputSpec"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "Layer"
     mtype: "<type \'type\'>"

diff --git a/keras/engine/input_spec.py b/keras/engine/input_spec.py
@@ -22,7 +22,9 @@
 from tensorflow.python.util.tf_export import tf_export
 
 
-@keras_export('keras.layers.InputSpec')
+@keras_export('keras.layers.InputSpec',
+              v1=['keras.layers.InputSpec',
+                  'keras.__internal__.legacy.layers.InputSpec'])
 @tf_export(v1=['layers.InputSpec'])
 class InputSpec(object):
   """Specifies the rank, dtype and shape of every input to a layer.