experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #25332 from lgeiger:monotonic-functions

Browse Source 
PiperOrigin-RevId: 232071191
This commit is contained in:
TensorFlower Gardener 2019-02-01 18:24:36 -08:00
commit 9167d6d159
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/core/grappler/op_types.cc
View File

@ -178,18 +178,14 @@ bool IsDiv(const NodeDef& node) { return node.op() == "Div"; }
bool IsElementWiseMonotonic(const NodeDef& node, bool* is_non_decreasing) { bool IsElementWiseMonotonic(const NodeDef& node, bool* is_non_decreasing) {
static const gtl::FlatSet<string>* const kMonotonicNonDecreasingOps = static const gtl::FlatSet<string>* const kMonotonicNonDecreasingOps =
CHECK_NOTNULL((new gtl::FlatSet<string>{ CHECK_NOTNULL((new gtl::FlatSet<string>{
"Asinh", "Atanh", "Ceil", "Elu", "Erf", "Exp", "Expm1", "Acosh", "Asin", "Asinh", "Atan", "Atanh", "Ceil",
"Floor", "Log", "Log1p", "Relu", "Relu", "Relu6", "Rint", "Elu", "Erf", "Exp", "Expm1", "Floor", "Log",
"Selu", "Sigmoid", "Sign", "Sinh", "Sqrt", "Tanh", "Log1p", "Relu", "Relu6", "Rint", "Selu", "Sigmoid",
"Sign", "Sinh", "Softsign", "Softplus", "Sqrt", "Tanh",
})); }));
static const gtl::FlatSet<string>* const kMonotonicNonIncreasingOps = static const gtl::FlatSet<string>* const kMonotonicNonIncreasingOps =
CHECK_NOTNULL((new gtl::FlatSet<string>{ CHECK_NOTNULL((new gtl::FlatSet<string>{"Acos", "Erfc", "Inv", "Neg",
"Inv", "Reciprocal", "Rsqrt"}));
"Reciprocal",
"Erfc",
"Rsqrt",
"Neg",
}));
if (kMonotonicNonDecreasingOps->count(node.op()) > 0) { if (kMonotonicNonDecreasingOps->count(node.op()) > 0) {
if (is_non_decreasing) { if (is_non_decreasing) {
*is_non_decreasing = true; *is_non_decreasing = true;