|
|
|
@@ -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>())}}; |
|
|
|
|