Modifier and Type | Method and Description |
---|---|
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.cudnnCreateConvolutionDescriptor(cudnnConvolutionDescriptor convDesc)
Create an instance of convolution descriptor
|
static int |
JCudnn.cudnnDestroyConvolutionDescriptor(cudnnConvolutionDescriptor convDesc)
Destroy an instance of convolution descriptor
|
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.cudnnGetConvolution2dDescriptor(cudnnConvolutionDescriptor convDesc,
int[] pad_h,
int[] pad_w,
int[] u,
int[] v,
int[] dilation_h,
int[] dilation_w,
int[] mode,
int[] computeType) |
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.cudnnGetConvolutionGroupCount(cudnnConvolutionDescriptor convDesc,
int[] groupCount) |
static int |
JCudnn.cudnnGetConvolutionMathType(cudnnConvolutionDescriptor convDesc,
int[] mathType) |
static int |
JCudnn.cudnnGetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc,
int arrayLengthRequested,
int[] arrayLength,
int[] padA,
int[] strideA,
int[] dilationA,
int[] mode,
int[] computeType)
Helper function to return the dimensions of the output tensor given a convolution descriptor
|
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.cudnnGetConvolutionReorderType(cudnnConvolutionDescriptor convDesc,
int[] reorderType) |
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.cudnnIm2Col(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
Pointer x,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
Pointer colBuffer) |
static int |
JCudnn.cudnnSetConvolution2dDescriptor(cudnnConvolutionDescriptor convDesc,
int pad_h,
int pad_w,
int u,
int v,
int dilation_h,
int dilation_w,
int mode,
int computeType) |
static int |
JCudnn.cudnnSetConvolutionGroupCount(cudnnConvolutionDescriptor convDesc,
int groupCount) |
static int |
JCudnn.cudnnSetConvolutionMathType(cudnnConvolutionDescriptor convDesc,
int mathType) |
static int |
JCudnn.cudnnSetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc,
int arrayLength,
int[] padA,
int[] filterStrideA,
int[] dilationA,
int mode,
int computeType) |
static int |
JCudnn.cudnnSetConvolutionReorderType(cudnnConvolutionDescriptor convDesc,
int reorderType) |
Copyright © 2020. All rights reserved.