Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.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 |
JCudnn.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 |
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_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 |
JCudnn.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 |
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)
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.cudnnFindConvolutionBackwardFilterAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnTensorDescriptor dyDesc,
cudnnConvolutionDescriptor convDesc,
cudnnFilterDescriptor dwDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf[] perfResults) |
static int |
JCudnn.cudnnFindConvolutionForwardAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf[] perfResults) |
static int |
JCudnn.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 |
JCudnn.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 |
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)
Helper function to return the minimum size of the workspace to be passed to the convolution given an algo
|
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)
Helper function to return the minimum size of the workspace to be passed to the convolution given an algo
|
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)
Helper function to return the minimum size of the workspace to be passed to the convolution given an algo
|
static int |
JCudnn.cudnnGetConvolutionNdDescriptor_v2(cudnnConvolutionDescriptor convDesc,
int arrayLengthRequested,
int[] arrayLength,
int[] padA,
int[] strideA,
int[] upscaleA,
int[] mode) |
static int |
JCudnn.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 |
JCudnn.cudnnGetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc,
int arrayLengthRequested,
int[] arrayLength,
int[] padA,
int[] strideA,
int[] upscaleA,
int[] mode,
int[] dataType) |
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.cudnnIm2Col(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
jcuda.Pointer colBuffer) |
static int |
JCudnn.cudnnSetConvolution2dDescriptor(cudnnConvolutionDescriptor convDesc,
int pad_h,
int pad_w,
int u,
int v,
int upscalex,
int upscaley,
int mode) |
static int |
JCudnn.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 |
JCudnn.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 |
JCudnn.cudnnSetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc,
int arrayLength,
int[] padA,
int[] filterStrideA,
int[] upscaleA,
int mode,
int dataType) |
Copyright © 2016. All rights reserved.