grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
Convolution.forward
grain
functions
binary
Convolution
Undocumented in source. Be warned that the author may not have intended to support it.
auto
forward
(Variable!(T, nbDims, HostStorage) x, Variable!(T, nbDims, HostStorage) w)
struct
Convolution
(T, size_t imDims, bool isConv = false, bool isNchw = true)
auto
forward
(
Variable
!(
T
,
nbDims
,
HostStorage
)
x
,
Variable
!(
T
,
nbDims
,
HostStorage
)
w
)
auto
forward
(Variable!(T, nbDims, DeviceStorage) x, Variable!(T, nbDims, DeviceStorage) w)
Meta
Source
See Implementation
grain
functions
binary
Convolution
aliases
RefImpl
functions
backward
forward
outShape
setDefault
mixins
__anonymous
variables
backwardAlgo
dilation
dw
dx
forwardAlgo
hw
hx
nbDims
ngroup
pad
stride