Skip to content

Commit 7f94247

Browse files
nhynestqchen
authored andcommitted
[Relay] Fix name of bias in testing.mlp (#2892)
1 parent 23aa24c commit 7f94247

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • python/tvm/relay/testing

python/tvm/relay/testing/mlp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def get_net(batch_size,
5050
dtype=dtype)
5151
data = relay.nn.batch_flatten(data)
5252
fc1 = relay.nn.dense(data, relay.var("fc1_weight"), units=128)
53-
fc1 = relay.nn.bias_add(fc1, relay.var("fc2_bias"))
53+
fc1 = relay.nn.bias_add(fc1, relay.var("fc1_bias"))
5454
act1 = relay.nn.relu(fc1)
5555
fc2 = relay.nn.dense(act1, relay.var("fc2_weight"), units=64)
5656
fc2 = relay.nn.bias_add(fc2, relay.var("fc2_bias"))

0 commit comments

Comments
 (0)