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