Skip to content

Commit 9dbcbde

Browse files
committed
add parsing of depth multiplier for 1D depthwise conv
1 parent 86b0c40 commit 9dbcbde

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

hls4ml/converters/keras/convolution.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,9 @@ def parse_conv1d_layer(keras_layer, input_names, input_shapes, data_reader):
2121

2222
layer['bias_data'] = get_weights_data(data_reader, layer['name'], 'bias')
2323

24+
if 'depth_multiplier' in keras_layer['config']:
25+
layer['depth_multiplier'] = keras_layer['config']['depth_multiplier']
26+
2427
if 'filters' in keras_layer['config']:
2528
layer['n_filt'] = keras_layer['config']['filters']
2629
else:

0 commit comments

Comments
 (0)