Skip to content

Commit

Permalink
fix axes null check
Browse files Browse the repository at this point in the history
  • Loading branch information
masahi committed May 27, 2021
1 parent 54fb723 commit 32698b7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/tvm/relay/op/_transform.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 32698b7

Please sign in to comment.