Skip to content

Commit e02b511

Browse files
authored
Merge pull request #1353 from JanFSchulte/timeDistFix
Fix missing return statement in time distributed parser
2 parents 1d1d53a + 46832a0 commit e02b511

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

hls4ml/converters/keras/recurrent.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,8 @@ def parse_time_distributed_layer(keras_layer, input_names, input_shapes, data_re
110110
layer['output_shape'] = output_shape[1:] # Remove the batch dimension
111111
layer['n_time_steps'] = output_shape[1]
112112

113+
return layer, output_shape
114+
113115

114116
@keras_handler('Bidirectional')
115117
def parse_bidirectional_layer(keras_layer, input_names, input_shapes, data_reader):

0 commit comments

Comments
 (0)