Skip to content

Commit 6c01d6b

Browse files
committed
fix typo
1 parent 0d09cc9 commit 6c01d6b

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

hls4ml/converters/keras/core.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -122,9 +122,8 @@ def parse_batchnorm_layer(keras_layer, input_names, input_shapes, data_reader, c
122122

123123
return layer, [shape for shape in input_shapes[0]]
124124

125-
embedding_layers = ['Embedding']
126-
@keras_handler(*embedding_layers)
127-
def parse_dense_layer(keras_layer, input_names, input_shapes, data_reader, config):
125+
@keras_handler('Embedding')
126+
def parse_embedding_layer(keras_layer, input_names, input_shapes, data_reader, config):
128127
assert('Embedding' in keras_layer['class_name'])
129128

130129
layer = parse_default_keras_layer(keras_layer, input_names)

0 commit comments

Comments
 (0)