Merge pull request #25535 from lgeiger:inv-non-monotonic
PiperOrigin-RevId: 233138653
This commit is contained in:
commit
14554b2371
@ -176,8 +176,7 @@ bool IsElementWiseMonotonic(const NodeDef& node, bool* is_non_decreasing) {
|
|||||||
"Sign", "Sinh", "Softsign", "Softplus", "Sqrt", "Tanh",
|
"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>{"Acos", "Erfc", "Inv", "Neg",
|
CHECK_NOTNULL((new gtl::FlatSet<string>{"Acos", "Erfc", "Neg", "Rsqrt"}));
|
||||||
"Reciprocal", "Rsqrt"}));
|
|
||||||
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;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user