grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
ConvolutionRefImpl.forward
grain
functions
binary
ConvolutionRefImpl
Undocumented in source.
struct
ConvolutionRefImpl
(T, size_t imDims, bool isConv = false, bool isNchw = true)
static
void
forward
(
const
T
[]
inputData
,
const
T
[]
filterData
,
T
[]
outputData
,
T
alpha
,
T
beta
,
const
int
[
nbDims
]
inDims
,
const
int
[
nbDims
]
filDims
,
const
int
[
nbDims
]
outDims
,
const
int
[
nbDims
]
inStride
,
const
int
[
nbDims
]
filStride
,
const
int
[
nbDims
]
outStride
,
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