diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-28 16:12:51 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-28 16:15:28 -0700 |
commit | 830c19c3f20816dcb5e8e9b6cb51f63cf8461442 (patch) | |
tree | 703848df8ce2caaa2271e4f24daaeeab067f4ce0 /tensorflow/core/grappler/op_types.cc | |
parent | 6cb3e6e0988a7bd123e683c13dae8470c71822af (diff) |
Add IsSquare bool to the grappler op_types.
PiperOrigin-RevId: 190852501
Diffstat (limited to 'tensorflow/core/grappler/op_types.cc')
-rw-r--r-- | tensorflow/core/grappler/op_types.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/op_types.cc b/tensorflow/core/grappler/op_types.cc index 1a6751befc..c31ac9b59c 100644 --- a/tensorflow/core/grappler/op_types.cc +++ b/tensorflow/core/grappler/op_types.cc @@ -309,6 +309,8 @@ bool IsSplitV(const NodeDef& node) { return node.op() == "SplitV"; } bool IsSqrtGrad(const NodeDef& node) { return node.op() == "SqrtGrad"; } +bool IsSquare(const NodeDef& node) { return node.op() == "Square"; } + bool IsSquaredDifference(const NodeDef& node) { return node.op() == "SquaredDifference"; } |