Skip to content

Commit e8fae54

Browse files
author
Enrico Lupi
committed
FIX posible directions for LSTM and GRU
1 parent b55cd04 commit e8fae54

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

hls4ml/model/layers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1353,7 +1353,7 @@ class LSTM(Layer):
13531353
Attribute('return_sequences', value_type=bool, default=False),
13541354
Attribute('return_state', value_type=bool, default=False),
13551355
Attribute('pass_initial_states', value_type=bool, default=False),
1356-
ChoiceAttribute('direction', ['forward', 'backward', 'bidirectional'], configurable=False, default='forward'),
1356+
ChoiceAttribute('direction', ['forward', 'backward'], configurable=False, default='forward'),
13571357
Attribute('time_major', value_type=bool, default=False),
13581358
WeightAttribute('weight'),
13591359
WeightAttribute('bias'),
@@ -1410,7 +1410,7 @@ class GRU(Layer):
14101410
Attribute('return_sequences', value_type=bool, default=False),
14111411
Attribute('return_state', value_type=bool, default=False),
14121412
Attribute('pass_initial_states', value_type=bool, default=False),
1413-
ChoiceAttribute('direction', ['forward', 'backward', 'bidirectional'], configurable=False, default='forward'),
1413+
ChoiceAttribute('direction', ['forward', 'backward'], configurable=False, default='forward'),
14141414
Attribute('time_major', value_type=bool, default=False),
14151415
ChoiceAttribute('apply_reset_gate', ['before', 'after'], configurable=False, default='after'),
14161416
WeightAttribute('weight'),

0 commit comments

Comments
 (0)