- Refactor learning rate schedule code to live in `keras/optimizers/schedules` (to match the API namespace).
- Export optimizers to `keras.optimizers.__init__` to resolve discrepancy between tf.keras and keras packages.
- Export schedules to `keras.optimizers.schedules.__init__` to resolve discrepancy between tf.keras and keras packages.
PiperOrigin-RevId: 435722952
diff --git a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-cosine-decay-restarts.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-cosine-decay-restarts.pbtxt
index 712dc5d..f465c90 100644
--- a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-cosine-decay-restarts.pbtxt
+++ b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-cosine-decay-restarts.pbtxt
@@ -1,7 +1,7 @@
path: "tensorflow.optimizers.schedules.CosineDecayRestarts"
tf_class {
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.CosineDecayRestarts\'>"
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.LearningRateSchedule\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.CosineDecayRestarts\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.LearningRateSchedule\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
diff --git a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-cosine-decay.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-cosine-decay.pbtxt
index 825a1b0..004d740 100644
--- a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-cosine-decay.pbtxt
+++ b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-cosine-decay.pbtxt
@@ -1,7 +1,7 @@
path: "tensorflow.optimizers.schedules.CosineDecay"
tf_class {
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.CosineDecay\'>"
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.LearningRateSchedule\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.CosineDecay\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.LearningRateSchedule\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
diff --git a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-exponential-decay.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-exponential-decay.pbtxt
index 8e60130..1f603e4 100644
--- a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-exponential-decay.pbtxt
+++ b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-exponential-decay.pbtxt
@@ -1,7 +1,7 @@
path: "tensorflow.optimizers.schedules.ExponentialDecay"
tf_class {
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.ExponentialDecay\'>"
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.LearningRateSchedule\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.ExponentialDecay\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.LearningRateSchedule\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
diff --git a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-inverse-time-decay.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-inverse-time-decay.pbtxt
index 154b0be..270148b 100644
--- a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-inverse-time-decay.pbtxt
+++ b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-inverse-time-decay.pbtxt
@@ -1,7 +1,7 @@
path: "tensorflow.optimizers.schedules.InverseTimeDecay"
tf_class {
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.InverseTimeDecay\'>"
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.LearningRateSchedule\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.InverseTimeDecay\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.LearningRateSchedule\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
diff --git a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-learning-rate-schedule.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-learning-rate-schedule.pbtxt
index 0a7f95d..583cd8f 100644
--- a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-learning-rate-schedule.pbtxt
+++ b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-learning-rate-schedule.pbtxt
@@ -1,6 +1,6 @@
path: "tensorflow.optimizers.schedules.LearningRateSchedule"
tf_class {
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.LearningRateSchedule\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.LearningRateSchedule\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
diff --git a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-piecewise-constant-decay.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-piecewise-constant-decay.pbtxt
index 19bfa0e..51b3f9c 100644
--- a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-piecewise-constant-decay.pbtxt
+++ b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-piecewise-constant-decay.pbtxt
@@ -1,7 +1,7 @@
path: "tensorflow.optimizers.schedules.PiecewiseConstantDecay"
tf_class {
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.PiecewiseConstantDecay\'>"
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.LearningRateSchedule\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.PiecewiseConstantDecay\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.LearningRateSchedule\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
diff --git a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-polynomial-decay.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-polynomial-decay.pbtxt
index 472b4f8..411262f 100644
--- a/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-polynomial-decay.pbtxt
+++ b/tensorflow/tools/api/golden/v2/tensorflow.optimizers.schedules.-polynomial-decay.pbtxt
@@ -1,7 +1,7 @@
path: "tensorflow.optimizers.schedules.PolynomialDecay"
tf_class {
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.PolynomialDecay\'>"
- is_instance: "<class \'keras.optimizers.learning_rate_schedule.LearningRateSchedule\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.PolynomialDecay\'>"
+ is_instance: "<class \'keras.optimizers.schedules.learning_rate_schedule.LearningRateSchedule\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
