From 3968cd18c768626fbb51c01a15251709e6b6cd65 Mon Sep 17 00:00:00 2001 From: Alexander Pivovarov Date: Wed, 12 Jun 2019 16:24:08 -0700 Subject: [PATCH] Update tflite schema version to 1.13 --- docker/install/ubuntu_install_tflite.sh | 2 +- tests/python/frontend/tflite/test_forward.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/install/ubuntu_install_tflite.sh b/docker/install/ubuntu_install_tflite.sh index d70f9890053d..802fb3b87d8c 100755 --- a/docker/install/ubuntu_install_tflite.sh +++ b/docker/install/ubuntu_install_tflite.sh @@ -35,7 +35,7 @@ pip2 install flatbuffers # Setup tflite from schema mkdir tflite cd tflite -wget -q https://raw.githubusercontent.com/tensorflow/tensorflow/r1.12/tensorflow/contrib/lite/schema/schema.fbs +wget -q https://raw.githubusercontent.com/tensorflow/tensorflow/r1.13/tensorflow/lite/schema/schema.fbs flatc --python schema.fbs cat <setup.py diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py index 5c2e3afb5a0d..23d46974b243 100644 --- a/tests/python/frontend/tflite/test_forward.py +++ b/tests/python/frontend/tflite/test_forward.py @@ -31,7 +31,7 @@ from tensorflow.python.ops import nn_ops from tensorflow.python.ops import array_ops from tensorflow.python.ops import variables -from tensorflow.contrib import lite as interpreter_wrapper +from tensorflow import lite as interpreter_wrapper import tvm.relay.testing.tf as tf_testing