aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/cc/ops
diff options
context:
space:
mode:
authorGravatar Geoffrey Irving <geoffreyi@google.com>2016-03-16 12:20:34 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-03-16 16:21:58 -0700
commit4c85a08666796faad743a47b63d350132d5c0c90 (patch)
tree5aa1c52226c40994493f755954e79ddbca24c57a /tensorflow/cc/ops
parentf1bccd320a742c67245ded0cbe5eba971294d769 (diff)
Rollforward of "Merge changes from github."
Change: 117375570
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);