|
|
|
@@ -470,7 +470,7 @@ INPUT_MAP(ApplyAdam) = {{1, INPUT_DESC(var)}, {2, INPUT_DESC(m)}, |
|
|
|
{10, INPUT_DESC(grad)}}; |
|
|
|
ATTR_MAP(ApplyAdam) = {{"use_locking", ATTR_DESC(use_locking, AnyTraits<bool>())}, |
|
|
|
{"use_nesterov", ATTR_DESC(use_nesterov, AnyTraits<bool>())}}; |
|
|
|
OUTPUT_MAP(ApplyAdam) = {{0, OUTPUT_DESC(var)}, {1, OUTPUT_DESC(m)}, {2, OUTPUT_DESC(v)}}; |
|
|
|
OUTPUT_MAP(ApplyAdam) = {{0, OUTPUT_DESC(var)}}; |
|
|
|
|
|
|
|
// Relu6 |
|
|
|
INPUT_MAP(Relu6) = {{1, INPUT_DESC(x)}}; |
|
|
|
@@ -823,7 +823,7 @@ OUTPUT_MAP(RealDiv) = {{0, OUTPUT_DESC(y)}}; |
|
|
|
// Cast |
|
|
|
INPUT_MAP(Cast) = {{1, INPUT_DESC(x)}}; |
|
|
|
INPUT_ATTR_MAP(Cast) = {{2, ATTR_DESC(dst_type, AnyTraits<GEType>())}}; |
|
|
|
ATTR_MAP(Cast) = {{"Truncate", ATTR_DESC(truncate, AnyTraits<bool>())}}; |
|
|
|
ATTR_MAP(Cast) = EMPTY_ATTR_MAP; |
|
|
|
OUTPUT_MAP(Cast) = {{0, OUTPUT_DESC(y)}}; |
|
|
|
|
|
|
|
// Reciprocal |
|
|
|
@@ -1153,7 +1153,7 @@ INPUT_MAP(SparseApplyAdagradD) = { |
|
|
|
{1, INPUT_DESC(var)}, {2, INPUT_DESC(accum)}, {3, INPUT_DESC(grad)}, {4, INPUT_DESC(indices)}}; |
|
|
|
ATTR_MAP(SparseApplyAdagradD) = {{"lr", ATTR_DESC(lr, AnyTraits<float>())}, |
|
|
|
{"use_locking", ATTR_DESC(use_locking, AnyTraits<bool>())}}; |
|
|
|
OUTPUT_MAP(SparseApplyAdagradD) = {{0, OUTPUT_DESC(var)}}; |
|
|
|
OUTPUT_MAP(SparseApplyAdagradD) = {{0, OUTPUT_DESC(var)}, {1, OUTPUT_DESC(accum)}}; |
|
|
|
|
|
|
|
// SparseApplyFtrlD |
|
|
|
INPUT_MAP(SparseApplyFtrlD) = {{1, INPUT_DESC(var)}, |
|
|
|
|