fix upsampling fuse pattern (#330)

82de081c · masahi · Tianqi Chen · 9998f330 · 82de081c
Commit 82de081c authored Jan 20, 2018 by masahi Committed by Tianqi Chen May 29, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

nnvm/python/nnvm/top/nn.py
+1 -1

No files found.
--- a/nnvm/python/nnvm/top/nn.py
+++ b/nnvm/python/nnvm/top/nn.py
@@ -264,4 +264,4 @@ def schedule_upsampling(_, outs, target):
    with tvm.target.create(target):
        return topi.generic.schedule_injective(outs)

-reg.register_pattern("upsampling", OpPattern.OUT_ELEMWISE_FUSABLE)
+reg.register_pattern("upsampling", OpPattern.INJECTIVE)