Fix OSS Keras API export issue for v1 endpoints.

PiperOrigin-RevId: 362173108
itrushkin · Mar 11, 2021 · f6c7ab3 · f6c7ab3
1 parent bcd8b23
commit f6c7ab3
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 9 deletions.
diff --git a/keras/backend.py b/keras/backend.py
@@ -1010,7 +1010,8 @@ def my_op(a):
   return tf.name_scope(name)
 
 # Export V1 version.
-keras_export(v1=['keras.backend.name_scope'], allow_multiple_exports=True)(tf.compat.v1.name_scope)
+_v1_name_scope = tf.compat.v1.name_scope
+keras_export(v1=['keras.backend.name_scope'], allow_multiple_exports=True)(_v1_name_scope)
 
 
 @keras_export('keras.backend.variable')

diff --git a/keras/initializers/initializers_v1.py b/keras/initializers/initializers_v1.py
@@ -21,20 +21,30 @@
 import tensorflow.compat.v2 as tf
 from tensorflow.python.util.tf_export import keras_export
 
+
+_v1_zeros_initializer = tf.compat.v1.zeros_initializer
+_v1_ones_initializer = tf.compat.v1.ones_initializer
+_v1_constant_initializer = tf.compat.v1.constant_initializer
+_v1_variance_scaling_initializer = tf.compat.v1.variance_scaling_initializer
+_v1_orthogonal_initializer = tf.compat.v1.orthogonal_initializer
+_v1_identity = tf.compat.v1.initializers.identity
+_v1_glorot_uniform_initializer = tf.compat.v1.glorot_uniform_initializer
+_v1_glorot_normal_initializer = tf.compat.v1.glorot_normal_initializer
+
 keras_export(v1=['keras.initializers.Zeros', 'keras.initializers.zeros'], allow_multiple_exports=True)(
-    tf.compat.v1.zeros_initializer)
+    _v1_zeros_initializer)
 keras_export(v1=['keras.initializers.Ones', 'keras.initializers.ones'], allow_multiple_exports=True)(
-    tf.compat.v1.ones_initializer)
+    _v1_ones_initializer)
 keras_export(v1=['keras.initializers.Constant', 'keras.initializers.constant'], allow_multiple_exports=True)(
-    tf.compat.v1.constant_initializer)
+    _v1_constant_initializer)
 keras_export(v1=['keras.initializers.VarianceScaling'], allow_multiple_exports=True)(
-    tf.compat.v1.variance_scaling_initializer)
+    _v1_variance_scaling_initializer)
 keras_export(v1=['keras.initializers.Orthogonal',
-                 'keras.initializers.orthogonal'], allow_multiple_exports=True)(tf.compat.v1.orthogonal_initializer)
+                 'keras.initializers.orthogonal'], allow_multiple_exports=True)(_v1_orthogonal_initializer)
 keras_export(v1=['keras.initializers.Identity',
-                 'keras.initializers.identity'], allow_multiple_exports=True)(tf.compat.v1.initializers.identity)
-keras_export(v1=['keras.initializers.glorot_uniform'], allow_multiple_exports=True)(tf.compat.v1.glorot_uniform_initializer)
-keras_export(v1=['keras.initializers.glorot_normal'], allow_multiple_exports=True)(tf.compat.v1.glorot_normal_initializer)
+                 'keras.initializers.identity'], allow_multiple_exports=True)(_v1_identity)
+keras_export(v1=['keras.initializers.glorot_uniform'], allow_multiple_exports=True)(_v1_glorot_uniform_initializer)
+keras_export(v1=['keras.initializers.glorot_normal'], allow_multiple_exports=True)(_v1_glorot_normal_initializer)
 
 
 @keras_export(v1=['keras.initializers.RandomNormal',