static int |
JCudnn.cudnnConvolutionBackwardData(cudnnHandle handle,
Pointer alpha,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor dyDesc,
Pointer dy,
cudnnConvolutionDescriptor convDesc,
int algo,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer beta,
cudnnTensorDescriptor dxDesc,
Pointer dx) |
static int |
JCudnn.cudnnConvolutionBackwardFilter(cudnnHandle handle,
Pointer alpha,
cudnnTensorDescriptor xDesc,
Pointer x,
cudnnTensorDescriptor dyDesc,
Pointer dy,
cudnnConvolutionDescriptor convDesc,
int algo,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer beta,
cudnnFilterDescriptor dwDesc,
Pointer dw) |
static int |
JCudnn.cudnnConvolutionBiasActivationForward(cudnnHandle handle,
Pointer alpha1,
cudnnTensorDescriptor xDesc,
Pointer x,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnConvolutionDescriptor convDesc,
int algo,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer alpha2,
cudnnTensorDescriptor zDesc,
Pointer z,
cudnnTensorDescriptor biasDesc,
Pointer bias,
cudnnActivationDescriptor activationDesc,
cudnnTensorDescriptor yDesc,
Pointer y)
Fused conv/bias/activation operation : y = Act( alpha1 * conv(x) + alpha2 * z + bias )
|
static int |
JCudnn.cudnnConvolutionForward(cudnnHandle handle,
Pointer alpha,
cudnnTensorDescriptor xDesc,
Pointer x,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnConvolutionDescriptor convDesc,
int algo,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer beta,
cudnnTensorDescriptor yDesc,
Pointer y)
Function to perform the forward pass for batch convolution
|
static int |
JCudnn.cudnnCreateFilterDescriptor(cudnnFilterDescriptor filterDesc)
Create an instance of FilterStruct
|
static int |
JCudnn.cudnnDestroyFilterDescriptor(cudnnFilterDescriptor filterDesc) |
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,
Pointer w,
cudnnTensorDescriptor dyDesc,
Pointer dy,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor dxDesc,
Pointer dx,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf[] perfResults,
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,
Pointer x,
cudnnTensorDescriptor dyDesc,
Pointer y,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor dwDesc,
Pointer dw,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf[] perfResults,
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,
Pointer x,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
Pointer y,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf[] perfResults,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
JCudnn.cudnnFindRNNBackwardDataAlgorithmEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor[] dyDesc,
Pointer dy,
cudnnTensorDescriptor dhyDesc,
Pointer dhy,
cudnnTensorDescriptor dcyDesc,
Pointer dcy,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnTensorDescriptor[] dxDesc,
Pointer dx,
cudnnTensorDescriptor dhxDesc,
Pointer dhx,
cudnnTensorDescriptor dcxDesc,
Pointer dcx,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformance[] perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor[] yDesc,
Pointer y,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformance[] perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterDescriptor dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnFindRNNForwardInferenceAlgorithmEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformance[] perfResults,
Pointer workspace,
long workSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnFindRNNForwardTrainingAlgorithmEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformance[] perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor inputTensorDesc,
cudnnFilterDescriptor filterDesc,
int[] n,
int[] c,
int[] h,
int[] w)
convolution data type
|
static int |
JCudnn.cudnnGetConvolutionBackwardDataAlgorithm_v7(cudnnHandle handle,
cudnnFilterDescriptor filterDesc,
cudnnTensorDescriptor diffDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor gradDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf[] perfResults) |
static int |
JCudnn.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 |
JCudnn.cudnnGetConvolutionBackwardFilterAlgorithm_v7(cudnnHandle handle,
cudnnTensorDescriptor srcDesc,
cudnnTensorDescriptor diffDesc,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor gradDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf[] perfResults) |
static int |
JCudnn.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 |
JCudnn.cudnnGetConvolutionForwardAlgorithm_v7(cudnnHandle handle,
cudnnTensorDescriptor srcDesc,
cudnnFilterDescriptor filterDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor destDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf[] perfResults) |
static int |
JCudnn.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 |
JCudnn.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 int |
JCudnn.cudnnGetFilter4dDescriptor(cudnnFilterDescriptor filterDesc,
int[] dataType,
int[] format,
int[] k,
int[] c,
int[] h,
int[] w)
width of each input filter
|
static int |
JCudnn.cudnnGetFilterNdDescriptor(cudnnFilterDescriptor filterDesc,
int nbDimsRequested,
int[] dataType,
int[] format,
int[] nbDims,
int[] filterDimA) |
static int |
JCudnn.cudnnGetFilterSizeInBytes(cudnnFilterDescriptor filterDesc,
long[] size) |
static int |
JCudnn.cudnnGetFoldedConvBackwardDataDescriptors(cudnnHandle handle,
cudnnFilterDescriptor filterDesc,
cudnnTensorDescriptor diffDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor gradDesc,
int transformFormat,
cudnnFilterDescriptor foldedFilterDesc,
cudnnTensorDescriptor paddedDiffDesc,
cudnnConvolutionDescriptor foldedConvDesc,
cudnnTensorDescriptor foldedGradDesc,
cudnnTensorTransformDescriptor filterFoldTransDesc,
cudnnTensorTransformDescriptor diffPadTransDesc,
cudnnTensorTransformDescriptor gradFoldTransDesc,
cudnnTensorTransformDescriptor gradUnfoldTransDesc)
Helper function to calculate folding descriptors for dgrad
|
static int |
JCudnn.cudnnGetRNNLinLayerBiasParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int pseudoLayer,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
Pointer w,
int linLayerID,
cudnnFilterDescriptor linLayerBiasDesc,
Pointer linLayerBias)
Deprecated.
|
static int |
JCudnn.cudnnGetRNNLinLayerMatrixParams(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int pseudoLayer,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
Pointer w,
int linLayerID,
cudnnFilterDescriptor linLayerMatDesc,
Pointer linLayerMat)
Deprecated.
|
static int |
JCudnn.cudnnIm2Col(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
Pointer x,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
Pointer colBuffer) |
static int |
JCudnn.cudnnReorderFilterAndBias(cudnnHandle handle,
cudnnFilterDescriptor filterDesc,
int reorderType,
Pointer filterData,
Pointer reorderedFilterData,
int reorderBias,
Pointer biasData,
Pointer reorderedBiasData) |
static int |
JCudnn.cudnnRNNBackwardData(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor[] dyDesc,
Pointer dy,
cudnnTensorDescriptor dhyDesc,
Pointer dhy,
cudnnTensorDescriptor dcyDesc,
Pointer dcy,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnTensorDescriptor[] dxDesc,
Pointer dx,
cudnnTensorDescriptor dhxDesc,
Pointer dhx,
cudnnTensorDescriptor dcxDesc,
Pointer dcx,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnRNNBackwardDataEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnRNNDataDescriptor yDesc,
Pointer y,
cudnnRNNDataDescriptor dyDesc,
Pointer dy,
cudnnRNNDataDescriptor dcDesc,
Pointer dcAttn,
cudnnTensorDescriptor dhyDesc,
Pointer dhy,
cudnnTensorDescriptor dcyDesc,
Pointer dcy,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnRNNDataDescriptor dxDesc,
Pointer dx,
cudnnTensorDescriptor dhxDesc,
Pointer dhx,
cudnnTensorDescriptor dcxDesc,
Pointer dcx,
cudnnRNNDataDescriptor dkDesc,
Pointer dkeys,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnRNNBackwardWeights(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor[] yDesc,
Pointer y,
Pointer workSpace,
long workSpaceSizeInBytes,
cudnnFilterDescriptor dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnRNNBackwardWeightsEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnRNNDataDescriptor xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnRNNDataDescriptor yDesc,
Pointer y,
Pointer workSpace,
long workSpaceSizeInBytes,
cudnnFilterDescriptor dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnRNNForwardInference(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
Pointer workSpace,
long workSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnRNNForwardInferenceEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnRNNDataDescriptor xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnRNNDataDescriptor yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
cudnnRNNDataDescriptor kDesc,
Pointer keys,
cudnnRNNDataDescriptor cDesc,
Pointer cAttn,
cudnnRNNDataDescriptor iDesc,
Pointer iAttn,
cudnnRNNDataDescriptor qDesc,
Pointer queries,
Pointer workSpace,
long workSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnRNNForwardTraining(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
int seqLength,
cudnnTensorDescriptor[] xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnTensorDescriptor[] yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnRNNForwardTrainingEx(cudnnHandle handle,
cudnnRNNDescriptor rnnDesc,
cudnnRNNDataDescriptor xDesc,
Pointer x,
cudnnTensorDescriptor hxDesc,
Pointer hx,
cudnnTensorDescriptor cxDesc,
Pointer cx,
cudnnFilterDescriptor wDesc,
Pointer w,
cudnnRNNDataDescriptor yDesc,
Pointer y,
cudnnTensorDescriptor hyDesc,
Pointer hy,
cudnnTensorDescriptor cyDesc,
Pointer cy,
cudnnRNNDataDescriptor kDesc,
Pointer keys,
cudnnRNNDataDescriptor cDesc,
Pointer cAttn,
cudnnRNNDataDescriptor iDesc,
Pointer iAttn,
cudnnRNNDataDescriptor qDesc,
Pointer queries,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes)
Deprecated.
|
static int |
JCudnn.cudnnSetFilter4dDescriptor(cudnnFilterDescriptor filterDesc,
int dataType,
int format,
int k,
int c,
int h,
int w) |
static int |
JCudnn.cudnnSetFilterNdDescriptor(cudnnFilterDescriptor filterDesc,
int dataType,
int format,
int nbDims,
int[] filterDimA)
width of each input filter
|
static int |
JCudnn.cudnnTransformFilter(cudnnHandle handle,
cudnnTensorTransformDescriptor transDesc,
Pointer alpha,
cudnnFilterDescriptor srcDesc,
Pointer srcData,
Pointer beta,
cudnnFilterDescriptor destDesc,
Pointer destData) |