Skip to content

Commit bc96589

Browse files
committed
Allowing forward from RNN
1 parent 1bdcf5e commit bc96589

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

src/nf/nf_layer_submodule.f90

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,19 @@ pure module subroutine forward(self, input)
197197
call this_layer % forward(prev_layer % output)
198198
end select
199199

200+
type is(rnn_layer)
201+
202+
! Upstream layers permitted: input1d, dense, rnn
203+
select type(prev_layer => input % p)
204+
type is(input1d_layer)
205+
call this_layer % forward(prev_layer % output)
206+
type is(dense_layer)
207+
call this_layer % forward(prev_layer % output)
208+
type is(rnn_layer)
209+
call this_layer % forward(prev_layer % output)
210+
end select
211+
212+
200213
end select
201214

202215
end subroutine forward

0 commit comments

Comments
 (0)