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