generator change

tensorflow · Sep 14, 2023 · a3d4f8f · a3d4f8f
1 parent 00d55c5
commit a3d4f8f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/codegen/operators/fully_connected.py b/codegen/operators/fully_connected.py
@@ -43,11 +43,14 @@ def generate_c_builtin_data(self) -> str:
         "    .activation = ${activation},\n"
         "    .weights_format = ${weights_format},\n"
         "    .keep_num_dims = ${keep_num_dims},\n"
+        "    .quantized_bias_type = ${quantized_bias_type},\n"
         "    .asymmetric_quantize_inputs = ${asymmetric_quantize_inputs}};")
     return builtin_template.substitute(
         activation=constants.ACTIVATION_FUNCS[
             self._builtin_options.fusedActivationFunction],
         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),
         asymmetric_quantize_inputs=utils.bool_to_c_str(
             self._builtin_options.asymmetricQuantizeInputs))