grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
ReLU.inplace
grain
functions
unary
ReLU
Undocumented in source.
struct
ReLU
(T, size_t dim)
bool
inplace
;
Meta
Source
See Implementation
grain
functions
unary
ReLU
functions
backward
forward
mixins
__anonymous
variables
dx
dy
hx
inplace
useCuDNN