diff options
author | 2016-03-16 12:20:34 -0800 | |
---|---|---|
committer | 2016-03-16 16:21:58 -0700 | |
commit | 4c85a08666796faad743a47b63d350132d5c0c90 (patch) | |
tree | 5aa1c52226c40994493f755954e79ddbca24c57a /tensorflow/cc/ops | |
parent | f1bccd320a742c67245ded0cbe5eba971294d769 (diff) |
Rollforward of "Merge changes from github."
Change: 117375570
Diffstat (limited to 'tensorflow/cc/ops')
-rw-r--r-- | tensorflow/cc/ops/const_op.cc | 3 | ||||
-rw-r--r-- | tensorflow/cc/ops/const_op.h | 1 |
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); |