Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dlib
Commits
7f53d7fe
Unverified
Commit
7f53d7fe
authored
Apr 13, 2021
by
Adrià Arrufat
Committed by
GitHub
Apr 12, 2021
Browse files
Make clipped-relu inplace and fix docs for elu (#2345)
parent
1b7c7a64
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
13 deletions
+8
-13
dlib/dnn/layers.h
dlib/dnn/layers.h
+6
-11
dlib/dnn/layers_abstract.h
dlib/dnn/layers_abstract.h
+2
-2
No files found.
dlib/dnn/layers.h
View file @
7f53d7fe
...
@@ -3479,24 +3479,19 @@ namespace dlib
...
@@ -3479,24 +3479,19 @@ namespace dlib
{
{
}
}
template
<
typename
SUBNET
>
void
forward_inplace
(
const
tensor
&
input
,
tensor
&
output
)
void
forward
(
SUBNET
&
sub
,
resizable_tensor
&
data_output
)
{
{
data_output
.
copy_size
(
sub
.
get_output
());
tt
::
clipped_relu
(
output
,
input
,
ceiling
);
tt
::
clipped_relu
(
data_output
,
sub
.
get_output
(),
ceiling
);
}
}
template
<
typename
SUBNET
>
void
backward_inplace
(
void
backward
(
const
tensor
&
computed_output
,
const
tensor
&
gradient_input
,
const
tensor
&
gradient_input
,
SUBNET
&
sub
,
tensor
&
data_grad
,
tensor
&
tensor
&
)
)
{
{
tt
::
clipped_relu_gradient
(
sub
.
get_gradient_input
(),
sub
.
get
_output
()
,
gradient_input
,
ceiling
);
tt
::
clipped_relu_gradient
(
data_grad
,
computed
_output
,
gradient_input
,
ceiling
);
}
}
inline
dpoint
map_input_to_output
(
const
dpoint
&
p
)
const
{
return
p
;
}
inline
dpoint
map_input_to_output
(
const
dpoint
&
p
)
const
{
return
p
;
}
...
...
dlib/dnn/layers_abstract.h
View file @
7f53d7fe
...
@@ -2537,8 +2537,8 @@ namespace dlib
...
@@ -2537,8 +2537,8 @@ namespace dlib
- returns the alpha parameter of the elu
- returns the alpha parameter of the elu
!*/
!*/
template
<
typename
SUBNET
>
void
setup
(
const
SUBNET
&
sub
);
template
<
typename
SUBNET
>
void
setup
(
const
SUBNET
&
sub
);
void
forward_inplace
(
const
tensor
&
input
,
tensor
&
output
);
template
<
typename
SUBNET
>
void
forward
(
const
SUBNET
&
sub
,
resizable_
tensor
&
data_
output
);
void
backward
_inplace
(
const
tensor
&
computed_output
,
const
tensor
&
gradient_input
,
tensor
&
data_grad
,
tensor
&
params_grad
);
template
<
typename
SUBNET
>
void
backward
(
const
tensor
&
gradient_input
,
SUBNET
&
sub
,
tensor
&
);
dpoint
map_input_to_output
(
dpoint
p
)
const
;
dpoint
map_input_to_output
(
dpoint
p
)
const
;
dpoint
map_output_to_input
(
dpoint
p
)
const
;
dpoint
map_output_to_input
(
dpoint
p
)
const
;
const
tensor
&
get_layer_params
()
const
;
const
tensor
&
get_layer_params
()
const
;
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment