aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/grappler/op_types.h
diff options
context:
space:
mode:
authorGravatar Yao Zhang <yaozhang@google.com>2018-01-09 11:31:05 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-01-09 11:39:24 -0800
commitfe613b051b5732f06af586f7b4f0ca28121f1ecc (patch)
tree910e7f7756df10ec0d9095584547e04e4f4ab873 /tensorflow/core/grappler/op_types.h
parent13d8bf3cba46361bf61c4dc3e8cd237c85e8f735 (diff)
Fix type attribute name for Cast.
PiperOrigin-RevId: 181348431
Diffstat (limited to 'tensorflow/core/grappler/op_types.h')
-rw-r--r--tensorflow/core/grappler/op_types.h1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/op_types.h b/tensorflow/core/grappler/op_types.h
index 3106739192..6b5e7e3391 100644
--- a/tensorflow/core/grappler/op_types.h
+++ b/tensorflow/core/grappler/op_types.h
@@ -36,6 +36,7 @@ bool IsBetainc(const NodeDef& node);
bool IsBiasAdd(const NodeDef& node);
bool IsBiasAddGrad(const NodeDef& node);
bool IsBitcast(const NodeDef& node);
+bool IsCast(const NodeDef& node);
bool IsComplex(const NodeDef& node);
bool IsComplexAbs(const NodeDef& node);
bool IsConj(const NodeDef& node);