Skip to content

Commit

Permalink
generator change
Browse files Browse the repository at this point in the history
  • Loading branch information
turbotoribio committed Sep 14, 2023
1 parent 00d55c5 commit a3d4f8f
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions codegen/operators/fully_connected.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))

0 comments on commit a3d4f8f

Please sign in to comment.