| Modifier and Type | Method and Description |
|---|---|
static int |
JCudnn.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 |
JCudnn.cudnnActivationForward(cudnnHandle handle,
cudnnActivationDescriptor activationDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
static int |
JCudnn.cudnnAddTensor(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor aDesc,
jcuda.Pointer A,
jcuda.Pointer beta,
cudnnTensorDescriptor cDesc,
jcuda.Pointer C) |
static int |
JCudnn.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 |
JCudnn.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 |
JCudnn.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 |
JCudnn.cudnnConvolutionBackwardBias(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
jcuda.Pointer beta,
cudnnTensorDescriptor dbDesc,
jcuda.Pointer db) |
static int |
JCudnn.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 |
JCudnn.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 |
JCudnn.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 |
JCudnn.cudnnCreateTensorDescriptor(cudnnTensorDescriptor tensorDesc) |
static int |
JCudnn.cudnnDeriveBNTensorDescriptor(cudnnTensorDescriptor derivedBnDesc,
cudnnTensorDescriptor xDesc,
int mode) |
static int |
JCudnn.cudnnDestroyTensorDescriptor(cudnnTensorDescriptor tensorDesc) |
static int |
JCudnn.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 |
JCudnn.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 |
JCudnn.cudnnDropoutBackward(cudnnHandle handle,
cudnnDropoutDescriptor dropoutDesc,
cudnnTensorDescriptor dydesc,
jcuda.Pointer dy,
cudnnTensorDescriptor dxdesc,
jcuda.Pointer dx,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnDropoutForward(cudnnHandle handle,
cudnnDropoutDescriptor dropoutDesc,
cudnnTensorDescriptor xdesc,
jcuda.Pointer x,
cudnnTensorDescriptor ydesc,
jcuda.Pointer y,
jcuda.Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
JCudnn.cudnnDropoutGetReserveSpaceSize(cudnnTensorDescriptor xdesc,
long[] sizeInBytes) |
static int |
JCudnn.cudnnFindConvolutionBackwardDataAlgorithm(cudnnHandle handle,
cudnnFilterDescriptor wDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor dxDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf[] perfResults) |
static int |
JCudnn.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 |
JCudnn.cudnnFindConvolutionBackwardFilterAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor dwDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf[] perfResults) |
static int |
JCudnn.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 |
JCudnn.cudnnFindConvolutionForwardAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf[] perfResults) |
static int |
JCudnn.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 |
JCudnn.cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor inputTensorDesc,
cudnnFilterDescriptor filterDesc,
int[] n,
int[] c,
int[] h,
int[] w) |
static int |
JCudnn.cudnnGetConvolutionBackwardDataAlgorithm(cudnnHandle handle,
cudnnFilterDescriptor wDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor dxDesc,
int preference,
long memoryLimitInBytes,
int[] algo) |
static int |
JCudnn.cudnnGetConvolutionBackwardDataWorkspaceSize(cudnnHandle handle,
cudnnFilterDescriptor wDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor dxDesc,
int algo,
long[] sizeInBytes) |
static int |
JCudnn.cudnnGetConvolutionBackwardFilterAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor dwDesc,
int preference,
long memoryLimitInBytes,
int[] algo) |
static int |
JCudnn.cudnnGetConvolutionBackwardFilterWorkspaceSize(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor gradDesc,
int algo,
long[] sizeInBytes) |
static int |
JCudnn.cudnnGetConvolutionForwardAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
int preference,
long memoryLimitInBytes,
int[] algo) |
static int |
JCudnn.cudnnGetConvolutionForwardWorkspaceSize(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
int algo,
long[] sizeInBytes) |
static int |
JCudnn.cudnnGetConvolutionNdForwardOutputDim(cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor inputTensorDesc,
cudnnFilterDescriptor filterDesc,
int nbDims,
int[] tensorOuputDimA) |
static int |
JCudnn.cudnnGetPooling2dForwardOutputDim(cudnnPoolingDescriptor poolingDesc,
cudnnTensorDescriptor inputTensorDesc,
int[] n,
int[] c,
int[] h,
int[] w) |
static int |
JCudnn.cudnnGetPoolingNdForwardOutputDim(cudnnPoolingDescriptor poolingDesc,
cudnnTensorDescriptor inputTensorDesc,
int nbDims,
int[] outputTensorDimA) |
static int |
JCudnn.cudnnGetRNNLinLayerBiasParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int layer,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
int linLayerID,
cudnnFilterDescriptor linLayerBiasDesc,
jcuda.Pointer linLayerBias) |
static int |
JCudnn.cudnnGetRNNLinLayerMatrixParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int layer,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
jcuda.Pointer w,
int linLayerID,
cudnnFilterDescriptor linLayerMatDesc,
jcuda.Pointer linLayerMat) |
static int |
JCudnn.cudnnGetRNNParamsSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnTensorDescriptor xDesc,
long[] sizeInBytes,
int dataType) |
static int |
JCudnn.cudnnGetRNNTrainingReserveSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
long[] sizeInBytes) |
static int |
JCudnn.cudnnGetRNNWorkspaceSize(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor xDesc,
long[] sizeInBytes) |
static int |
JCudnn.cudnnGetTensor4dDescriptor(cudnnTensorDescriptor tensorDesc,
int[] dataType,
int[] n,
int[] c,
int[] h,
int[] w,
int[] nStride,
int[] cStride,
int[] hStride,
int[] wStride) |
static int |
JCudnn.cudnnGetTensorNdDescriptor(cudnnTensorDescriptor tensorDesc,
int nbDimsRequested,
int[] dataType,
int[] nbDims,
int[] dimA,
int[] strideA) |
static int |
JCudnn.cudnnIm2Col(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
jcuda.Pointer colBuffer) |
static int |
JCudnn.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 |
JCudnn.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 |
JCudnn.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 |
JCudnn.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 |
JCudnn.cudnnPoolingForward(cudnnHandle handle,
cudnnPoolingDescriptor poolingDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
static int |
JCudnn.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 |
JCudnn.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 |
JCudnn.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 |
JCudnn.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 |
JCudnn.cudnnScaleTensor(cudnnHandle handle,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
jcuda.Pointer alpha) |
static int |
JCudnn.cudnnSetTensor(cudnnHandle handle,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
jcuda.Pointer valuePtr) |
static int |
JCudnn.cudnnSetTensor4dDescriptor(cudnnTensorDescriptor tensorDesc,
int format,
int dataType,
int n,
int c,
int h,
int w) |
static int |
JCudnn.cudnnSetTensor4dDescriptorEx(cudnnTensorDescriptor tensorDesc,
int dataType,
int n,
int c,
int h,
int w,
int nStride,
int cStride,
int hStride,
int wStride) |
static int |
JCudnn.cudnnSetTensorNdDescriptor(cudnnTensorDescriptor tensorDesc,
int dataType,
int nbDims,
int[] dimA,
int[] strideA) |
static int |
JCudnn.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 |
JCudnn.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 |
JCudnn.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 |
JCudnn.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 |
JCudnn.cudnnTransformTensor(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
Copyright © 2017. All rights reserved.