fix bugs for DCGAN and mobilenet (apache#27)

tmoreau89 · Jan 2, 2019 · bb6b024 · bb6b024
1 parent bbfbde3
commit bb6b024
Showing 1 changed file with 1 addition and 1 deletion.
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())