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.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.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.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.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_v4(cudnnFilterDescriptor filterDesc,
int[] dataType,
int[] format,
jcuda.Pointer k,
jcuda.Pointer c,
jcuda.Pointer h,
jcuda.Pointer w) |
static int |
JCudnn.cudnnGetFilter4dDescriptor(cudnnFilterDescriptor filterDesc,
int[] dataType,
jcuda.Pointer k,
jcuda.Pointer c,
jcuda.Pointer h,
jcuda.Pointer w) |
static int |
JCudnn.cudnnGetFilterNdDescriptor_v4(cudnnFilterDescriptor filterDesc,
int nbDimsRequested,
int[] dataType,
int[] format,
jcuda.Pointer nbDims,
int[] filterDimA) |
static int |
JCudnn.cudnnGetFilterNdDescriptor(cudnnFilterDescriptor filterDesc,
int nbDimsRequested,
int[] dataType,
jcuda.Pointer nbDims,
int[] filterDimA) |
static int |
JCudnn.cudnnIm2Col(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
jcuda.Pointer colBuffer) |
static int |
JCudnn.cudnnSetFilter4dDescriptor_v4(cudnnFilterDescriptor filterDesc,
int dataType,
int format,
int k,
int c,
int h,
int w) |
static int |
JCudnn.cudnnSetFilter4dDescriptor(cudnnFilterDescriptor filterDesc,
int dataType,
int k,
int c,
int h,
int w) |
static int |
JCudnn.cudnnSetFilterNdDescriptor_v4(cudnnFilterDescriptor filterDesc,
int dataType,
int format,
int nbDims,
int[] filterDimA) |
static int |
JCudnn.cudnnSetFilterNdDescriptor(cudnnFilterDescriptor filterDesc,
int dataType,
int nbDims,
int[] filterDimA) |
Copyright © 2016. All rights reserved.