diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py index 0de00517e9a8..50221c7baf28 100644 --- a/python/tvm/relay/frontend/tensorflow.py +++ b/python/tvm/relay/frontend/tensorflow.py @@ -19,6 +19,7 @@ """TF: Tensorflow frontend.""" import warnings from collections import defaultdict +from collections import deque # Numpy support import numpy as np @@ -1778,7 +1779,6 @@ def _impl(inputs, attr, params, mod): s0 = list(s0.asnumpy().reshape([-1])) s1 = list(s1.asnumpy().reshape([-1])) s0_size, s1_size = len(s0), len(s1) - from collections import deque out = deque([]) for i in range(1, min(s0_size, s1_size) + 1): diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py index 2e4237b34ab1..f8d3e2386070 100644 --- a/tests/python/frontend/tensorflow/test_forward.py +++ b/tests/python/frontend/tensorflow/test_forward.py @@ -3620,7 +3620,7 @@ def test_forward_logical(): ####################################################################### -# Where, Select +# Where, Select, SelectV2 # ------------- def test_forward_where(): """ Where: return elements depending on conditions"""