Skip to content

Commit b942637

Browse files
committed
Resolved merge conflicts
1 parent c7e682b commit b942637

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

src/nf/nf_layer_submodule.f90

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -367,6 +367,16 @@ pure module subroutine get_output_2d(self, output)
367367

368368
type is(input2d_layer)
369369
allocate(output, source=this_layer % output)
370+
type is(maxpool1d_layer)
371+
allocate(output, source=this_layer % output)
372+
type is(locally_connected_1d_layer)
373+
allocate(output, source=this_layer % output)
374+
type is(reshape2d_layer)
375+
allocate(output, source=this_layer % output)
376+
type is(linear2d_layer)
377+
allocate(output, source=this_layer % output)
378+
type is(self_attention_layer)
379+
allocate(output, source=this_layer % output)
370380
class default
371381
error stop '2-d output can only be read from an input2d or linear2d layer.'
372382

@@ -557,6 +567,8 @@ module function get_gradients(self) result(gradients)
557567
! No gradients to get.
558568
type is (flatten_layer)
559569
! No gradients to get.
570+
type is (reshape2d_layer)
571+
! No parameters to get.
560572
type is (reshape3d_layer)
561573
! No gradients to get.
562574
type is (linear2d_layer)

0 commit comments

Comments
 (0)