aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/cc/ops
diff options
context:
space:
mode:
authorGravatar Martin Wicke <wicke@google.com>2016-03-15 17:53:33 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-03-16 16:19:29 -0700
commit3ae663ccc5d08976e0f547d5b2ece35067a6673e (patch)
tree45810d02a11acc9b45e4268fd00548929ea91ffd /tensorflow/cc/ops
parenta0d21ec39cc3f18781d2d37798aa328e12f92844 (diff)
Merge changes from github.
Change: 117301677
Diffstat (limited to 'tensorflow/cc/ops')
-rw-r--r--tensorflow/cc/ops/const_op.cc3
-rw-r--r--tensorflow/cc/ops/const_op.h1
2 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/cc/ops/const_op.cc b/tensorflow/cc/ops/const_op.cc
index cdf653a880..ddfa2a5b0e 100644
--- a/tensorflow/cc/ops/const_op.cc
+++ b/tensorflow/cc/ops/const_op.cc
@@ -87,6 +87,9 @@ DEFINE_CONST(bool, bool_val);
DEFINE_CONST_IMPL(complex64, proto.add_scomplex_val(t.begin()->real());
proto.add_scomplex_val(t.begin()->imag()););
+DEFINE_CONST_IMPL(complex128, proto.add_dcomplex_val(t.begin()->real());
+ proto.add_dcomplex_val(t.begin()->imag()););
+
Node* Const(StringPiece s, const GraphDefBuilder::Options& options) {
if (options.HaveError()) return nullptr;
NodeBuilder node_builder(options.GetNameForOp(OpName()), OpName(),
diff --git a/tensorflow/cc/ops/const_op.h b/tensorflow/cc/ops/const_op.h
index 36a97f8e4c..0a1ee3f1e0 100644
--- a/tensorflow/cc/ops/const_op.h
+++ b/tensorflow/cc/ops/const_op.h
@@ -49,6 +49,7 @@ DECLARE_CONST(uint8);
DECLARE_CONST(int16);
DECLARE_CONST(int8);
DECLARE_CONST(complex64);
+DECLARE_CONST(complex128);
DECLARE_CONST(int64);
DECLARE_CONST(bool);