grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
ReLU.backward
grain
functions
unary
ReLU
Undocumented in source. Be warned that the author may not have intended to support it.
auto
backward
(Variable!(T, dim, HostStorage) gy)
struct
ReLU
(T, size_t dim)
auto
backward
(
Variable
!(
T
,
dim
,
HostStorage
)
gy
)
auto
backward
(Variable!(T, dim, DeviceStorage) gy)
Meta
Source
See Implementation
grain
functions
unary
ReLU
functions
backward
forward
mixins
__anonymous
variables
dx
dy
hx
inplace
useCuDNN