Skip to content

Commit c5c45c8

Browse files
committed
Allowing backward 1D from dense to RNN
1 parent 0479825 commit c5c45c8

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
@@ -33,6 +33,8 @@ pure module subroutine backward_1d(self, previous, gradient)
3333
call this_layer % backward(prev_layer % output, gradient)
3434
type is(flatten_layer)
3535
call this_layer % backward(prev_layer % output, gradient)
36+
type is(rnn_layer)
37+
call this_layer % backward(prev_layer % output, gradient)
3638
end select
3739

3840
type is(flatten_layer)

0 commit comments

Comments
 (0)