Skip to content

Commit 31e4e54

Browse files
committed
support serializable hgq objects (tf>=2.13)
1 parent 1e3ef52 commit 31e4e54

File tree

3 files changed

+5
-2
lines changed

3 files changed

+5
-2
lines changed

hls4ml/converters/keras/hgq_proxy_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from hls4ml.converters.keras_to_hls import KerasReader, keras_handler, parse_default_keras_layer
22

33

4-
@keras_handler('FixedPointQuantizer')
4+
@keras_handler('FixedPointQuantizer', 'HGQ>FixedPointQuantizer')
55
def fixedpoint_quantizer_handler(keras_layer, input_names, input_shapes, data_reader: KerasReader):
66
config = parse_default_keras_layer(keras_layer, input_names)
77

@@ -21,7 +21,7 @@ def fixedpoint_quantizer_handler(keras_layer, input_names, input_shapes, data_re
2121
return layer, input_shapes[0]
2222

2323

24-
@keras_handler('UnaryLUT')
24+
@keras_handler('UnaryLUT', 'HGQ>UnaryLUT')
2525
def unary_lut_keras_handler(keras_layer, input_names, input_shapes, data_reader: KerasReader):
2626
config = parse_default_keras_layer(keras_layer, input_names)
2727

hls4ml/converters/keras_to_hls.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -206,6 +206,7 @@ def parse_keras_model(model_arch, reader):
206206
'TernaryTanh',
207207
'HardActivation',
208208
'UnaryLUT',
209+
'HGQ>UnaryLUT',
209210
]
210211
# Recurrent layers
211212
recurrent_layers = ['SimpleRNN', 'LSTM', 'GRU']

hls4ml/model/optimizer/passes/hgq_proxy_model.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -145,5 +145,7 @@ def transform(self, model, node: FixedPointQuantizer):
145145

146146
def register_hgq_proxy_model():
147147
register_layer('FixedPointQuantizer', FixedPointQuantizer)
148+
register_layer('HGQ>FixedPointQuantizer', FixedPointQuantizer)
148149
register_layer('UnaryLUT', UnaryLUT)
150+
register_layer('HGQ>UnaryLUT', UnaryLUT)
149151
register_pass('enforce_proxy_model_embedded_config', EnforceProxyModelEmbeddedConfig)

0 commit comments

Comments
 (0)