From: Sammy <samshang.wang@mail.utoronto.ca>
Date: Tue, 25 Jun 2019 03:55:55 +0000 (-0400)
Subject: Fixing package path in tflite test (#3427)
X-Git-Tag: upstream/0.7.0~2260
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=e97c01012d3a04c1f2a39abbc810c34395c74080;p=platform%2Fupstream%2Ftvm.git

Fixing package path in tflite test (#3427)
---

diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py
index 41147fe..c9fd0dc 100644
--- a/tests/python/frontend/tflite/test_forward.py
+++ b/tests/python/frontend/tflite/test_forward.py
@@ -32,7 +32,10 @@ from tensorflow.python.ops import math_ops
 from tensorflow.python.ops import nn_ops
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import variables
-from tensorflow import lite as interpreter_wrapper
+try:
+    from tensorflow import lite as interpreter_wrapper
+except ImportError:
+    from tensorflow.contrib import lite as interpreter_wrapper
 
 import tvm.relay.testing.tf as tf_testing
 
@@ -131,7 +134,7 @@ def compare_tflite_with_tvm(in_data, in_name, input_tensors,
         if init_global_variables:
             sess.run(variables.global_variables_initializer())
         # convert to tflite model
-        converter = tf.contrib.lite.TFLiteConverter.from_session(
+        converter = interpreter_wrapper.TFLiteConverter.from_session(
             sess, input_tensors, output_tensors)
         tflite_model_buffer = converter.convert()
         tflite_output = run_tflite_graph(tflite_model_buffer, in_data)