grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
Convolution.hx
grain
functions
binary
Convolution
Undocumented in source.
Variable
!(
T
,
nbDims
,
HostStorage
)
hx
;
struct
Convolution
(T, size_t imDims, bool isConv = false, bool isNchw = true)
Variable
!(
T
,
nbDims
,
HostStorage
)
hx
;
Variable
!(
T
,
nbDims
,
HostStorage
)
hw
;
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