diff options
author | 2018-06-27 18:11:03 -0700 | |
---|---|---|
committer | 2018-06-27 18:13:25 -0700 | |
commit | 128b69878362dcec736326d52828d2167ce291cb (patch) | |
tree | 7cc93883a8248c3ee8f799502e4d39646a31a419 /tensorflow/contrib/lite/model.cc | |
parent | 2b3b5054c7ceff0bc2811cfe0ebc063947801ce0 (diff) |
Add complex64 support to tf.lite runtime.
PiperOrigin-RevId: 202403235
Diffstat (limited to 'tensorflow/contrib/lite/model.cc')
-rw-r--r-- | tensorflow/contrib/lite/model.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/model.cc b/tensorflow/contrib/lite/model.cc index 793a72272d..f54db3af87 100644 --- a/tensorflow/contrib/lite/model.cc +++ b/tensorflow/contrib/lite/model.cc @@ -63,6 +63,9 @@ TfLiteStatus ConvertTensorType(TensorType tensor_type, TfLiteType* type, case TensorType_BOOL: *type = kTfLiteBool; break; + case TensorType_COMPLEX64: + *type = kTfLiteComplex64; + break; default: error_reporter->Report("Unimplemented data type %s (%d) in tensor\n", EnumNameTensorType(tensor_type), tensor_type); |