Skip to content

Commit

Permalink
Extenting uses
Browse files Browse the repository at this point in the history
  • Loading branch information
castelao committed Sep 15, 2024
1 parent 64e1b69 commit 8c11911
Show file tree
Hide file tree
Showing 4 changed files with 9 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/nf/nf_layer_constructors.f90
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ module nf_layer_constructors
implicit none

private
public :: conv2d, dense, flatten, input, maxpool2d, reshape
public :: conv2d, dense, flatten, input, maxpool2d, reshape, rnn

interface input

Expand Down
1 change: 1 addition & 0 deletions src/nf/nf_layer_constructors_submodule.f90
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
use nf_input3d_layer, only: input3d_layer
use nf_maxpool2d_layer, only: maxpool2d_layer
use nf_reshape_layer, only: reshape3d_layer
use nf_rnn_layer, only: rnn_layer
use nf_activation, only: activation_function, relu, sigmoid

implicit none
Expand Down
3 changes: 3 additions & 0 deletions src/nf/nf_layer_submodule.f90
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
use nf_input3d_layer, only: input3d_layer
use nf_maxpool2d_layer, only: maxpool2d_layer
use nf_reshape_layer, only: reshape3d_layer
use nf_rnn_layer, only: rnn_layer
use nf_optimizers, only: optimizer_base_type

contains
Expand Down Expand Up @@ -292,6 +293,8 @@ elemental module function get_num_params(self) result(num_params)
num_params = 0
type is (reshape3d_layer)
num_params = 0
type is (rnn_layer)
num_params = this_layer % get_num_params()
class default
error stop 'Unknown layer type.'
end select
Expand Down
5 changes: 4 additions & 1 deletion src/nf/nf_network_submodule.f90
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,11 @@
use nf_input3d_layer, only: input3d_layer
use nf_maxpool2d_layer, only: maxpool2d_layer
use nf_reshape_layer, only: reshape3d_layer
use nf_rnn_layer, only: rnn_layer
use nf_io_hdf5, only: get_hdf5_dataset
use nf_keras, only: get_keras_h5_layers, keras_layer
use nf_layer, only: layer
use nf_layer_constructors, only: conv2d, dense, flatten, input, maxpool2d, reshape
use nf_layer_constructors, only: conv2d, dense, flatten, input, maxpool2d, reshape, rnn
use nf_loss, only: quadratic
use nf_optimizers, only: optimizer_base_type, sgd
use nf_parallel, only: tile_indices
Expand Down

0 comments on commit 8c11911

Please sign in to comment.