Fix slice size

apache · Jun 12, 2020 · e64a00f · e64a00f
1 parent 7f20b9d
commit e64a00f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py
@@ -1155,6 +1155,9 @@ def _impl(inputs, attr, params, mod):
         if not isinstance(begin, (_expr.Call, _expr.Var)):
             for _ in range(len(begin), data_dim):
                 begin.append(0)
+        elif not isinstance(size, (_expr.Call, _expr.Var)):
+            for _ in range(len(size), data_dim):
+                size.append(-1)
         return _op.strided_slice(inputs[0], begin=begin, end=size,
                                  strides=strides, slice_mode="size")
     return _impl