Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
Laurawly committed Nov 2, 2020
1 parent dd8d2b7 commit 8380dce
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions python/tvm/relay/op/image/_image.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,8 @@ def resize_func(attrs, inputs, _):
if letter == "C":
channel_axis = i
size = get_const_tuple(attrs.size)
return [_resize_func(inputs[0], convert(size), convert(height_axis), convert(width_axis), convert(channel_axis))]
return [_resize_func(inputs[0], convert(size), convert(height_axis),
convert(width_axis), convert(channel_axis))]


@reg.register_compute("image.resize3d")
Expand Down Expand Up @@ -201,4 +202,4 @@ def grid_sample_func(attrs, inputs, _):
"""
Shape function for grid_sample op.
"""
return [_grid_sample_func(inputs[0], inputs[1])]
return [_grid_sample_func(inputs[0], inputs[1])]

0 comments on commit 8380dce

Please sign in to comment.