grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
Pool.forward
grain
functions
unary
Pool
Undocumented in source. Be warned that the author may not have intended to support it.
auto
forward
(Variable!(T, dim, HostStorage) x)
struct
Pool
(bool isMax, T, size_t poolDims)
auto
forward
(
Variable
!(
T
,
dim
,
HostStorage
)
x
)
auto
forward
(Variable!(T, dim, DeviceStorage) x)
Meta
Source
See Implementation
grain
functions
unary
Pool
aliases
CpuImpl
functions
backward
forward
manifest constants
dim
static functions
pool
variables
dx
dy
hx
hy
pad
stride
window