Skip to content
Snippets Groups Projects
Commit a152a9cb authored by Yuwei Hu's avatar Yuwei Hu Committed by Tianqi Chen
Browse files

[TOPI] modify conv2d_transpose schedule (#613)

parent 685f78d0
No related branches found
No related tags found
No related merge requests found
......@@ -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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment