grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
PoolRefImpl.backward
grain
functions
unary
PoolRefImpl
Undocumented in source. Be warned that the author may not have intended to support it.
struct
PoolRefImpl
(alias poolFun)
static
void
backward
(
T
size_t
poolDims
size_t
tensorDims
)
(
ref
Variable
!(
T
,
tensorDims
,
HostStorage
)
gx
,
Variable
!(
T
,
tensorDims
,
HostStorage
)
x
,
Variable
!(
T
,
tensorDims
,
HostStorage
)
gy
,
Variable
!(
T
,
tensorDims
,
HostStorage
)
y
,
int
[
poolDims
]
windowA
,
int
[
poolDims
]
padA
,
int
[
poolDims
]
strideA
)
Meta
Source
See Implementation
grain
functions
unary
PoolRefImpl
static functions
backward
forward