From 32698b74df211829777e5493e82bf7425364acb4 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Sat, 22 May 2021 13:11:01 +0900 Subject: [PATCH] fix axes null check --- python/tvm/relay/op/_transform.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/op/_transform.py b/python/tvm/relay/op/_transform.py index 2622b129c424..c11239ca2bbb 100644 --- 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