grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
ConvolutionRefImpl.backwardWeight
grain
functions
binary
ConvolutionRefImpl
Undocumented in source.
struct
ConvolutionRefImpl
(T, size_t imDims, bool isConv = false, bool isNchw = true)
static
void
backwardWeight
(
const
T
[]
image
,
const
T
[]
diffData
,
float
alpha
,
float
beta
,
scope
T
[]
output
,
const
int
[
nbDims
]
inDims
,
const
int
[
nbDims
]
filDims
,
const
int
[
nbDims
]
diffDims
,
const
int
[
nbDims
]
inStride
,
const
int
[
nbDims
]
filterStride
,
const
int
[
nbDims
]
diffStride
,
const
int
[
imDims
]
stride
,
const
int
[
imDims
]
pad
,
const
int
[
imDims
]
dilation
)
Meta
Source
See Implementation
grain
functions
binary
ConvolutionRefImpl
static functions
backwardData
backwardWeight
forward
variables
nbDims