From 48959a13a164f3474cf8c28f87a89ca54333c6d0 Mon Sep 17 00:00:00 2001 From: Siju Samuel Date: Wed, 10 Jul 2019 14:43:02 +0530 Subject: [PATCH] Deprecated tf.Session removed in toco_from_protos_test.py --- tensorflow/lite/toco/python/toco_from_protos_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/lite/toco/python/toco_from_protos_test.py b/tensorflow/lite/toco/python/toco_from_protos_test.py index bb2cab3e1ed..e673f3fd32c 100644 --- a/tensorflow/lite/toco/python/toco_from_protos_test.py +++ b/tensorflow/lite/toco/python/toco_from_protos_test.py @@ -80,7 +80,7 @@ class TocoFromProtosTest(googletest.TestCase): def test_toco(self): """Run a couple of TensorFlow graphs against TOCO through the python bin.""" - with tf.Session() as sess: + with tf.compat.v1.Session() as sess: img = tf.placeholder(name="img", dtype=tf.float32, shape=(1, 64, 64, 3)) val = img + tf.constant([1., 2., 3.]) + tf.constant([1., 4., 4.]) out = tf.identity(val, name="out")