fixed

tensorflow · Sep 14, 2023 · 15b142d · 15b142d
1 parent a3d4f8f
commit 15b142d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/codegen/examples/hello_world/hello_world_model.cc b/codegen/examples/hello_world/hello_world_model.cc
@@ -122,6 +122,7 @@ struct Node0_0 {
       .activation = kTfLiteActRelu,
       .weights_format = kTfLiteFullyConnectedWeightsFormatDefault,
       .keep_num_dims = false,
+      .quantized_bias_type = 0,
       .asymmetric_quantize_inputs = false};
 } node_0_0;
 
@@ -139,6 +140,7 @@ struct Node0_1 {
       .activation = kTfLiteActRelu,
       .weights_format = kTfLiteFullyConnectedWeightsFormatDefault,
       .keep_num_dims = false,
+      .quantized_bias_type = 0,
       .asymmetric_quantize_inputs = false};
 } node_0_1;
 
@@ -156,6 +158,7 @@ struct Node0_2 {
       .activation = kTfLiteActNone,
       .weights_format = kTfLiteFullyConnectedWeightsFormatDefault,
       .keep_num_dims = false,
+      .quantized_bias_type = 0,
       .asymmetric_quantize_inputs = false};
 } node_0_2;
 

diff --git a/codegen/operators/fully_connected.py b/codegen/operators/fully_connected.py
@@ -51,6 +51,6 @@ def generate_c_builtin_data(self) -> str:
         weights_format=_WEIGHTS_FORMATS[self._builtin_options.weightsFormat],
         keep_num_dims=utils.bool_to_c_str(self._builtin_options.keepNumDims),
         quantized_bias_type=utils.bool_to_c_str(
-            self._builtin_options.quantized_bias_type),
+            self._builtin_options.quantizedBiasType),
         asymmetric_quantize_inputs=utils.bool_to_c_str(
             self._builtin_options.asymmetricQuantizeInputs))