Commit a152a9cb by Yuwei Hu Committed by Tianqi Chen

[TOPI] modify conv2d_transpose schedule (#613)

parent 685f78d0
......@@ -75,7 +75,7 @@ def schedule_conv2d_transpose_small_batch(outs):
def traverse(OP):
# inline all one-to-one-mapping operators except the last stage (output)
if tag.is_broadcast(OP.tag):
if tag.is_injective(OP.tag):
if OP not in s.outputs:
s[OP].compute_inline()
for tensor in OP.input_tensors:
......
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