From 7abdb37caaad54a161abdf66d8b35e992d163011 Mon Sep 17 00:00:00 2001 From: Nick Hynes Date: Mon, 25 Mar 2019 13:45:05 -0700 Subject: [PATCH] [Relay] Fix name of bias in testing.mlp (#2892) --- python/tvm/relay/testing/mlp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/testing/mlp.py b/python/tvm/relay/testing/mlp.py index 7d7d984f7526..3eac0a8f2654 100644 --- a/python/tvm/relay/testing/mlp.py +++ b/python/tvm/relay/testing/mlp.py @@ -50,7 +50,7 @@ def get_net(batch_size, dtype=dtype) data = relay.nn.batch_flatten(data) fc1 = relay.nn.dense(data, relay.var("fc1_weight"), units=128) - fc1 = relay.nn.bias_add(fc1, relay.var("fc2_bias")) + fc1 = relay.nn.bias_add(fc1, relay.var("fc1_bias")) act1 = relay.nn.relu(fc1) fc2 = relay.nn.dense(act1, relay.var("fc2_weight"), units=64) fc2 = relay.nn.bias_add(fc2, relay.var("fc2_bias"))