Skip to content

Commit 8876c99

Browse files
committed
Enable conv2d->conv2d in layer % forward() method
1 parent 6eaf191 commit 8876c99

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/nf_layer_submodule.f90

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,12 @@ pure module subroutine forward(self, input)
5151

5252
type is(conv2d_layer)
5353

54-
! Input layers permitted: input3d
54+
! Input layers permitted: input3d, conv2d
5555
select type(prev_layer => input % p)
5656
type is(input3d_layer)
5757
call this_layer % forward(prev_layer % output)
58+
type is(conv2d_layer)
59+
call this_layer % forward(prev_layer % output)
5860
end select
5961

6062
end select

0 commit comments

Comments
 (0)