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(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 |
cudnnActivationForward(cudnnHandle handle,
cudnnActivationDescriptor activationDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
static int |
cudnnAddTensor(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor aDesc,
jcuda.Pointer A,
jcuda.Pointer beta,
cudnnTensorDescriptor cDesc,
jcuda.Pointer C) |
static int |
cudnnBatchNormalizationBackward(cudnnHandle handle,
int mode,
jcuda.Pointer alphaDataDiff,
jcuda.Pointer betaDataDiff,
jcuda.Pointer alphaParamDiff,
jcuda.Pointer betaParamDiff,
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) |
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 estimatedVariance,
double epsilon) |
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 resultRunningVariance,
double epsilon,
jcuda.Pointer resultSaveMean,
jcuda.Pointer resultSaveInvVariance) |
static int |
cudnnConvolutionBackwardBias(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
jcuda.Pointer beta,
cudnnTensorDescriptor dbDesc,
jcuda.Pointer db) |
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(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) |
static int |
cudnnCreate(cudnnHandle handle) |
static int |
cudnnCreateActivationDescriptor(cudnnActivationDescriptor activationDesc) |
static int |
cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor convDesc) |
static int |
cudnnCreateDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc) |
static int |
cudnnCreateFilterDescriptor(cudnnFilterDescriptor filterDesc) |
static int |
cudnnCreateLRNDescriptor(cudnnLRNDescriptor normDesc) |
static int |
cudnnCreateOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc) |
static int |
cudnnCreatePoolingDescriptor(cudnnPoolingDescriptor poolingDesc) |
static int |
cudnnCreateRNNDescriptor(cudnnRNNDescriptor rnnDesc) |
static int |
cudnnCreateSpatialTransformerDescriptor(cudnnSpatialTransformerDescriptor stDesc) |
static int |
cudnnCreateTensorDescriptor(cudnnTensorDescriptor tensorDesc) |
static int |
cudnnDeriveBNTensorDescriptor(cudnnTensorDescriptor derivedBnDesc,
cudnnTensorDescriptor xDesc,
int mode) |
static int |
cudnnDestroy(cudnnHandle handle) |
static int |
cudnnDestroyActivationDescriptor(cudnnActivationDescriptor activationDesc) |
static int |
cudnnDestroyConvolutionDescriptor(cudnnConvolutionDescriptor convDesc) |
static int |
cudnnDestroyDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc) |
static int |
cudnnDestroyFilterDescriptor(cudnnFilterDescriptor filterDesc) |
static int |
cudnnDestroyLRNDescriptor(cudnnLRNDescriptor lrnDesc) |
static int |
cudnnDestroyOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc) |
static int |
cudnnDestroyPoolingDescriptor(cudnnPoolingDescriptor poolingDesc) |
static int |
cudnnDestroyRNNDescriptor(cudnnRNNDescriptor rnnDesc) |
static int |
cudnnDestroySpatialTransformerDescriptor(cudnnSpatialTransformerDescriptor stDesc) |
static int |
cudnnDestroyTensorDescriptor(cudnnTensorDescriptor tensorDesc) |
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) |
static int |
cudnnDropoutBackward(cudnnHandle handle,
cudnnDropoutDescriptor dropoutDesc,
cudnnTensorDescriptor dydesc,
jcuda.Pointer dy,
cudnnTensorDescriptor dxdesc,
jcuda.Pointer dx,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnDropoutForward(cudnnHandle handle,
cudnnDropoutDescriptor dropoutDesc,
cudnnTensorDescriptor xdesc,
jcuda.Pointer x,
cudnnTensorDescriptor ydesc,
jcuda.Pointer y,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnDropoutGetReserveSpaceSize(cudnnTensorDescriptor xdesc,
long[] sizeInBytes) |
static int |
cudnnDropoutGetStatesSize(cudnnHandle handle,
long[] sizeInBytes) |
static int |
cudnnFindConvolutionBackwardDataAlgorithm(cudnnHandle handle,
cudnnFilterDescriptor wDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor dxDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf[] perfResults) |
static int |
cudnnFindConvolutionBackwardDataAlgorithmEx(cudnnHandle handle,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf[] perfResults,
jcuda.Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindConvolutionBackwardFilterAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor dwDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf[] perfResults) |
static int |
cudnnFindConvolutionBackwardFilterAlgorithmEx(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer y,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor dwDesc,
jcuda.Pointer dw,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf[] perfResults,
jcuda.Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindConvolutionForwardAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf[] perfResults) |
static int |
cudnnFindConvolutionForwardAlgorithmEx(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf[] perfResults,
jcuda.Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnGetActivationDescriptor(cudnnActivationDescriptor activationDesc,
int[] mode,
int[] reluNanOpt,
double[] reluCeiling) |
static int |
cudnnGetConvolution2dDescriptor_v5(cudnnConvolutionDescriptor convDesc,
int[] pad_h,
int[] pad_w,
int[] u,
int[] v,
int[] upscalex,
int[] upscaley,
int[] mode,
int[] dataType) |
static int |
cudnnGetConvolution2dDescriptor(cudnnConvolutionDescriptor convDesc,
int[] pad_h,
int[] pad_w,
int[] u,
int[] v,
int[] upscalex,
int[] upscaley,
int[] mode) |
static int |
cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor inputTensorDesc,
cudnnFilterDescriptor filterDesc,
int[] n,
int[] c,
int[] h,
int[] w) |
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) |
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) |
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) |
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) |
static String |
cudnnGetErrorString(int status) |
static int |
cudnnGetFilter4dDescriptor(cudnnFilterDescriptor filterDesc,
int[] dataType,
int[] format,
int[] k,
int[] c,
int[] h,
int[] w) |
static int |
cudnnGetFilterNdDescriptor(cudnnFilterDescriptor filterDesc,
int nbDimsRequested,
int[] dataType,
int[] format,
int[] nbDims,
int[] filterDimA) |
static int |
cudnnGetLRNDescriptor(cudnnLRNDescriptor normDesc,
int[] lrnN,
double[] lrnAlpha,
double[] lrnBeta,
double[] lrnK) |
static int |
cudnnGetOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc,
int[] opTensorOp,
int[] opTensorCompType,
int[] opTensorNanOpt) |
static int |
cudnnGetPooling2dDescriptor(cudnnPoolingDescriptor poolingDesc,
int[] mode,
int[] maxpoolingNanOpt,
int[] windowHeight,
int[] windowWidth,
int[] verticalPadding,
int[] horizontalPadding,
int[] verticalStride,
int[] horizontalStride) |
static int |
cudnnGetPooling2dForwardOutputDim(cudnnPoolingDescriptor poolingDesc,
cudnnTensorDescriptor inputTensorDesc,
int[] n,
int[] c,
int[] h,
int[] w) |
static int |
cudnnGetPoolingNdDescriptor(cudnnPoolingDescriptor poolingDesc,
int nbDimsRequested,
int[] mode,
int[] maxpoolingNanOpt,
int[] nbDims,
int[] windowDimA,
int[] paddingA,
int[] strideA) |
static int |
cudnnGetPoolingNdForwardOutputDim(cudnnPoolingDescriptor poolingDesc,
cudnnTensorDescriptor inputTensorDesc,
int nbDims,
int[] outputTensorDimA) |
static int |
cudnnGetRNNLinLayerBiasParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int layer,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
int linLayerID,
cudnnFilterDescriptor linLayerBiasDesc,
jcuda.Pointer linLayerBias) |
static int |
cudnnGetRNNLinLayerMatrixParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int layer,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
int linLayerID,
cudnnFilterDescriptor linLayerMatDesc,
jcuda.Pointer linLayerMat) |
static int |
cudnnGetRNNParamsSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnTensorDescriptor xDesc,
long[] sizeInBytes,
int dataType) |
static int |
cudnnGetRNNTrainingReserveSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
long[] sizeInBytes) |
static int |
cudnnGetRNNWorkspaceSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
long[] sizeInBytes) |
static int |
cudnnGetStream(cudnnHandle handle,
jcuda.runtime.cudaStream_t streamId) |
static int |
cudnnGetTensor4dDescriptor(cudnnTensorDescriptor tensorDesc,
int[] dataType,
int[] n,
int[] c,
int[] h,
int[] w,
int[] nStride,
int[] cStride,
int[] hStride,
int[] 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) |
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) |
static int |
cudnnOpTensor(cudnnHandle handle,
cudnnOpTensorDescriptor opTensorDesc,
jcuda.Pointer alpha1,
cudnnTensorDescriptor aDesc,
jcuda.Pointer A,
jcuda.Pointer alpha2,
cudnnTensorDescriptor bDesc,
jcuda.Pointer B,
jcuda.Pointer beta,
cudnnTensorDescriptor cDesc,
jcuda.Pointer C) |
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) |
static int |
cudnnPoolingForward(cudnnHandle handle,
cudnnPoolingDescriptor poolingDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
static int |
cudnnRNNBackwardData(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor dhyDesc,
jcuda.Pointer dhy,
cudnnTensorDescriptor dcyDesc,
jcuda.Pointer dcy,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnTensorDescriptor hxDesc,
jcuda.Pointer hx,
cudnnTensorDescriptor cxDesc,
jcuda.Pointer cx,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx,
cudnnTensorDescriptor dhxDesc,
jcuda.Pointer dhx,
cudnnTensorDescriptor dcxDesc,
jcuda.Pointer dcx,
jcuda.Pointer workspace,
long workSpaceSizeInBytes,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNBackwardWeights(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor hxDesc,
jcuda.Pointer hx,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
jcuda.Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterDescriptor dwDesc,
jcuda.Pointer dw,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNForwardInference(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor hxDesc,
jcuda.Pointer hx,
cudnnTensorDescriptor cxDesc,
jcuda.Pointer cx,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor hyDesc,
jcuda.Pointer hy,
cudnnTensorDescriptor cyDesc,
jcuda.Pointer cy,
jcuda.Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnRNNForwardTraining(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor hxDesc,
jcuda.Pointer hx,
cudnnTensorDescriptor cxDesc,
jcuda.Pointer cx,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor hyDesc,
jcuda.Pointer hy,
cudnnTensorDescriptor cyDesc,
jcuda.Pointer cy,
jcuda.Pointer workspace,
long workSpaceSizeInBytes,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnScaleTensor(cudnnHandle handle,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
jcuda.Pointer alpha) |
static int |
cudnnSetActivationDescriptor(cudnnActivationDescriptor activationDesc,
int mode,
int reluNanOpt,
double reluCeiling) |
static int |
cudnnSetConvolution2dDescriptor_v5(cudnnConvolutionDescriptor convDesc,
int pad_h,
int pad_w,
int u,
int v,
int upscalex,
int upscaley,
int mode,
int dataType) |
static int |
cudnnSetConvolution2dDescriptor(cudnnConvolutionDescriptor convDesc,
int pad_h,
int pad_w,
int u,
int v,
int upscalex,
int upscaley,
int mode) |
static int |
cudnnSetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc,
int arrayLength,
int[] padA,
int[] filterStrideA,
int[] upscaleA,
int mode,
int dataType) |
static int |
cudnnSetDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc,
cudnnHandle handle,
float dropout,
jcuda.Pointer states,
long stateSizeInBytes,
long seed) |
static int |
cudnnSetFilter4dDescriptor(cudnnFilterDescriptor filterDesc,
int dataType,
int format,
int k,
int c,
int h,
int w) |
static int |
cudnnSetFilterNdDescriptor(cudnnFilterDescriptor filterDesc,
int dataType,
int format,
int nbDims,
int[] filterDimA) |
static int |
cudnnSetLRNDescriptor(cudnnLRNDescriptor normDesc,
int lrnN,
double lrnAlpha,
double lrnBeta,
double lrnK) |
static int |
cudnnSetOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc,
int opTensorOp,
int opTensorCompType,
int opTensorNanOpt) |
static int |
cudnnSetPooling2dDescriptor(cudnnPoolingDescriptor poolingDesc,
int mode,
int maxpoolingNanOpt,
int windowHeight,
int windowWidth,
int verticalPadding,
int horizontalPadding,
int verticalStride,
int horizontalStride) |
static int |
cudnnSetPoolingNdDescriptor(cudnnPoolingDescriptor poolingDesc,
int mode,
int maxpoolingNanOpt,
int nbDims,
int[] windowDimA,
int[] paddingA,
int[] strideA) |
static int |
cudnnSetRNNDescriptor(cudnnRNNDescriptor rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutDescriptor dropoutDesc,
int inputMode,
int direction,
int mode,
int dataType) |
static int |
cudnnSetSpatialTransformerNdDescriptor(cudnnSpatialTransformerDescriptor stDesc,
int samplerType,
int dataType,
int nbDims,
int[] dimA) |
static int |
cudnnSetStream(cudnnHandle handle,
jcuda.runtime.cudaStream_t streamId) |
static int |
cudnnSetTensor(cudnnHandle handle,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
jcuda.Pointer valuePtr) |
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) |
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) |
static int |
cudnnSpatialTfGridGeneratorBackward(cudnnHandle handle,
cudnnSpatialTransformerDescriptor stDesc,
jcuda.Pointer dgrid,
jcuda.Pointer dtheta) |
static int |
cudnnSpatialTfGridGeneratorForward(cudnnHandle handle,
cudnnSpatialTransformerDescriptor stDesc,
jcuda.Pointer theta,
jcuda.Pointer grid) |
static int |
cudnnSpatialTfSamplerBackward(cudnnHandle handle,
cudnnSpatialTransformerDescriptor stDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx,
jcuda.Pointer alphaDgrid,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
jcuda.Pointer grid,
jcuda.Pointer betaDgrid,
jcuda.Pointer dgrid) |
static int |
cudnnSpatialTfSamplerForward(cudnnHandle handle,
cudnnSpatialTransformerDescriptor stDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer grid,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
static int |
cudnnTransformTensor(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer 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, int[] n, int[] c, int[] h, int[] w, int[] nStride, int[] cStride, int[] hStride, int[] 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 aDesc, jcuda.Pointer A, jcuda.Pointer beta, cudnnTensorDescriptor cDesc, jcuda.Pointer C)
public static int cudnnCreateOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc)
public static int cudnnSetOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc, int opTensorOp, int opTensorCompType, int opTensorNanOpt)
public static int cudnnGetOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc, int[] opTensorOp, int[] opTensorCompType, int[] opTensorNanOpt)
public static int cudnnDestroyOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc)
public static int cudnnOpTensor(cudnnHandle handle, cudnnOpTensorDescriptor opTensorDesc, jcuda.Pointer alpha1, cudnnTensorDescriptor aDesc, jcuda.Pointer A, jcuda.Pointer alpha2, cudnnTensorDescriptor bDesc, jcuda.Pointer B, jcuda.Pointer beta, cudnnTensorDescriptor cDesc, jcuda.Pointer C)
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 format, int k, int c, int h, int w)
public static int cudnnGetFilter4dDescriptor(cudnnFilterDescriptor filterDesc, int[] dataType, int[] format, int[] k, int[] c, int[] h, int[] w)
public static int cudnnSetFilterNdDescriptor(cudnnFilterDescriptor filterDesc, int dataType, int format, int nbDims, int[] filterDimA)
public static int cudnnGetFilterNdDescriptor(cudnnFilterDescriptor filterDesc, int nbDimsRequested, int[] dataType, int[] format, int[] 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 cudnnSetConvolution2dDescriptor_v5(cudnnConvolutionDescriptor convDesc, int pad_h, int pad_w, int u, int v, int upscalex, int upscaley, int mode, int dataType)
public static int cudnnGetConvolution2dDescriptor(cudnnConvolutionDescriptor convDesc, int[] pad_h, int[] pad_w, int[] u, int[] v, int[] upscalex, int[] upscaley, int[] mode)
public static int cudnnGetConvolution2dDescriptor_v5(cudnnConvolutionDescriptor convDesc, int[] pad_h, int[] pad_w, int[] u, int[] v, int[] upscalex, int[] upscaley, int[] mode, int[] dataType)
public static int cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor inputTensorDesc, cudnnFilterDescriptor filterDesc, int[] n, int[] c, int[] h, int[] 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 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 cudnnFindConvolutionForwardAlgorithmEx(cudnnHandle handle, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnFilterDescriptor wDesc, jcuda.Pointer w, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor yDesc, jcuda.Pointer y, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionFwdAlgoPerf[] perfResults, jcuda.Pointer workSpace, long workSpaceSizeInBytes)
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 cudnnFindConvolutionBackwardFilterAlgorithmEx(cudnnHandle handle, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor dyDesc, jcuda.Pointer y, cudnnConvolutionDescriptor convDesc, cudnnFilterDescriptor dwDesc, jcuda.Pointer dw, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf[] perfResults, jcuda.Pointer workSpace, long workSpaceSizeInBytes)
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 cudnnFindConvolutionBackwardDataAlgorithm(cudnnHandle handle, cudnnFilterDescriptor wDesc, cudnnTensorDescriptor dyDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor dxDesc, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf[] perfResults)
public static int cudnnFindConvolutionBackwardDataAlgorithmEx(cudnnHandle handle, cudnnFilterDescriptor wDesc, jcuda.Pointer w, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf[] perfResults, jcuda.Pointer workSpace, long workSpaceSizeInBytes)
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 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 maxpoolingNanOpt, int windowHeight, int windowWidth, int verticalPadding, int horizontalPadding, int verticalStride, int horizontalStride)
public static int cudnnGetPooling2dDescriptor(cudnnPoolingDescriptor poolingDesc, int[] mode, int[] maxpoolingNanOpt, int[] windowHeight, int[] windowWidth, int[] verticalPadding, int[] horizontalPadding, int[] verticalStride, int[] horizontalStride)
public static int cudnnSetPoolingNdDescriptor(cudnnPoolingDescriptor poolingDesc, int mode, int maxpoolingNanOpt, int nbDims, int[] windowDimA, int[] paddingA, int[] strideA)
public static int cudnnGetPoolingNdDescriptor(cudnnPoolingDescriptor poolingDesc, int nbDimsRequested, int[] mode, int[] maxpoolingNanOpt, int[] 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, int[] n, int[] c, int[] h, int[] 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, double[] reluCeiling)
public static int cudnnDestroyActivationDescriptor(cudnnActivationDescriptor activationDesc)
public static int cudnnActivationForward(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, 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)
public static int cudnnSetLRNDescriptor(cudnnLRNDescriptor normDesc, int lrnN, double lrnAlpha, double lrnBeta, double lrnK)
public static int cudnnGetLRNDescriptor(cudnnLRNDescriptor normDesc, int[] lrnN, double[] lrnAlpha, double[] lrnBeta, double[] lrnK)
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)
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 resultRunningVariance, 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 estimatedVariance, double epsilon)
public static int cudnnBatchNormalizationBackward(cudnnHandle handle, int mode, jcuda.Pointer alphaDataDiff, jcuda.Pointer betaDataDiff, jcuda.Pointer alphaParamDiff, jcuda.Pointer betaParamDiff, 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 cudnnCreateSpatialTransformerDescriptor(cudnnSpatialTransformerDescriptor stDesc)
public static int cudnnSetSpatialTransformerNdDescriptor(cudnnSpatialTransformerDescriptor stDesc, int samplerType, int dataType, int nbDims, int[] dimA)
public static int cudnnDestroySpatialTransformerDescriptor(cudnnSpatialTransformerDescriptor stDesc)
public static int cudnnSpatialTfGridGeneratorForward(cudnnHandle handle, cudnnSpatialTransformerDescriptor stDesc, jcuda.Pointer theta, jcuda.Pointer grid)
public static int cudnnSpatialTfGridGeneratorBackward(cudnnHandle handle, cudnnSpatialTransformerDescriptor stDesc, jcuda.Pointer dgrid, jcuda.Pointer dtheta)
public static int cudnnSpatialTfSamplerForward(cudnnHandle handle, cudnnSpatialTransformerDescriptor stDesc, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer grid, jcuda.Pointer beta, cudnnTensorDescriptor yDesc, jcuda.Pointer y)
public static int cudnnSpatialTfSamplerBackward(cudnnHandle handle, cudnnSpatialTransformerDescriptor stDesc, jcuda.Pointer alpha, cudnnTensorDescriptor xDesc, jcuda.Pointer x, jcuda.Pointer beta, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx, jcuda.Pointer alphaDgrid, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, jcuda.Pointer grid, jcuda.Pointer betaDgrid, jcuda.Pointer dgrid)
public static int cudnnCreateDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc)
public static int cudnnDestroyDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc)
public static int cudnnDropoutGetStatesSize(cudnnHandle handle, long[] sizeInBytes)
public static int cudnnDropoutGetReserveSpaceSize(cudnnTensorDescriptor xdesc, long[] sizeInBytes)
public static int cudnnSetDropoutDescriptor(cudnnDropoutDescriptor dropoutDesc, cudnnHandle handle, float dropout, jcuda.Pointer states, long stateSizeInBytes, long seed)
public static int cudnnDropoutForward(cudnnHandle handle, cudnnDropoutDescriptor dropoutDesc, cudnnTensorDescriptor xdesc, jcuda.Pointer x, cudnnTensorDescriptor ydesc, jcuda.Pointer y, jcuda.Pointer reserveSpace, long reserveSpaceSizeInBytes)
public static int cudnnDropoutBackward(cudnnHandle handle, cudnnDropoutDescriptor dropoutDesc, cudnnTensorDescriptor dydesc, jcuda.Pointer dy, cudnnTensorDescriptor dxdesc, jcuda.Pointer dx, jcuda.Pointer reserveSpace, long reserveSpaceSizeInBytes)
public static int cudnnCreateRNNDescriptor(cudnnRNNDescriptor rnnDesc)
public static int cudnnDestroyRNNDescriptor(cudnnRNNDescriptor rnnDesc)
public static int cudnnSetRNNDescriptor(cudnnRNNDescriptor rnnDesc, int hiddenSize, int numLayers, cudnnDropoutDescriptor dropoutDesc, int inputMode, int direction, int mode, int dataType)
public static int cudnnGetRNNWorkspaceSize(cudnnHandle handle, cudnnRNNDescriptor rnnDesc, int seqLength, cudnnTensorDescriptor xDesc, long[] sizeInBytes)
public static int cudnnGetRNNTrainingReserveSize(cudnnHandle handle, cudnnRNNDescriptor rnnDesc, int seqLength, cudnnTensorDescriptor xDesc, long[] sizeInBytes)
public static int cudnnGetRNNParamsSize(cudnnHandle handle, cudnnRNNDescriptor rnnDesc, cudnnTensorDescriptor xDesc, long[] sizeInBytes, int dataType)
public static int cudnnGetRNNLinLayerMatrixParams(cudnnHandle handle, cudnnRNNDescriptor rnnDesc, int layer, cudnnTensorDescriptor xDesc, cudnnFilterDescriptor wDesc, jcuda.Pointer w, int linLayerID, cudnnFilterDescriptor linLayerMatDesc, jcuda.Pointer linLayerMat)
public static int cudnnGetRNNLinLayerBiasParams(cudnnHandle handle, cudnnRNNDescriptor rnnDesc, int layer, cudnnTensorDescriptor xDesc, cudnnFilterDescriptor wDesc, jcuda.Pointer w, int linLayerID, cudnnFilterDescriptor linLayerBiasDesc, jcuda.Pointer linLayerBias)
public static int cudnnRNNForwardInference(cudnnHandle handle, cudnnRNNDescriptor rnnDesc, int seqLength, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor hxDesc, jcuda.Pointer hx, cudnnTensorDescriptor cxDesc, jcuda.Pointer cx, cudnnFilterDescriptor wDesc, jcuda.Pointer w, cudnnTensorDescriptor yDesc, jcuda.Pointer y, cudnnTensorDescriptor hyDesc, jcuda.Pointer hy, cudnnTensorDescriptor cyDesc, jcuda.Pointer cy, jcuda.Pointer workspace, long workSpaceSizeInBytes)
public static int cudnnRNNForwardTraining(cudnnHandle handle, cudnnRNNDescriptor rnnDesc, int seqLength, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor hxDesc, jcuda.Pointer hx, cudnnTensorDescriptor cxDesc, jcuda.Pointer cx, cudnnFilterDescriptor wDesc, jcuda.Pointer w, cudnnTensorDescriptor yDesc, jcuda.Pointer y, cudnnTensorDescriptor hyDesc, jcuda.Pointer hy, cudnnTensorDescriptor cyDesc, jcuda.Pointer cy, jcuda.Pointer workspace, long workSpaceSizeInBytes, jcuda.Pointer reserveSpace, long reserveSpaceSizeInBytes)
public static int cudnnRNNBackwardData(cudnnHandle handle, cudnnRNNDescriptor rnnDesc, int seqLength, cudnnTensorDescriptor yDesc, jcuda.Pointer y, cudnnTensorDescriptor dyDesc, jcuda.Pointer dy, cudnnTensorDescriptor dhyDesc, jcuda.Pointer dhy, cudnnTensorDescriptor dcyDesc, jcuda.Pointer dcy, cudnnFilterDescriptor wDesc, jcuda.Pointer w, cudnnTensorDescriptor hxDesc, jcuda.Pointer hx, cudnnTensorDescriptor cxDesc, jcuda.Pointer cx, cudnnTensorDescriptor dxDesc, jcuda.Pointer dx, cudnnTensorDescriptor dhxDesc, jcuda.Pointer dhx, cudnnTensorDescriptor dcxDesc, jcuda.Pointer dcx, jcuda.Pointer workspace, long workSpaceSizeInBytes, jcuda.Pointer reserveSpace, long reserveSpaceSizeInBytes)
public static int cudnnRNNBackwardWeights(cudnnHandle handle, cudnnRNNDescriptor rnnDesc, int seqLength, cudnnTensorDescriptor xDesc, jcuda.Pointer x, cudnnTensorDescriptor hxDesc, jcuda.Pointer hx, cudnnTensorDescriptor yDesc, jcuda.Pointer y, jcuda.Pointer workspace, long workSpaceSizeInBytes, cudnnFilterDescriptor dwDesc, jcuda.Pointer dw, jcuda.Pointer reserveSpace, long reserveSpaceSizeInBytes)
Copyright © 2017. All rights reserved.