Skip to content

Commit 4a550bb

Browse files
junminjangandifeCopilot
authored
Fix typo in variable name 'bn_bais' to 'bn_bias' (#299)
Signed-off-by: junminjjang <junminjang@hanyang.ac.kr> Signed-off-by: Andreas Fehlner <fehlner@arcor.de> Co-authored-by: Andreas Fehlner <fehlner@arcor.de> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
1 parent cf1ef8e commit 4a550bb

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

onnxoptimizer/passes/fuse_bn_into_conv.h

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -52,24 +52,24 @@ struct FuseBNIntoConv final : public PredicateBasedPass {
5252
const auto& conv_inputs = conv->inputs();
5353

5454
auto bn_scale = *FetchConstantTensor(bn_inputs[1]);
55-
auto bn_bais = *FetchConstantTensor(bn_inputs[2]);
55+
auto bn_bias = *FetchConstantTensor(bn_inputs[2]);
5656
auto bn_mean = *FetchConstantTensor(bn_inputs[3]);
5757
auto bn_var = *FetchConstantTensor(bn_inputs[4]);
5858
auto conv_W = *FetchConstantTensor(conv_inputs[1]);
5959
bn_scale.setName(graph.getNextUniqueName());
60-
bn_bais.setName(graph.getNextUniqueName());
60+
bn_bias.setName(graph.getNextUniqueName());
6161
bn_mean.setName(graph.getNextUniqueName());
6262
bn_var.setName(graph.getNextUniqueName());
6363
conv_W.setName(graph.getNextUniqueName());
6464

65-
/// scale bais mean var must be the same shape (C)
66-
ONNX_ASSERT(bn_scale.sizes() == bn_bais.sizes());
65+
/// scale bias mean var must be the same shape (C)
66+
ONNX_ASSERT(bn_scale.sizes() == bn_bias.sizes());
6767
ONNX_ASSERT(bn_scale.sizes() == bn_mean.sizes());
6868
ONNX_ASSERT(bn_scale.sizes() == bn_var.sizes());
6969
ONNX_ASSERT(bn_scale.sizes().size() == 1);
7070
int64_t C = bn_scale.sizes()[0];
7171
ONNX_ASSERT(conv_W.sizes().size() > 2 && conv_W.sizes()[0] == C);
72-
if (bn_scale.elem_type() != bn_bais.elem_type() ||
72+
if (bn_scale.elem_type() != bn_bias.elem_type() ||
7373
bn_scale.elem_type() != bn_mean.elem_type() ||
7474
bn_scale.elem_type() != bn_var.elem_type() ||
7575
bn_scale.elem_type() != conv_W.elem_type()) {
@@ -160,7 +160,7 @@ struct FuseBNIntoConv final : public PredicateBasedPass {
160160
Node* bias_add = graph.create(kAdd, 1);
161161
bias_add->insertAfter(mul);
162162
bias_add->addInput(mul->output());
163-
bias_add->addInput(graph.addInitializerAndCreateValue(bn_bais));
163+
bias_add->addInput(graph.addInitializerAndCreateValue(bn_bias));
164164

165165
Value* old_w_value = conv_inputs[1];
166166
conv->replaceInput(1, mul_w->output());

0 commit comments

Comments
 (0)