Merge pull request #42624 from ashwin-phadke:master

PiperOrigin-RevId: 328839156 Change-Id: I97efb1f11939fcc41f20ceaa504989377637787b
2020-08-27 16:35:26 -07:00 · 2020-08-27 16:35:26 -07:00 · e6f238fbaf
commit e6f238fbaf
parent ea2a96351a 14f41cad58
2 changed files with 12 additions and 0 deletions
--- a/tensorflow/core/api_def/base_api/api_def_ApplyCenteredRMSProp.pbtxt
+++ b/tensorflow/core/api_def/base_api/api_def_ApplyCenteredRMSProp.pbtxt
@ -34,6 +34,12 @@ END
    name: "rho"
    description: <<END
 Decay rate. Must be a scalar.
+END
+  }
+  in_arg {
+    name: "momentum"
+    description: <<END
+Momentum Scale. Must be a scalar.
 END
  }
  in_arg {
--- a/tensorflow/core/api_def/base_api/api_def_ResourceApplyCenteredRMSProp.pbtxt
+++ b/tensorflow/core/api_def/base_api/api_def_ResourceApplyCenteredRMSProp.pbtxt
@ -34,6 +34,12 @@ END
    name: "rho"
    description: <<END
 Decay rate. Must be a scalar.
+END
+  }
+  in_arg {
+    name: "momentum"
+    description: <<END
+Momentum Scale. Must be a scalar.
 END
  }
  in_arg {