From 579e96da4481aa4919d22abba934a0db0f736a6a Mon Sep 17 00:00:00 2001 From: Alexander Pivovarov Date: Thu, 13 Jun 2019 08:52:25 -0700 Subject: [PATCH] Update tflite schema version to 1.13 (#3356) --- 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 d70f989..802fb3b 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 5c2e3af..23d4697 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 math_ops 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 -- 2.7.4