MCPcopy Create free account
hub / github.com/arrayfire/arrayfire / cudnnConvolutionBackwardData

Function cudnnConvolutionBackwardData

src/backend/cuda/cudnn.cpp:163–174  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

161}
162
163cudnnStatus_t cudnnConvolutionBackwardData(
164 cudnnHandle_t handle, const void *alpha,
165 const cudnnFilterDescriptor_t wDesc, const void *w,
166 const cudnnTensorDescriptor_t dyDesc, const void *dy,
167 const cudnnConvolutionDescriptor_t convDesc,
168 cudnnConvolutionBwdDataAlgo_t algo, void *workSpace,
169 size_t workSpaceSizeInBytes, const void *beta,
170 const cudnnTensorDescriptor_t dxDesc, void *dx) {
171 return getCudnnPlugin().cudnnConvolutionBackwardData(
172 handle, alpha, wDesc, w, dyDesc, dy, convDesc, algo, workSpace,
173 workSpaceSizeInBytes, beta, dxDesc, dx);
174}
175
176cudnnStatus_t cudnnGetConvolutionNdForwardOutputDim(
177 const cudnnConvolutionDescriptor_t convDesc,

Callers 1

data_gradient_cudnnFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected