diff options
Diffstat (limited to 'tensorflow/python/lib/core/py_func.cc')
-rw-r--r-- | tensorflow/python/lib/core/py_func.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/python/lib/core/py_func.cc b/tensorflow/python/lib/core/py_func.cc index 5701e8fe70..1949913ec8 100644 --- a/tensorflow/python/lib/core/py_func.cc +++ b/tensorflow/python/lib/core/py_func.cc @@ -99,6 +99,9 @@ Status TfDTypeToNpDType(const DataType& tf, int* np) { case DT_COMPLEX64: *np = NPY_COMPLEX64; break; + case DT_COMPLEX128: + *np = NPY_COMPLEX128; + break; case DT_STRING: *np = NPY_OBJECT; break; @@ -210,6 +213,9 @@ Status NumericNpDTypeToTfDType(const int np, DataType* tf) { case NPY_COMPLEX64: *tf = DT_COMPLEX64; break; + case NPY_COMPLEX128: + *tf = DT_COMPLEX128; + break; default: return errors::Unimplemented("Unsupported numpy type ", np); } |