From 128b69878362dcec736326d52828d2167ce291cb Mon Sep 17 00:00:00 2001
From: RJ Ryan <rjryan@google.com>
Date: Wed, 27 Jun 2018 18:11:03 -0700
Subject: Add complex64 support to tf.lite runtime.

PiperOrigin-RevId: 202403235
---
 tensorflow/contrib/lite/model.cc | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'tensorflow/contrib/lite/model.cc')

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);
-- 
cgit v1.2.3