| Modifier and Type | Method and Description |
|---|---|
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.cudnnCreateFilterDescriptor(cudnnFilterDescriptor filterDesc) |
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,
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.cudnnGetFilter4dDescriptor(cudnnFilterDescriptor filterDesc,
int[] dataType,
int[] format,
int[] k,
int[] c,
int[] h,
int[] w) |
static int |
JCudnn.cudnnGetFilterNdDescriptor(cudnnFilterDescriptor filterDesc,
int nbDimsRequested,
int[] dataType,
int[] format,
int[] nbDims,
int[] filterDimA) |
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.cudnnIm2Col(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
jcuda.Pointer colBuffer) |
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.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) |
Copyright © 2017. All rights reserved.