Skip to content

Commit 14a7b20

Browse files
committed
Enable conv2d and maxpool2d layers as input to conv2d layer
1 parent 2b8ccc6 commit 14a7b20

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/nf_layer_submodule.f90

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,10 +52,14 @@ pure module subroutine forward(self, input)
5252

5353
type is(conv2d_layer)
5454

55-
! Input layers permitted: input3d
55+
! Input layers permitted: input3d, conv2d, maxpool2d
5656
select type(prev_layer => input % p)
5757
type is(input3d_layer)
5858
call this_layer % forward(prev_layer % output)
59+
type is(conv2d_layer)
60+
call this_layer % forward(prev_layer % output)
61+
type is(maxpool2d_layer)
62+
call this_layer % forward(prev_layer % output)
5963
end select
6064

6165
type is(maxpool2d_layer)

0 commit comments

Comments
 (0)