ConvolutionRefImpl

Reference CPU implementation of Convolution function

static

struct ConvolutionRefImpl (

size_t imDims

bool isConv = false

bool isNchw = true

) {

}

Members

backwardData void backwardData(T[] weight, T[] top_diff, T[] output, float alpha, float beta, int[nbDims] inDims, int[nbDims] filDims, int[nbDims] outDims, int[nbDims] inStride, int[nbDims] filterStride, int[nbDims] outStride, int[imDims] stride, int[imDims] pad, int[imDims] dilation): Undocumented in source.
backwardWeight void backwardWeight(T[] image, T[] diffData, float alpha, float beta, T[] output, int[nbDims] inDims, int[nbDims] filDims, int[nbDims] diffDims, int[nbDims] inStride, int[nbDims] filterStride, int[nbDims] diffStride, int[imDims] stride, int[imDims] pad, int[imDims] dilation): Undocumented in source.
forward void forward(T[] inputData, T[] filterData, T[] outputData, T alpha, T beta, int[nbDims] inDims, int[nbDims] filDims, int[nbDims] outDims, int[nbDims] inStride, int[nbDims] filStride, int[nbDims] outStride, int[imDims] stride, int[imDims] pad, int[imDims] dilation): Undocumented in source.