grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
Pool.dy
grain
functions
unary
Pool
Undocumented in source.
Variable
!(
T
,
dim
,
DeviceStorage
)
dx
;
Variable
!(
T
,
dim
,
DeviceStorage
)
dy
;
struct
Pool
(bool isMax, T, size_t poolDims)
version(grain_cuda)
Variable
!(
T
,
dim
,
DeviceStorage
)
dy
;
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