Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
Laurawly committed Sep 5, 2019
1 parent 745af6b commit 7cf3365
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions topi/python/topi/intel_graphics/depthwise_conv2d.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
from ..util import traverse_inline
from .. import tag
from .. import generic, nn
from ..nn.depthwise_conv2d import depthwise_conv2d_NCHWc, depthwise_conv2d_infer_layout
from ..nn.depthwise_conv2d import depthwise_conv2d_infer_layout

# register original implementation of depthwise_conv2d_nchw since we don't need to change this part
autotvm.register_topi_compute(nn.depthwise_conv2d_nchw, ['intel_graphics'], 'direct',
Expand Down Expand Up @@ -313,7 +313,7 @@ def traverse(OP):
return s

@depthwise_conv2d_infer_layout.register("intel_graphics")
def _depthwise_conv2d_infer_layout(workload, cfg):
def _depthwise_conv2d_infer_layout(workload, _):
"""Infer input/output shapes and layouts from a workload and cfg.
Parameters
Expand All @@ -329,7 +329,7 @@ def _depthwise_conv2d_infer_layout(workload, cfg):
Output : [tuple of tuple and str, tuple of tuple and str]
Input shapes and layouts, and output shapes and layouts
"""
_, data, kernel, strides, padding, dilation, dtype = workload
_, data, kernel, strides, padding, _, _ = workload
batch_size, in_channel, in_height, in_width = data[:-1]
filter_channel, channel_multiplier, k_height, k_width = kernel[:-1]
out_channel = filter_channel * channel_multiplier
Expand Down

0 comments on commit 7cf3365

Please sign in to comment.