Skip to content

T
tic
Commit 82de081c by masahi Committed by Tianqi Chen
Options

fix upsampling fuse pattern (#330)

parent 9998f330
Showing with 1 additions and 1 deletions
nnvm/python/nnvm/top/nn.py
...@@ -264,4 +264,4 @@ def schedule_upsampling(_, outs, target): ...@@ -264,4 +264,4 @@ def schedule_upsampling(_, outs, target):
with tvm.target.create(target): with tvm.target.create(target):
return topi.generic.schedule_injective(outs) return topi.generic.schedule_injective(outs)
reg.register_pattern("upsampling", OpPattern.OUT_ELEMWISE_FUSABLE) reg.register_pattern("upsampling", OpPattern.INJECTIVE)
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