From b6a0ee9f6c947cef39bb7a225fa7b6e0b43d99be Mon Sep 17 00:00:00 2001
From: zhaojichen <zhaojichen1@huawei.com>
Date: Wed, 3 Jun 2020 07:53:33 -0400
Subject: [PATCH] fix applyrmsprop bug

---
 mindspore/ccsrc/transform/op_declare.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/transform/op_declare.cc b/mindspore/ccsrc/transform/op_declare.cc
index 59ccb2416..c3f8646b3 100644
--- a/mindspore/ccsrc/transform/op_declare.cc
+++ b/mindspore/ccsrc/transform/op_declare.cc
@@ -1226,7 +1226,7 @@ OUTPUT_MAP(Atan2) = {{0, OUTPUT_DESC(y)}};
 
 // ApplyRMSPropD
 INPUT_MAP(ApplyRMSPropD) = {
-  {1, INPUT_DESC(var)}, {2, INPUT_DESC(ms)}, {3, INPUT_DESC(mom)}, {4, INPUT_DESC(grad)}, {5, INPUT_DESC(lr)}};
+  {1, INPUT_DESC(var)}, {2, INPUT_DESC(ms)}, {3, INPUT_DESC(mom)}, {4, INPUT_DESC(lr)}, {5, INPUT_DESC(grad)}};
 INPUT_ATTR_MAP(ApplyRMSPropD) = {{6, ATTR_DESC(rho, AnyTraits<float>())},
                                  {7, ATTR_DESC(momentum, AnyTraits<float>())},
                                  {8, ATTR_DESC(epsilon, AnyTraits<float>())}};
-- 
GitLab