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

Remove reciprocal from monotonic decreasing ops

Browse Source
This commit is contained in:
Lukas Geiger 2019-02-06 02:48:31 +01:00
parent d08271d73c
commit f7fba1b59b
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -184,8 +184,7 @@ bool IsElementWiseMonotonic(const NodeDef& node, bool* is_non_decreasing) {
"Sign", "Sinh", "Softsign", "Softplus", "Sqrt", "Tanh",
}));
static const gtl::FlatSet<string>* const kMonotonicNonIncreasingOps =
CHECK_NOTNULL((new gtl::FlatSet<string>{"Acos", "Erfc", "Inv", "Neg",
"Reciprocal", "Rsqrt"}));
CHECK_NOTNULL((new gtl::FlatSet<string>{"Acos", "Erfc", "Neg", "Rsqrt"}));
if (kMonotonicNonDecreasingOps->count(node.op()) > 0) {
if (is_non_decreasing) {
*is_non_decreasing = true;