Skip to content

Commit

Permalink
Update conv2d.py
Browse files Browse the repository at this point in the history
Fixed merge issue
  • Loading branch information
Wheest committed Dec 21, 2020
1 parent 955844b commit c57bd78
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions python/tvm/topi/nn/conv2d.py
Original file line number Diff line number Diff line change
Expand Up @@ -170,10 +170,7 @@ def _get_workload(data, kernel, stride, padding, out_dtype, data_layout="NCHW"):
else:
KH, KW, CIG, CO = get_const_tuple(kernel.shape)

if asymmetric_pad:
pt, pl, pb, pr = get_pad_tuple(padding, (get_const_int(KH), get_const_int(KW)))
else:
HPAD, WPAD, _, _ = get_pad_tuple(padding, (get_const_int(KH), get_const_int(KW)))
HPAD, WPAD, _, _ = get_pad_tuple(padding, (get_const_int(KH), get_const_int(KW)))
GRPS = CI // CIG
if isinstance(stride, (tuple, list)):
HSTR, WSTR = stride
Expand Down

0 comments on commit c57bd78

Please sign in to comment.