We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 23aa24c commit 7f94247Copy full SHA for 7f94247
1 file changed
python/tvm/relay/testing/mlp.py
@@ -50,7 +50,7 @@ def get_net(batch_size,
50
dtype=dtype)
51
data = relay.nn.batch_flatten(data)
52
fc1 = relay.nn.dense(data, relay.var("fc1_weight"), units=128)
53
- fc1 = relay.nn.bias_add(fc1, relay.var("fc2_bias"))
+ fc1 = relay.nn.bias_add(fc1, relay.var("fc1_bias"))
54
act1 = relay.nn.relu(fc1)
55
fc2 = relay.nn.dense(act1, relay.var("fc2_weight"), units=64)
56
fc2 = relay.nn.bias_add(fc2, relay.var("fc2_bias"))
0 commit comments