More fix

nnvm/python/nnvm/top/nn.py

@@ -121,6 +121,13 @@ def compute_conv2d(attrs, inputs, _):
     else:
         raise ValueError("not support arbitrary group number for now")
 
+    if attrs.get_bool("use_bias"):
+        bias = inputs[2]
+        expand_axis = 1 if layout == "NCHW" else 0
+        bias = topi.expand_dims(bias, axis=expand_axis, num_newaxis=2)
+        out = topi.add(out, bias)
+    return out
+
 @reg.register_schedule("conv2d")
 def schedule_conv2d(attrs, outs, target):
     """Schedule definition of conv2d"""