Skip to content

Commit 7528043

Browse files
committed
make reshape handler compatiable with keras v3
1 parent 6b8a44c commit 7528043

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hls4ml/converters/keras/reshape.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def parse_reshape_layer(keras_layer, input_names, input_shapes, data_reader):
2424
layer = parse_default_keras_layer(keras_layer, input_names)
2525

2626
layer['target_shape'] = keras_layer['config']['target_shape']
27-
output_shape = input_shapes[0][:1] + keras_layer['config']['target_shape']
27+
output_shape = input_shapes[0][:1] + list(keras_layer['config']['target_shape'])
2828

2929
return layer, output_shape
3030

0 commit comments

Comments
 (0)