@@ -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