diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-25 10:36:13 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-25 10:39:02 -0700 |
commit | bf94bc682f1494aa912ef38628d82942bf4a1fc7 (patch) | |
tree | ff12ed9532be3d23d9bcbe1ea3a7a4faf8b3b437 /tensorflow/core/grappler/op_types.cc | |
parent | 169f88708959de64374ff4c5e7728d9b75d95b4d (diff) |
Automated g4 rollback of changelist 194186080
PiperOrigin-RevId: 194261487
Diffstat (limited to 'tensorflow/core/grappler/op_types.cc')
-rw-r--r-- | tensorflow/core/grappler/op_types.cc | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/tensorflow/core/grappler/op_types.cc b/tensorflow/core/grappler/op_types.cc index f595cf6456..fba6c5810d 100644 --- a/tensorflow/core/grappler/op_types.cc +++ b/tensorflow/core/grappler/op_types.cc @@ -506,10 +506,13 @@ bool IsUnaryElementWise(const NodeDef& node) { "Atan2", "Atanh", "Ceil", + "CheckNumerics", "ComplexAbs", "Conj", "Cos", "Cosh", + "DebugGradientIdentity", + "DeepCopy" "Digamma", "Elu" "Erf", @@ -517,36 +520,37 @@ bool IsUnaryElementWise(const NodeDef& node) { "Exp", "Expm1", "Floor", + "Identity", "Inv", "Invert", - "Isinf", - "Isnan", - "Isfinite", "Lgamma", "Log", "Log1p", "LogicalNot", "Neg", + "PreventGradient", + "Print", "Reciprocal", "Relu", "Relu6", "Rint", "Round", - "Selu", "Rsqrt", + "Selu", "Sigmoid", "Sign", "Sin", "SinH", + "Snapshot", "Softplus", "Softsign", "Sqrt", "Square", + "StopGradient", "Tan" "Tanh", })); - return element_wise_ops->count(node.op()) > 0 || - (!IsIdentityN(node) && IsValueAndOrderPreserving(node)); + return element_wise_ops->find(node.op()) != element_wise_ops->end(); } bool HasOpDef(const NodeDef& node) { |