99
1010
1111def get_quantizer_from_config (keras_layer , quantizer_var ):
12- quantizer_config = keras_layer ['config' ][f'{ quantizer_var } _quantizer' ]
12+ quantizer_config = keras_layer ['config' ].get (f'{ quantizer_var } _quantizer' , None )
13+ if quantizer_config is None :
14+ return None # No quantizer specified in the layer
1315 if keras_layer ['class_name' ] == 'QBatchNormalization' :
1416 return QKerasQuantizer (quantizer_config )
1517 elif 'binary' in quantizer_config ['class_name' ]:
@@ -25,10 +27,7 @@ def parse_qdense_layer(keras_layer, input_names, input_shapes, data_reader):
2527 layer , output_shape = parse_dense_layer (keras_layer , input_names , input_shapes , data_reader )
2628
2729 layer ['weight_quantizer' ] = get_quantizer_from_config (keras_layer , 'kernel' )
28- if keras_layer ['config' ]['bias_quantizer' ] is not None :
29- layer ['bias_quantizer' ] = get_quantizer_from_config (keras_layer , 'bias' )
30- else :
31- layer ['bias_quantizer' ] = None
30+ layer ['bias_quantizer' ] = get_quantizer_from_config (keras_layer , 'bias' )
3231
3332 return layer , output_shape
3433
@@ -43,10 +42,7 @@ def parse_qconv_layer(keras_layer, input_names, input_shapes, data_reader):
4342 layer , output_shape = parse_conv2d_layer (keras_layer , input_names , input_shapes , data_reader )
4443
4544 layer ['weight_quantizer' ] = get_quantizer_from_config (keras_layer , 'kernel' )
46- if keras_layer ['config' ]['bias_quantizer' ] is not None :
47- layer ['bias_quantizer' ] = get_quantizer_from_config (keras_layer , 'bias' )
48- else :
49- layer ['bias_quantizer' ] = None
45+ layer ['bias_quantizer' ] = get_quantizer_from_config (keras_layer , 'bias' )
5046
5147 return layer , output_shape
5248
@@ -56,11 +52,7 @@ def parse_qdepthwiseqconv_layer(keras_layer, input_names, input_shapes, data_rea
5652 layer , output_shape = parse_conv2d_layer (keras_layer , input_names , input_shapes , data_reader )
5753
5854 layer ['depthwise_quantizer' ] = get_quantizer_from_config (keras_layer , 'depthwise' )
59-
60- if keras_layer ['config' ]['bias_quantizer' ] is not None :
61- layer ['bias_quantizer' ] = get_quantizer_from_config (keras_layer , 'bias' )
62- else :
63- layer ['bias_quantizer' ] = None
55+ layer ['bias_quantizer' ] = get_quantizer_from_config (keras_layer , 'bias' )
6456
6557 return layer , output_shape
6658
@@ -76,11 +68,7 @@ def parse_qsepconv_layer(keras_layer, input_names, input_shapes, data_reader):
7668
7769 layer ['depthwise_quantizer' ] = get_quantizer_from_config (keras_layer , 'depthwise' )
7870 layer ['pointwise_quantizer' ] = get_quantizer_from_config (keras_layer , 'pointwise' )
79-
80- if keras_layer ['config' ]['bias_quantizer' ] is not None :
81- layer ['bias_quantizer' ] = get_quantizer_from_config (keras_layer , 'bias' )
82- else :
83- layer ['bias_quantizer' ] = None
71+ layer ['bias_quantizer' ] = get_quantizer_from_config (keras_layer , 'bias' )
8472
8573 return layer , output_shape
8674
@@ -93,11 +81,7 @@ def parse_qrnn_layer(keras_layer, input_names, input_shapes, data_reader):
9381
9482 layer ['weight_quantizer' ] = get_quantizer_from_config (keras_layer , 'kernel' )
9583 layer ['recurrent_quantizer' ] = get_quantizer_from_config (keras_layer , 'recurrent' )
96-
97- if keras_layer ['config' ]['bias_quantizer' ] is not None :
98- layer ['bias_quantizer' ] = get_quantizer_from_config (keras_layer , 'bias' )
99- else :
100- layer ['bias_quantizer' ] = None
84+ layer ['bias_quantizer' ] = get_quantizer_from_config (keras_layer , 'bias' )
10185
10286 return layer , output_shape
10387
0 commit comments