Skip to content

Commit 1bdcf5e

Browse files
committed
Allowing forward from dense to RNN
1 parent d4e55b6 commit 1bdcf5e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/nf/nf_layer_submodule.f90

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,8 @@ pure module subroutine forward(self, input)
139139
call this_layer % forward(prev_layer % output)
140140
type is(flatten_layer)
141141
call this_layer % forward(prev_layer % output)
142+
type is(rnn_layer)
143+
call this_layer % forward(prev_layer % output)
142144
end select
143145

144146
type is(conv2d_layer)

0 commit comments

Comments
 (0)