fix axes null check

masahi · May 27, 2021 · 32698b7 · 32698b7
1 parent 54fb723
commit 32698b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/op/_transform.py b/python/tvm/relay/op/_transform.py
@@ -297,7 +297,7 @@ def strided_slice_shape_func(attrs, inputs, _):
     Shape func for strided_slice
     """
     slice_mode = convert(0 if attrs.slice_mode == "end" else 1)
-    if len(attrs.axes) == 0:
+    if attrs.axes is None:
         return [
             _strided_slice_shape_func_input_shape(
                 inputs[0], attrs.begin, attrs.end, attrs.strides, slice_mode