moving import and referencing test for SelectV2

srinidhigoud · Apr 24, 2021 · 3df4763 · 3df4763
1 parent 867ae40
commit 3df4763
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -3620,7 +3620,7 @@ def test_forward_logical():
 
 
 #######################################################################
-# Where, Select
+# Where, Select, SelectV2
 # -------------
 def test_forward_where():
     """ Where: return elements depending on conditions"""