Commit 3c2569a0 by Yuwei HU Committed by Tianqi Chen

minor fix (#313)

parent e4b500b6
...@@ -103,7 +103,7 @@ def schedule_depthwise_conv2d_map(op): ...@@ -103,7 +103,7 @@ def schedule_depthwise_conv2d_map(op):
if OP not in s.outputs: if OP not in s.outputs:
s[OP].compute_inline() s[OP].compute_inline()
for tensor in OP.input_tensors: for tensor in OP.input_tensors:
if str(tensor.op.input_tensors) != str([]): if tensor.op.input_tensors:
traverse(tensor.op) traverse(tensor.op)
# schedule depthwise_conv2d # schedule depthwise_conv2d
if OP.tag == 'depthwise_conv2d': if OP.tag == 'depthwise_conv2d':
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment