aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/grappler/op_types.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-06-28 22:11:49 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-28 22:13:59 -0700
commit6874e1ef40c4189d96c105227f60b507953f95d3 (patch)
treeca8c6d0f04aec4c2a32c99582084116606f9adaa /tensorflow/core/grappler/op_types.cc
parent50839154899377f89367d851f6d1e2c45fcd6431 (diff)
Convert exp(x-1) into expm1(x).
PiperOrigin-RevId: 202598404
Diffstat (limited to 'tensorflow/core/grappler/op_types.cc')
-rw-r--r--tensorflow/core/grappler/op_types.cc2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/op_types.cc b/tensorflow/core/grappler/op_types.cc
index bdeb5c66fc..653b088b1d 100644
--- a/tensorflow/core/grappler/op_types.cc
+++ b/tensorflow/core/grappler/op_types.cc
@@ -161,6 +161,8 @@ bool IsExit(const NodeDef& node) {
return op == "Exit" || op == "RefExit";
}
+bool IsExp(const NodeDef& node) { return node.op() == "Exp"; }
+
bool IsFill(const NodeDef& node) { return node.op() == "Fill"; }
bool IsFloorDiv(const NodeDef& node) { return node.op() == "FloorDiv"; }