Skip to content

Commit e4cb526

Browse files
committed
linear2d_layer: temporarily remove api
1 parent dd1297e commit e4cb526

File tree

4 files changed

+32
-32
lines changed

4 files changed

+32
-32
lines changed

src/nf/nf_layer_constructors.f90

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -185,8 +185,8 @@ module function reshape(output_shape) result(res)
185185
!! Resulting layer instance
186186
end function reshape
187187

188-
module function linear2d(sequence_length, in_features, out_features, batch_size) result(res)
189-
integer, intent(in) :: batch_size, sequence_length, in_features, out_features
188+
module function linear2d(sequence_length, in_features, out_features) result(res)
189+
integer, intent(in) :: sequence_length, in_features, out_features
190190
type(layer) :: res
191191
end function linear2d
192192

src/nf/nf_layer_constructors_submodule.f90

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -149,13 +149,13 @@ module function reshape(output_shape) result(res)
149149

150150
end function reshape
151151

152-
module function linear2d(sequence_length, in_features, out_features, batch_size) result(res)
153-
integer, intent(in) :: batch_size, sequence_length, in_features, out_features
152+
module function linear2d(sequence_length, in_features, out_features) result(res)
153+
integer, intent(in) :: sequence_length, in_features, out_features
154154
type(layer) :: res
155155

156156
res % name = 'linear2d'
157-
res % layer_shape = [sequence_length, out_features, batch_size]
158-
allocate(res % p, source=linear2d_layer(sequence_length, in_features, out_features, batch_size))
157+
res % layer_shape = [sequence_length, out_features]
158+
allocate(res % p, source=linear2d_layer(sequence_length, in_features, out_features))
159159
end function linear2d
160160

161161
end submodule nf_layer_constructors_submodule

src/nf/nf_layer_submodule.f90

Lines changed: 24 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ pure module subroutine backward_1d(self, previous, gradient)
4848
call this_layer % backward(prev_layer % output, gradient)
4949
type is(maxpool2d_layer)
5050
call this_layer % backward(prev_layer % output, gradient)
51-
type is(linear2d_layer)
52-
call this_layer % backward(prev_layer % output, gradient)
51+
! type is(linear2d_layer)
52+
! call this_layer % backward(prev_layer % output, gradient)
5353
end select
5454

5555
end select
@@ -119,11 +119,11 @@ pure module subroutine backward_3d(self, previous, gradient)
119119
call this_layer % backward(prev_layer % output, gradient)
120120
end select
121121

122-
type is(linear2d_layer)
123-
select type(prev_layer => previous % p)
124-
type is(input3d_layer)
125-
call this_layer % backward(prev_layer % output, gradient)
126-
end select
122+
! type is(linear2d_layer)
123+
! select type(prev_layer => previous % p)
124+
! type is(input3d_layer)
125+
! call this_layer % backward(prev_layer % output, gradient)
126+
! end select
127127

128128
end select
129129

@@ -191,8 +191,8 @@ pure module subroutine forward(self, input)
191191
call this_layer % forward(prev_layer % output)
192192
type is(reshape3d_layer)
193193
call this_layer % forward(prev_layer % output)
194-
type is(linear2d_layer)
195-
call this_layer % forward(prev_layer % output)
194+
! type is(linear2d_layer)
195+
! call this_layer % forward(prev_layer % output)
196196
end select
197197

198198
type is(reshape3d_layer)
@@ -207,13 +207,13 @@ pure module subroutine forward(self, input)
207207
call this_layer % forward(prev_layer % output)
208208
end select
209209

210-
type is(linear2d_layer)
211-
select type(prev_layer => input % p)
212-
type is(input3d_layer)
213-
call this_layer % forward(prev_layer % output)
214-
type is(linear2d_layer)
215-
call this_layer % forward(prev_layer % output)
216-
end select
210+
! type is(linear2d_layer)
211+
! select type(prev_layer => input % p)
212+
! type is(input3d_layer)
213+
! call this_layer % forward(prev_layer % output)
214+
! type is(linear2d_layer)
215+
! call this_layer % forward(prev_layer % output)
216+
! end select
217217

218218
end select
219219

@@ -347,8 +347,8 @@ elemental module function get_num_params(self) result(num_params)
347347
num_params = 0
348348
type is (reshape3d_layer)
349349
num_params = 0
350-
type is (linear2d_layer)
351-
num_params = this_layer % get_num_params()
350+
! type is (linear2d_layer)
351+
! num_params = this_layer % get_num_params()
352352
class default
353353
error stop 'Unknown layer type.'
354354
end select
@@ -376,8 +376,8 @@ module function get_params(self) result(params)
376376
! No parameters to get.
377377
type is (reshape3d_layer)
378378
! No parameters to get.
379-
type is (linear2d_layer)
380-
params = this_layer % get_params()
379+
! type is (linear2d_layer)
380+
! params = this_layer % get_params()
381381
class default
382382
error stop 'Unknown layer type.'
383383
end select
@@ -405,8 +405,8 @@ module function get_gradients(self) result(gradients)
405405
! No gradients to get.
406406
type is (reshape3d_layer)
407407
! No gradients to get.
408-
type is (linear2d_layer)
409-
gradients = this_layer % get_gradients()
408+
! type is (linear2d_layer)
409+
! gradients = this_layer % get_gradients()
410410
class default
411411
error stop 'Unknown layer type.'
412412
end select
@@ -472,8 +472,8 @@ module subroutine set_params(self, params)
472472
class default
473473
error stop 'Unknown layer type.'
474474

475-
type is (linear2d_layer)
476-
call this_layer % set_params(params)
475+
! type is (linear2d_layer)
476+
! call this_layer % set_params(params)
477477
end select
478478

479479
end subroutine set_params

src/nf/nf_network_submodule.f90

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -157,8 +157,8 @@ module subroutine backward(self, output, loss)
157157

158158
type is(reshape3d_layer)
159159
call self % layers(n) % backward(self % layers(n - 1), next_layer % gradient)
160-
type is(linear2d_layer)
161-
call self % layers(n) % backward(self % layers(n - 1), next_layer % gradient)
160+
! type is(linear2d_layer)
161+
! call self % layers(n) % backward(self % layers(n - 1), next_layer % gradient)
162162
end select
163163
end if
164164

0 commit comments

Comments
 (0)