public class JCudnn extends Object
Modifier and Type | Field and Description |
---|---|
static double |
CUDNN_BN_MIN_EPSILON
Minimum epsilon allowed to be used in the Batch Normalization formula
|
static int |
CUDNN_DIM_MAX
Maximum supported number of tensor dimensions
|
static int |
CUDNN_MAJOR |
static int |
CUDNN_MINOR |
static int |
CUDNN_PATCHLEVEL |
static int |
CUDNN_VERSION |
Modifier and Type | Method and Description |
---|---|
static int |
cudnnActivationBackward_v4(cudnnHandle handle,
cudnnActivationDescriptor activationDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx) |
static int |
cudnnActivationBackward(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx)
Function to perform backward activation
|
static int |
cudnnActivationForward_v4(cudnnHandle handle,
cudnnActivationDescriptor activationDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
static int |
cudnnActivationForward(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Function to perform forward activation
|
static int |
cudnnAddTensor_v2(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor bDesc,
jcuda.Pointer b,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
static int |
cudnnAddTensor_v3(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor bDesc,
jcuda.Pointer b,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
cudnnAddTensor_v3 is now mapped to cudnnAddTensor
and will be removed at the same time as cudnnAddTensor_v2
Use cudnnAddTensor instead
|
static int |
cudnnAddTensor(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor bDesc,
jcuda.Pointer b,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Tensor Bias addition : y = alpha * b + beta * y
|
static int |
cudnnBatchNormalizationBackward(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
jcuda.Pointer beta,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx,
cudnnTensorDescriptor dBnScaleBiasDesc,
jcuda.Pointer bnScale,
jcuda.Pointer dBnScaleResult,
jcuda.Pointer dBnBiasResult,
double epsilon,
jcuda.Pointer savedMean,
jcuda.Pointer savedInvVariance)
Performs backward pass of Batch Normalization layer.
|
static int |
cudnnBatchNormalizationForwardInference(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
jcuda.Pointer beta,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor bnScaleBiasMeanVarDesc,
jcuda.Pointer bnScale,
jcuda.Pointer bnBias,
jcuda.Pointer estimatedMean,
jcuda.Pointer estimatedInvVariance,
double epsilon)
Performs Batch Normalization during Inference:
y[i] = bnScale[k]*(x[i]-estimatedMean[k])*estimatedInvVariance[k] + bnBias[k]
with bnScale, bnBias, runningMean, runningInvVariance tensors indexed
according to spatial or per-activation mode.
|
static int |
cudnnBatchNormalizationForwardTraining(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
jcuda.Pointer beta,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor bnScaleBiasMeanVarDesc,
jcuda.Pointer bnScale,
jcuda.Pointer bnBias,
double exponentialAverageFactor,
jcuda.Pointer resultRunningMean,
jcuda.Pointer resultRunningInvVariance,
double epsilon,
jcuda.Pointer resultSaveMean,
jcuda.Pointer resultSaveInvVariance)
Computes y = BN(x).
|
static int |
cudnnConvolutionBackwardBias(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
jcuda.Pointer beta,
cudnnTensorDescriptor dbDesc,
jcuda.Pointer db)
Function to compute the bias gradient for batch convolution
|
static int |
cudnnConvolutionBackwardData_v2(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnFilterDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnConvolutionDescriptor convDesc,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx) |
static int |
cudnnConvolutionBackwardData_v3(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnConvolutionDescriptor convDesc,
int algo,
jcuda.Pointer workSpace,
long workSpaceSizeInBytes,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx)
cudnnConvolutionBackwardData_v3 is now mapped to cudnnConvolutionBackwardData
and will be removed at the same time thancudnnConvolutionBackwardData_v2
Use cudnnConvolutionBackwardData instead
|
static int |
cudnnConvolutionBackwardData(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnConvolutionDescriptor convDesc,
int algo,
jcuda.Pointer workSpace,
long workSpaceSizeInBytes,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx) |
static int |
cudnnConvolutionBackwardFilter_v2(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnConvolutionDescriptor convDesc,
jcuda.Pointer beta,
cudnnFilterDescriptor dxDesc,
jcuda.Pointer dx) |
static int |
cudnnConvolutionBackwardFilter_v3(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnConvolutionDescriptor convDesc,
int algo,
jcuda.Pointer workSpace,
long workSpaceSizeInBytes,
jcuda.Pointer beta,
cudnnFilterDescriptor dwDesc,
jcuda.Pointer dw)
cudnnConvolutionBackwardFilter_v3 is now mapped to cudnnConvolutionBackwardFilter
and will be removed at the same time thancudnnConvolutionBackwardFilter_v2
Use cudnnConvolutionBackwardFilter instead
|
static int |
cudnnConvolutionBackwardFilter(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnConvolutionDescriptor convDesc,
int algo,
jcuda.Pointer workSpace,
long workSpaceSizeInBytes,
jcuda.Pointer beta,
cudnnFilterDescriptor dwDesc,
jcuda.Pointer dw) |
static int |
cudnnConvolutionForward(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnConvolutionDescriptor convDesc,
int algo,
jcuda.Pointer workSpace,
long workSpaceSizeInBytes,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Function to perform the forward pass for batch convolution
|
static int |
cudnnCreate(cudnnHandle handle) |
static int |
cudnnCreateActivationDescriptor(cudnnActivationDescriptor activationDesc)
Activation functions: All of the form "output = alpha * Op(inputs) + beta * output"
|
static int |
cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor convDesc)
Create an instance of convolution descriptor
|
static int |
cudnnCreateFilterDescriptor(cudnnFilterDescriptor filterDesc)
Create an instance of FilterStruct
|
static int |
cudnnCreateLRNDescriptor(cudnnLRNDescriptor normDesc)
Create an instance of LRN (Local Response Normalization) descriptor
Uses lrnN=5, lrnAlpha=1e-4, lrnBeta=0.75, lrnK=2.0 as defaults from Krizhevsky'12 ImageNet paper
|
static int |
cudnnCreatePoolingDescriptor(cudnnPoolingDescriptor poolingDesc)
Create an instance of pooling descriptor
|
static int |
cudnnCreateTensorDescriptor(cudnnTensorDescriptor tensorDesc)
Create an instance of a generic Tensor descriptor
|
static int |
cudnnDeriveBNTensorDescriptor(cudnnTensorDescriptor derivedBnDesc,
cudnnTensorDescriptor xDesc,
int mode)
Derives a tensor descriptor from layer data descriptor for BatchNormalization
scale, invVariance, bnBias, bnScale tensors.
|
static int |
cudnnDestroy(cudnnHandle handle) |
static int |
cudnnDestroyActivationDescriptor(cudnnActivationDescriptor activationDesc) |
static int |
cudnnDestroyConvolutionDescriptor(cudnnConvolutionDescriptor convDesc)
Destroy an instance of convolution descriptor
|
static int |
cudnnDestroyFilterDescriptor(cudnnFilterDescriptor filterDesc) |
static int |
cudnnDestroyLRNDescriptor(cudnnLRNDescriptor lrnDesc)
Destroy an instance of LRN descriptor
|
static int |
cudnnDestroyPoolingDescriptor(cudnnPoolingDescriptor poolingDesc)
Destroy an instance of pooling descriptor
|
static int |
cudnnDestroyTensorDescriptor(cudnnTensorDescriptor tensorDesc)
Destroy an instance of Tensor4d descriptor
|
static int |
cudnnDivisiveNormalizationBackward(cudnnHandle handle,
cudnnLRNDescriptor normDesc,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer means,
jcuda.Pointer dy,
jcuda.Pointer temp,
jcuda.Pointer temp2,
jcuda.Pointer beta,
cudnnTensorDescriptor dXdMeansDesc,
jcuda.Pointer dx,
jcuda.Pointer dMeans) |
static int |
cudnnDivisiveNormalizationForward(cudnnHandle handle,
cudnnLRNDescriptor normDesc,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer means,
jcuda.Pointer temp,
jcuda.Pointer temp2,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
LCN/divisive normalization functions: y = alpha * normalize(x) + beta * y
|
static int |
cudnnFindConvolutionBackwardDataAlgorithm(cudnnHandle handle,
cudnnFilterDescriptor wDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor dxDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf[] perfResults) |
static int |
cudnnFindConvolutionBackwardFilterAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor dwDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf[] perfResults) |
static int |
cudnnFindConvolutionForwardAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf[] perfResults) |
static int |
cudnnGetActivationDescriptor(cudnnActivationDescriptor activationDesc,
int[] mode,
int[] reluNanOpt,
jcuda.Pointer reluCeiling) |
static int |
cudnnGetConvolution2dDescriptor(cudnnConvolutionDescriptor convDesc,
jcuda.Pointer pad_h,
jcuda.Pointer pad_w,
jcuda.Pointer u,
jcuda.Pointer v,
jcuda.Pointer upscalex,
jcuda.Pointer upscaley,
int[] mode) |
static int |
cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor inputTensorDesc,
cudnnFilterDescriptor filterDesc,
jcuda.Pointer n,
jcuda.Pointer c,
jcuda.Pointer h,
jcuda.Pointer w)
Helper function to return the dimensions of the output tensor given a convolution descriptor
|
static int |
cudnnGetConvolutionBackwardDataAlgorithm(cudnnHandle handle,
cudnnFilterDescriptor wDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor dxDesc,
int preference,
long memoryLimitInBytes,
int[] algo) |
static int |
cudnnGetConvolutionBackwardDataWorkspaceSize(cudnnHandle handle,
cudnnFilterDescriptor wDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor dxDesc,
int algo,
long[] sizeInBytes)
Helper function to return the minimum size of the workspace to be passed to the convolution given an algo
|
static int |
cudnnGetConvolutionBackwardFilterAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor dwDesc,
int preference,
long memoryLimitInBytes,
int[] algo) |
static int |
cudnnGetConvolutionBackwardFilterWorkspaceSize(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor gradDesc,
int algo,
long[] sizeInBytes)
Helper function to return the minimum size of the workspace to be passed to the convolution given an algo
|
static int |
cudnnGetConvolutionForwardAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
int preference,
long memoryLimitInBytes,
int[] algo) |
static int |
cudnnGetConvolutionForwardWorkspaceSize(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
int algo,
long[] sizeInBytes)
Helper function to return the minimum size of the workspace to be passed to the convolution given an algo
|
static int |
cudnnGetConvolutionNdDescriptor_v2(cudnnConvolutionDescriptor convDesc,
int arrayLengthRequested,
int[] arrayLength,
int[] padA,
int[] strideA,
int[] upscaleA,
int[] mode) |
static int |
cudnnGetConvolutionNdDescriptor_v3(cudnnConvolutionDescriptor convDesc,
int arrayLengthRequested,
int[] arrayLength,
int[] padA,
int[] strideA,
int[] upscaleA,
int[] mode,
int[] dataType)
cudnnGetConvolutionNdDescriptor_v3 is now mapped to cudnnGetConvolutionNdDescriptor
and will be removed at the same time thancudnnGetConvolutionNdDescriptor_v2
Use cudnnGetConvolutionNdDescriptor instead
|
static int |
cudnnGetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc,
int arrayLengthRequested,
int[] arrayLength,
int[] padA,
int[] strideA,
int[] upscaleA,
int[] mode,
int[] dataType) |
static int |
cudnnGetConvolutionNdForwardOutputDim(cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor inputTensorDesc,
cudnnFilterDescriptor filterDesc,
int nbDims,
int[] tensorOuputDimA)
Helper function to return the dimensions of the output tensor given a convolution descriptor
|
static String |
cudnnGetErrorString(int status) |
static int |
cudnnGetFilter4dDescriptor_v4(cudnnFilterDescriptor filterDesc,
int[] dataType,
int[] format,
jcuda.Pointer k,
jcuda.Pointer c,
jcuda.Pointer h,
jcuda.Pointer w) |
static int |
cudnnGetFilter4dDescriptor(cudnnFilterDescriptor filterDesc,
int[] dataType,
jcuda.Pointer k,
jcuda.Pointer c,
jcuda.Pointer h,
jcuda.Pointer w) |
static int |
cudnnGetFilterNdDescriptor_v4(cudnnFilterDescriptor filterDesc,
int nbDimsRequested,
int[] dataType,
int[] format,
jcuda.Pointer nbDims,
int[] filterDimA) |
static int |
cudnnGetFilterNdDescriptor(cudnnFilterDescriptor filterDesc,
int nbDimsRequested,
int[] dataType,
jcuda.Pointer nbDims,
int[] filterDimA) |
static int |
cudnnGetLRNDescriptor(cudnnLRNDescriptor normDesc,
int[] lrnN,
jcuda.Pointer lrnAlpha,
jcuda.Pointer lrnBeta,
jcuda.Pointer lrnK)
Retrieve the settings currently stored in an LRN layer descriptor
Any of the provided pointers can be NULL (no corresponding value will be returned)
|
static int |
cudnnGetPooling2dDescriptor_v4(cudnnPoolingDescriptor poolingDesc,
int[] mode,
int[] maxpoolingNanOpt,
jcuda.Pointer windowHeight,
jcuda.Pointer windowWidth,
jcuda.Pointer verticalPadding,
jcuda.Pointer horizontalPadding,
jcuda.Pointer verticalStride,
jcuda.Pointer horizontalStride) |
static int |
cudnnGetPooling2dDescriptor(cudnnPoolingDescriptor poolingDesc,
int[] mode,
jcuda.Pointer windowHeight,
jcuda.Pointer windowWidth,
jcuda.Pointer verticalPadding,
jcuda.Pointer horizontalPadding,
jcuda.Pointer verticalStride,
jcuda.Pointer horizontalStride) |
static int |
cudnnGetPooling2dForwardOutputDim(cudnnPoolingDescriptor poolingDesc,
cudnnTensorDescriptor inputTensorDesc,
jcuda.Pointer n,
jcuda.Pointer c,
jcuda.Pointer h,
jcuda.Pointer w) |
static int |
cudnnGetPoolingNdDescriptor_v4(cudnnPoolingDescriptor poolingDesc,
int nbDimsRequested,
int[] mode,
int[] maxpoolingNanOpt,
jcuda.Pointer nbDims,
int[] windowDimA,
int[] paddingA,
int[] strideA) |
static int |
cudnnGetPoolingNdDescriptor(cudnnPoolingDescriptor poolingDesc,
int nbDimsRequested,
int[] mode,
jcuda.Pointer nbDims,
int[] windowDimA,
int[] paddingA,
int[] strideA) |
static int |
cudnnGetPoolingNdForwardOutputDim(cudnnPoolingDescriptor poolingDesc,
cudnnTensorDescriptor inputTensorDesc,
int nbDims,
int[] outputTensorDimA) |
static int |
cudnnGetStream(cudnnHandle handle,
jcuda.runtime.cudaStream_t streamId) |
static int |
cudnnGetTensor4dDescriptor(cudnnTensorDescriptor tensorDesc,
int[] dataType,
jcuda.Pointer n,
jcuda.Pointer c,
jcuda.Pointer h,
jcuda.Pointer w,
jcuda.Pointer nStride,
jcuda.Pointer cStride,
jcuda.Pointer hStride,
jcuda.Pointer wStride) |
static int |
cudnnGetTensorNdDescriptor(cudnnTensorDescriptor tensorDesc,
int nbDimsRequested,
int[] dataType,
int[] nbDims,
int[] dimA,
int[] strideA) |
static long |
cudnnGetVersion() |
static int |
cudnnIm2Col(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
jcuda.Pointer colBuffer) |
static int |
cudnnLRNCrossChannelBackward(cudnnHandle handle,
cudnnLRNDescriptor normDesc,
int lrnMode,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx)
LRN cross-channel backward computation.
|
static int |
cudnnLRNCrossChannelForward(cudnnHandle handle,
cudnnLRNDescriptor normDesc,
int lrnMode,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
LRN cross-channel forward computation.
|
static int |
cudnnPoolingBackward(cudnnHandle handle,
cudnnPoolingDescriptor poolingDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx)
Function to perform backward pooling
|
static int |
cudnnPoolingForward(cudnnHandle handle,
cudnnPoolingDescriptor poolingDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Function to perform forward pooling
|
static int |
cudnnScaleTensor(cudnnHandle handle,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
jcuda.Pointer alpha)
Scale all values of a tensor by a given factor : y[i] = alpha * y[i]
|
static int |
cudnnSetActivationDescriptor(cudnnActivationDescriptor activationDesc,
int mode,
int reluNanOpt,
double reluCeiling) |
static int |
cudnnSetConvolution2dDescriptor(cudnnConvolutionDescriptor convDesc,
int pad_h,
int pad_w,
int u,
int v,
int upscalex,
int upscaley,
int mode) |
static int |
cudnnSetConvolutionNdDescriptor_v2(cudnnConvolutionDescriptor convDesc,
int arrayLength,
int[] padA,
int[] filterStrideA,
int[] upscaleA,
int mode)
DEPRECATED API THAT WILL BE REMOVED FROM UPCOMING RELEASES
|
static int |
cudnnSetConvolutionNdDescriptor_v3(cudnnConvolutionDescriptor convDesc,
int arrayLength,
int[] padA,
int[] filterStrideA,
int[] upscaleA,
int mode,
int dataType)
cudnnSetConvolutionNdDescriptor_v3 is now mapped to cudnnSetConvolutionNdDescriptor
and will be removed at the same time than cudnnSetConvolutionNdDescriptor_v2
Use cudnnSetConvolutionNdDescriptor instead
|
static int |
cudnnSetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc,
int arrayLength,
int[] padA,
int[] filterStrideA,
int[] upscaleA,
int mode,
int dataType) |
static int |
cudnnSetFilter4dDescriptor_v4(cudnnFilterDescriptor filterDesc,
int dataType,
int format,
int k,
int c,
int h,
int w) |
static int |
cudnnSetFilter4dDescriptor(cudnnFilterDescriptor filterDesc,
int dataType,
int k,
int c,
int h,
int w) |
static int |
cudnnSetFilterNdDescriptor_v4(cudnnFilterDescriptor filterDesc,
int dataType,
int format,
int nbDims,
int[] filterDimA) |
static int |
cudnnSetFilterNdDescriptor(cudnnFilterDescriptor filterDesc,
int dataType,
int nbDims,
int[] filterDimA) |
static int |
cudnnSetLRNDescriptor(cudnnLRNDescriptor normDesc,
int lrnN,
double lrnAlpha,
double lrnBeta,
double lrnK)
Uses a window [center-lookBehind, center+lookAhead], where
lookBehind = floor( (lrnN-1)/2 ), lookAhead = lrnN-lookBehind-1.
|
static int |
cudnnSetPooling2dDescriptor_v4(cudnnPoolingDescriptor poolingDesc,
int mode,
int maxpoolingNanOpt,
int windowHeight,
int windowWidth,
int verticalPadding,
int horizontalPadding,
int verticalStride,
int horizontalStride) |
static int |
cudnnSetPooling2dDescriptor(cudnnPoolingDescriptor poolingDesc,
int mode,
int windowHeight,
int windowWidth,
int verticalPadding,
int horizontalPadding,
int verticalStride,
int horizontalStride) |
static int |
cudnnSetPoolingNdDescriptor_v4(cudnnPoolingDescriptor poolingDesc,
int mode,
int maxpoolingNanOpt,
int nbDims,
int[] windowDimA,
int[] paddingA,
int[] strideA) |
static int |
cudnnSetPoolingNdDescriptor(cudnnPoolingDescriptor poolingDesc,
int mode,
int nbDims,
int[] windowDimA,
int[] paddingA,
int[] strideA) |
static int |
cudnnSetStream(cudnnHandle handle,
jcuda.runtime.cudaStream_t streamId) |
static int |
cudnnSetTensor(cudnnHandle handle,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
jcuda.Pointer valuePtr)
Set all values of a tensor to a given value : y[i] = value[0]
|
static int |
cudnnSetTensor4dDescriptor(cudnnTensorDescriptor tensorDesc,
int format,
int dataType,
int n,
int c,
int h,
int w) |
static int |
cudnnSetTensor4dDescriptorEx(cudnnTensorDescriptor tensorDesc,
int dataType,
int n,
int c,
int h,
int w,
int nStride,
int cStride,
int hStride,
int wStride) |
static int |
cudnnSetTensorNdDescriptor(cudnnTensorDescriptor tensorDesc,
int dataType,
int nbDims,
int[] dimA,
int[] strideA) |
static int |
cudnnSoftmaxBackward(cudnnHandle handle,
int algo,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx)
Function to perform backward softmax
|
static int |
cudnnSoftmaxForward(cudnnHandle handle,
int algo,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Function to perform forward softmax
|
static int |
cudnnTransformTensor(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Tensor layout conversion helper (y = alpha * x + beta * y)
|
static void |
initialize()
Initializes the native library.
|
static void |
setExceptionsEnabled(boolean enabled)
Enables or disables exceptions.
|
static void |
setLogLevel(jcuda.LogLevel logLevel)
Set the specified log level for the JCudnn library.
|
public static final int CUDNN_MAJOR
public static final int CUDNN_MINOR
public static final int CUDNN_PATCHLEVEL
public static final int CUDNN_VERSION
public static final int CUDNN_DIM_MAX
public static final double CUDNN_BN_MIN_EPSILON
public static void initialize()
public static void setExceptionsEnabled(boolean enabled)
cudnnStatus
from the native methods.
If exceptions are enabled, a CudaException with a detailed error
message will be thrown if a method is about to set a result code
that is not cudnnStatus.CUDNN_STATUS_SUCCESSenabled
- Whether exceptions are enabledpublic static void setLogLevel(jcuda.LogLevel logLevel)
logLevel
- The log level to use.public static long cudnnGetVersion()
public static String cudnnGetErrorString(int status)
public static int cudnnCreate(cudnnHandle handle)
public static int cudnnDestroy(cudnnHandle handle)
public static int cudnnSetStream(cudnnHandle handle, jcuda.runtime.cudaStream_t streamId)
public static int cudnnGetStream(cudnnHandle handle, jcuda.runtime.cudaStream_t streamId)
public static int cudnnCreateTensorDescriptor(cudnnTensorDescriptor tensorDesc)
public static int cudnnSetTensor4dDescriptor(cudnnTensorDescriptor tensorDesc, int format, int dataType, int n, int c, int h, int w)
public static int cudnnSetTensor4dDescriptorEx(cudnnTensorDescriptor tensorDesc, int dataType, int n, int c, int h, int w, int nStride, int cStride, int hStride, int wStride)
public static int cudnnGetTensor4dDescriptor(cudnnTensorDescriptor tensorDesc, int[] dataType, jcuda.Pointer n, jcuda.Pointer c, jcuda.Pointer h, jcuda.Pointer w, jcuda.Pointer nStride, jcuda.Pointer cStride, jcuda.Pointer hStride, jcuda.Pointer wStride)
public static int cudnnSetTensorNdDescriptor(cudnnTensorDescriptor tensorDesc, int dataType, int nbDims, int[] dimA, int[] strideA)
public static int cudnnGetTensorNdDescriptor(cudnnTensorDescriptor tensorDesc, int nbDimsRequested, int[] dataType, int[] nbDims, int[] dimA, int[] strideA)
public static int cudnnDestroyTensorDescriptor(cudnnTensorDescriptor tensorDesc)
public static int cudnnTransformTensor(cudnnHandle handle, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnAddTensor(cudnnHandle handle, jcuda.Pointer alpha, cudnnTensorDescriptor bDesc, jcuda.Pointer b, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnAddTensor_v3(cudnnHandle handle, jcuda.Pointer alpha, cudnnTensorDescriptor bDesc, jcuda.Pointer b, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
cudnnAddTensor_v3 is now mapped to cudnnAddTensor and will be removed at the same time as cudnnAddTensor_v2 Use cudnnAddTensor instead
public static int cudnnSetTensor(cudnnHandle handle, cudnnTensorDescriptor yDesc, jcuda.Pointer y, jcuda.Pointer valuePtr)
public static int cudnnScaleTensor(cudnnHandle handle, cudnnTensorDescriptor yDesc, jcuda.Pointer y, jcuda.Pointer alpha)
public static int cudnnCreateFilterDescriptor(cudnnFilterDescriptor filterDesc)
public static int cudnnSetFilter4dDescriptor(cudnnFilterDescriptor filterDesc, int dataType, int k, int c, int h, int w)
public static int cudnnSetFilter4dDescriptor_v4(cudnnFilterDescriptor filterDesc, int dataType, int format, int k, int c, int h, int w)
public static int cudnnGetFilter4dDescriptor(cudnnFilterDescriptor filterDesc, int[] dataType, jcuda.Pointer k, jcuda.Pointer c, jcuda.Pointer h, jcuda.Pointer w)
public static int cudnnGetFilter4dDescriptor_v4(cudnnFilterDescriptor filterDesc, int[] dataType, int[] format, jcuda.Pointer k, jcuda.Pointer c, jcuda.Pointer h, jcuda.Pointer w)
public static int cudnnSetFilterNdDescriptor(cudnnFilterDescriptor filterDesc, int dataType, int nbDims, int[] filterDimA)
public static int cudnnSetFilterNdDescriptor_v4(cudnnFilterDescriptor filterDesc, int dataType, int format, int nbDims, int[] filterDimA)
public static int cudnnGetFilterNdDescriptor(cudnnFilterDescriptor filterDesc, int nbDimsRequested, int[] dataType, jcuda.Pointer nbDims, int[] filterDimA)
public static int cudnnGetFilterNdDescriptor_v4(cudnnFilterDescriptor filterDesc, int nbDimsRequested, int[] dataType, int[] format, jcuda.Pointer nbDims, int[] filterDimA)
public static int cudnnDestroyFilterDescriptor(cudnnFilterDescriptor filterDesc)
public static int cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor convDesc)
public static int cudnnSetConvolution2dDescriptor(cudnnConvolutionDescriptor convDesc, int pad_h, int pad_w, int u, int v, int upscalex, int upscaley, int mode)
public static int cudnnGetConvolution2dDescriptor(cudnnConvolutionDescriptor convDesc, jcuda.Pointer pad_h, jcuda.Pointer pad_w, jcuda.Pointer u, jcuda.Pointer v, jcuda.Pointer upscalex, jcuda.Pointer upscaley, int[] mode)
public static int cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor inputTensorDesc, cudnnFilterDescriptor filterDesc, jcuda.Pointer n, jcuda.Pointer c, jcuda.Pointer h, jcuda.Pointer w)
public static int cudnnSetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc, int arrayLength, int[] padA, int[] filterStrideA, int[] upscaleA, int mode, int dataType)
public static int cudnnGetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc, int arrayLengthRequested, int[] arrayLength, int[] padA, int[] strideA, int[] upscaleA, int[] mode, int[] dataType)
public static int cudnnSetConvolutionNdDescriptor_v3(cudnnConvolutionDescriptor convDesc, int arrayLength, int[] padA, int[] filterStrideA, int[] upscaleA, int mode, int dataType)
public static int cudnnGetConvolutionNdDescriptor_v3(cudnnConvolutionDescriptor convDesc, int arrayLengthRequested, int[] arrayLength, int[] padA, int[] strideA, int[] upscaleA, int[] mode, int[] dataType)
cudnnGetConvolutionNdDescriptor_v3 is now mapped to cudnnGetConvolutionNdDescriptor and will be removed at the same time thancudnnGetConvolutionNdDescriptor_v2 Use cudnnGetConvolutionNdDescriptor instead
public static int cudnnGetConvolutionNdForwardOutputDim(cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor inputTensorDesc, cudnnFilterDescriptor filterDesc, int nbDims, int[] tensorOuputDimA)
public static int cudnnDestroyConvolutionDescriptor(cudnnConvolutionDescriptor convDesc)
public static int cudnnFindConvolutionForwardAlgorithm(cudnnHandle handle, cudnnTensorDescriptor xDesc, cudnnFilterDescriptor wDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor yDesc, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionFwdAlgoPerf[] perfResults)
public static int cudnnGetConvolutionForwardAlgorithm(cudnnHandle handle, cudnnTensorDescriptor xDesc, cudnnFilterDescriptor wDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor yDesc, int preference, long memoryLimitInBytes, int[] algo)
public static int cudnnGetConvolutionForwardWorkspaceSize(cudnnHandle handle, cudnnTensorDescriptor xDesc, cudnnFilterDescriptor wDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor yDesc, int algo, long[] sizeInBytes)
public static int cudnnConvolutionForward(cudnnHandle handle, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnFilterDescriptor wDesc, jcuda.Pointer w, cudnnConvolutionDescriptor convDesc, int algo, jcuda.Pointer workSpace, long workSpaceSizeInBytes, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnConvolutionBackwardBias(cudnnHandle handle, jcuda.Pointer alpha, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, jcuda.Pointer beta, cudnnTensorDescriptor dbDesc, jcuda.Pointer db)
public static int cudnnFindConvolutionBackwardFilterAlgorithm(cudnnHandle handle, cudnnTensorDescriptor xDesc, cudnnTensorDescriptor dyDesc, cudnnConvolutionDescriptor convDesc, cudnnFilterDescriptor dwDesc, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf[] perfResults)
public static int cudnnGetConvolutionBackwardFilterAlgorithm(cudnnHandle handle, cudnnTensorDescriptor xDesc, cudnnTensorDescriptor dyDesc, cudnnConvolutionDescriptor convDesc, cudnnFilterDescriptor dwDesc, int preference, long memoryLimitInBytes, int[] algo)
public static int cudnnGetConvolutionBackwardFilterWorkspaceSize(cudnnHandle handle, cudnnTensorDescriptor xDesc, cudnnTensorDescriptor dyDesc, cudnnConvolutionDescriptor convDesc, cudnnFilterDescriptor gradDesc, int algo, long[] sizeInBytes)
public static int cudnnConvolutionBackwardFilter(cudnnHandle handle, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnConvolutionDescriptor convDesc, int algo, jcuda.Pointer workSpace, long workSpaceSizeInBytes, jcuda.Pointer beta, cudnnFilterDescriptor dwDesc, jcuda.Pointer dw)
public static int cudnnConvolutionBackwardFilter_v3(cudnnHandle handle, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnConvolutionDescriptor convDesc, int algo, jcuda.Pointer workSpace, long workSpaceSizeInBytes, jcuda.Pointer beta, cudnnFilterDescriptor dwDesc, jcuda.Pointer dw)
public static int cudnnFindConvolutionBackwardDataAlgorithm(cudnnHandle handle, cudnnFilterDescriptor wDesc, cudnnTensorDescriptor dyDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor dxDesc, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf[] perfResults)
public static int cudnnGetConvolutionBackwardDataAlgorithm(cudnnHandle handle, cudnnFilterDescriptor wDesc, cudnnTensorDescriptor dyDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor dxDesc, int preference, long memoryLimitInBytes, int[] algo)
public static int cudnnGetConvolutionBackwardDataWorkspaceSize(cudnnHandle handle, cudnnFilterDescriptor wDesc, cudnnTensorDescriptor dyDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor dxDesc, int algo, long[] sizeInBytes)
public static int cudnnConvolutionBackwardData(cudnnHandle handle, jcuda.Pointer alpha, cudnnFilterDescriptor wDesc, jcuda.Pointer w, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnConvolutionDescriptor convDesc, int algo, jcuda.Pointer workSpace, long workSpaceSizeInBytes, jcuda.Pointer beta, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx)
public static int cudnnConvolutionBackwardData_v3(cudnnHandle handle, jcuda.Pointer alpha, cudnnFilterDescriptor wDesc, jcuda.Pointer w, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnConvolutionDescriptor convDesc, int algo, jcuda.Pointer workSpace, long workSpaceSizeInBytes, jcuda.Pointer beta, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx)
public static int cudnnIm2Col(cudnnHandle handle, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnFilterDescriptor wDesc, cudnnConvolutionDescriptor convDesc, jcuda.Pointer colBuffer)
public static int cudnnSoftmaxForward(cudnnHandle handle, int algo, int mode, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnSoftmaxBackward(cudnnHandle handle, int algo, int mode, jcuda.Pointer alpha, cudnnTensorDescriptor yDesc, jcuda.Pointer y, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, jcuda.Pointer beta, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx)
public static int cudnnCreatePoolingDescriptor(cudnnPoolingDescriptor poolingDesc)
public static int cudnnSetPooling2dDescriptor(cudnnPoolingDescriptor poolingDesc, int mode, int windowHeight, int windowWidth, int verticalPadding, int horizontalPadding, int verticalStride, int horizontalStride)
public static int cudnnSetPooling2dDescriptor_v4(cudnnPoolingDescriptor poolingDesc, int mode, int maxpoolingNanOpt, int windowHeight, int windowWidth, int verticalPadding, int horizontalPadding, int verticalStride, int horizontalStride)
public static int cudnnGetPooling2dDescriptor(cudnnPoolingDescriptor poolingDesc, int[] mode, jcuda.Pointer windowHeight, jcuda.Pointer windowWidth, jcuda.Pointer verticalPadding, jcuda.Pointer horizontalPadding, jcuda.Pointer verticalStride, jcuda.Pointer horizontalStride)
public static int cudnnGetPooling2dDescriptor_v4(cudnnPoolingDescriptor poolingDesc, int[] mode, int[] maxpoolingNanOpt, jcuda.Pointer windowHeight, jcuda.Pointer windowWidth, jcuda.Pointer verticalPadding, jcuda.Pointer horizontalPadding, jcuda.Pointer verticalStride, jcuda.Pointer horizontalStride)
public static int cudnnSetPoolingNdDescriptor(cudnnPoolingDescriptor poolingDesc, int mode, int nbDims, int[] windowDimA, int[] paddingA, int[] strideA)
public static int cudnnSetPoolingNdDescriptor_v4(cudnnPoolingDescriptor poolingDesc, int mode, int maxpoolingNanOpt, int nbDims, int[] windowDimA, int[] paddingA, int[] strideA)
public static int cudnnGetPoolingNdDescriptor(cudnnPoolingDescriptor poolingDesc, int nbDimsRequested, int[] mode, jcuda.Pointer nbDims, int[] windowDimA, int[] paddingA, int[] strideA)
public static int cudnnGetPoolingNdDescriptor_v4(cudnnPoolingDescriptor poolingDesc, int nbDimsRequested, int[] mode, int[] maxpoolingNanOpt, jcuda.Pointer nbDims, int[] windowDimA, int[] paddingA, int[] strideA)
public static int cudnnGetPoolingNdForwardOutputDim(cudnnPoolingDescriptor poolingDesc, cudnnTensorDescriptor inputTensorDesc, int nbDims, int[] outputTensorDimA)
public static int cudnnGetPooling2dForwardOutputDim(cudnnPoolingDescriptor poolingDesc, cudnnTensorDescriptor inputTensorDesc, jcuda.Pointer n, jcuda.Pointer c, jcuda.Pointer h, jcuda.Pointer w)
public static int cudnnDestroyPoolingDescriptor(cudnnPoolingDescriptor poolingDesc)
public static int cudnnPoolingForward(cudnnHandle handle, cudnnPoolingDescriptor poolingDesc, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnPoolingBackward(cudnnHandle handle, cudnnPoolingDescriptor poolingDesc, jcuda.Pointer alpha, cudnnTensorDescriptor yDesc, jcuda.Pointer y, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx)
public static int cudnnCreateActivationDescriptor(cudnnActivationDescriptor activationDesc)
public static int cudnnSetActivationDescriptor(cudnnActivationDescriptor activationDesc, int mode, int reluNanOpt, double reluCeiling)
public static int cudnnGetActivationDescriptor(cudnnActivationDescriptor activationDesc, int[] mode, int[] reluNanOpt, jcuda.Pointer reluCeiling)
public static int cudnnDestroyActivationDescriptor(cudnnActivationDescriptor activationDesc)
public static int cudnnActivationForward(cudnnHandle handle, int mode, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnActivationForward_v4(cudnnHandle handle, cudnnActivationDescriptor activationDesc, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnActivationBackward(cudnnHandle handle, int mode, jcuda.Pointer alpha, cudnnTensorDescriptor yDesc, jcuda.Pointer y, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx)
public static int cudnnActivationBackward_v4(cudnnHandle handle, cudnnActivationDescriptor activationDesc, jcuda.Pointer alpha, cudnnTensorDescriptor yDesc, jcuda.Pointer y, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx)
public static int cudnnCreateLRNDescriptor(cudnnLRNDescriptor normDesc)
Create an instance of LRN (Local Response Normalization) descriptor Uses lrnN=5, lrnAlpha=1e-4, lrnBeta=0.75, lrnK=2.0 as defaults from Krizhevsky'12 ImageNet paper
public static int cudnnSetLRNDescriptor(cudnnLRNDescriptor normDesc, int lrnN, double lrnAlpha, double lrnBeta, double lrnK)
Uses a window [center-lookBehind, center+lookAhead], where lookBehind = floor( (lrnN-1)/2 ), lookAhead = lrnN-lookBehind-1. Values of double parameters cast to tensor data type.
public static int cudnnGetLRNDescriptor(cudnnLRNDescriptor normDesc, int[] lrnN, jcuda.Pointer lrnAlpha, jcuda.Pointer lrnBeta, jcuda.Pointer lrnK)
Retrieve the settings currently stored in an LRN layer descriptor Any of the provided pointers can be NULL (no corresponding value will be returned)
public static int cudnnDestroyLRNDescriptor(cudnnLRNDescriptor lrnDesc)
public static int cudnnLRNCrossChannelForward(cudnnHandle handle, cudnnLRNDescriptor normDesc, int lrnMode, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnLRNCrossChannelBackward(cudnnHandle handle, cudnnLRNDescriptor normDesc, int lrnMode, jcuda.Pointer alpha, cudnnTensorDescriptor yDesc, jcuda.Pointer y, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx)
public static int cudnnDivisiveNormalizationForward(cudnnHandle handle, cudnnLRNDescriptor normDesc, int mode, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer means, jcuda.Pointer temp, jcuda.Pointer temp2, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnDivisiveNormalizationBackward(cudnnHandle handle, cudnnLRNDescriptor normDesc, int mode, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer means, jcuda.Pointer dy, jcuda.Pointer temp, jcuda.Pointer temp2, jcuda.Pointer beta, cudnnTensorDescriptor dXdMeansDesc, jcuda.Pointer dx, jcuda.Pointer dMeans)
public static int cudnnDeriveBNTensorDescriptor(cudnnTensorDescriptor derivedBnDesc, cudnnTensorDescriptor xDesc, int mode)
Derives a tensor descriptor from layer data descriptor for BatchNormalization scale, invVariance, bnBias, bnScale tensors. Use this tensor desc for bnScaleBiasMeanVarDesc and bnScaleBiasDiffDesc in Batch Normalization forward and backward functions.
public static int cudnnBatchNormalizationForwardTraining(cudnnHandle handle, int mode, jcuda.Pointer alpha, jcuda.Pointer beta, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor yDesc, jcuda.Pointer y, cudnnTensorDescriptor bnScaleBiasMeanVarDesc, jcuda.Pointer bnScale, jcuda.Pointer bnBias, double exponentialAverageFactor, jcuda.Pointer resultRunningMean, jcuda.Pointer resultRunningInvVariance, double epsilon, jcuda.Pointer resultSaveMean, jcuda.Pointer resultSaveInvVariance)
public static int cudnnBatchNormalizationForwardInference(cudnnHandle handle, int mode, jcuda.Pointer alpha, jcuda.Pointer beta, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor yDesc, jcuda.Pointer y, cudnnTensorDescriptor bnScaleBiasMeanVarDesc, jcuda.Pointer bnScale, jcuda.Pointer bnBias, jcuda.Pointer estimatedMean, jcuda.Pointer estimatedInvVariance, double epsilon)
Performs Batch Normalization during Inference: y[i] = bnScale[k]*(x[i]-estimatedMean[k])*estimatedInvVariance[k] + bnBias[k] with bnScale, bnBias, runningMean, runningInvVariance tensors indexed according to spatial or per-activation mode. Refer to cudnnBatchNormalizationForwardTraining above for notes on function arguments.
public static int cudnnBatchNormalizationBackward(cudnnHandle handle, int mode, jcuda.Pointer alpha, jcuda.Pointer beta, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx, cudnnTensorDescriptor dBnScaleBiasDesc, jcuda.Pointer bnScale, jcuda.Pointer dBnScaleResult, jcuda.Pointer dBnBiasResult, double epsilon, jcuda.Pointer savedMean, jcuda.Pointer savedInvVariance)
public static int cudnnSetConvolutionNdDescriptor_v2(cudnnConvolutionDescriptor convDesc, int arrayLength, int[] padA, int[] filterStrideA, int[] upscaleA, int mode)
public static int cudnnGetConvolutionNdDescriptor_v2(cudnnConvolutionDescriptor convDesc, int arrayLengthRequested, int[] arrayLength, int[] padA, int[] strideA, int[] upscaleA, int[] mode)
public static int cudnnAddTensor_v2(cudnnHandle handle, int mode, jcuda.Pointer alpha, cudnnTensorDescriptor bDesc, jcuda.Pointer b, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnConvolutionBackwardFilter_v2(cudnnHandle handle, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnConvolutionDescriptor convDesc, jcuda.Pointer beta, cudnnFilterDescriptor dxDesc, jcuda.Pointer dx)
public static int cudnnConvolutionBackwardData_v2(cudnnHandle handle, jcuda.Pointer alpha, cudnnFilterDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnConvolutionDescriptor convDesc, jcuda.Pointer beta, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx)
Copyright © 2016. All rights reserved.