File tree Expand file tree Collapse file tree 1 file changed +30
-0
lines changed Expand file tree Collapse file tree 1 file changed +30
-0
lines changed Original file line number Diff line number Diff line change 1+ module nf_keras
2+
3+ ! ! This module provides procedures to read and parse Keras models
4+ ! ! from HDF5 files.
5+
6+ implicit none
7+
8+ private
9+ public :: get_keras_h5_layers
10+
11+ type :: keras_layer
12+ ! ! Intermediate container to convey the Keras layer information
13+ ! ! to neural-fortran layer constructors.
14+ character (:), allocatable :: type
15+ character (:), allocatable :: name
16+ character (:), allocatable :: activation
17+ integer , allocatable :: num_elements(:)
18+ end type keras_layer
19+
20+ interface
21+
22+ module function get_keras_h5_layers (filename ) result(res)
23+ character (* ), intent (in ) :: filename
24+ ! ! HDF5 file name
25+ type (keras_layer), allocatable :: res(:)
26+ end function get_keras_h5_layers
27+
28+ end interface
29+
30+ end module nf_keras
You can’t perform that action at this time.
0 commit comments