fix bugs for DCGAN and mobilenet (apache#27)

jroesch · Nov 25, 2018 · c38ad70 · c38ad70
1 parent dc11b26
commit c38ad70
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/tvm/autotvm/task/nnvm_integration.py b/python/tvm/autotvm/task/nnvm_integration.py
@@ -60,8 +60,8 @@ def __init__(self, wanted_symbols):
  # NOTE: To add more symbols, you only need to change the following lists
  # nnvm symbol -> topi compute
  self.symbol2topi = {
- nnvm.sym.conv2d: [topi.nn.conv2d, topi.nn.depthwise_conv2d_nchw,
- topi.nn.group_conv2d_nchw],
+ nnvm.sym.conv2d: [topi.nn.conv2d, topi.nn.depthwise_conv2d_nchw],
+ #topi.nn.group_conv2d_nchw],
  nnvm.sym.conv2d_transpose: [topi.nn.conv2d_transpose_nchw],
  nnvm.sym.dense: [topi.nn.dense],
  }

diff --git a/vta/python/vta/top/op.py b/vta/python/vta/top/op.py
@@ -149,5 +149,5 @@ def schedule_conv2d_transpose(attrs, outs, target):
  raise RuntimeError("not support target %s" % target)
 
  with tvm.target.arm_cpu(tvm.target.current_target().model):
- return _nn.schedule_conv2d_transpose(attrs, outs, target)
+ return _nn.schedule_conv2d_transpose(attrs, outs, tvm.target.current_target())