Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.cudnnActivationBackward_v4(cudnnHandle handle,
cudnnActivationDescriptor activationDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx) |
static int |
JCudnn.cudnnActivationBackward(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx)
Function to perform backward activation
|
static int |
JCudnn.cudnnActivationForward_v4(cudnnHandle handle,
cudnnActivationDescriptor activationDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
static int |
JCudnn.cudnnActivationForward(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Function to perform forward activation
|
static int |
JCudnn.cudnnAddTensor_v2(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor bDesc,
jcuda.Pointer b,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
static int |
JCudnn.cudnnAddTensor_v3(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor bDesc,
jcuda.Pointer b,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
cudnnAddTensor_v3 is now mapped to cudnnAddTensor
and will be removed at the same time as cudnnAddTensor_v2
Use cudnnAddTensor instead
|
static int |
JCudnn.cudnnAddTensor(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor bDesc,
jcuda.Pointer b,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Tensor Bias addition : y = alpha * b + beta * y
|
static int |
JCudnn.cudnnBatchNormalizationBackward(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
jcuda.Pointer beta,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx,
cudnnTensorDescriptor dBnScaleBiasDesc,
jcuda.Pointer bnScale,
jcuda.Pointer dBnScaleResult,
jcuda.Pointer dBnBiasResult,
double epsilon,
jcuda.Pointer savedMean,
jcuda.Pointer savedInvVariance)
Performs backward pass of Batch Normalization layer.
|
static int |
JCudnn.cudnnBatchNormalizationForwardInference(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
jcuda.Pointer beta,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor bnScaleBiasMeanVarDesc,
jcuda.Pointer bnScale,
jcuda.Pointer bnBias,
jcuda.Pointer estimatedMean,
jcuda.Pointer estimatedInvVariance,
double epsilon)
Performs Batch Normalization during Inference:
y[i] = bnScale[k]*(x[i]-estimatedMean[k])*estimatedInvVariance[k] + bnBias[k]
with bnScale, bnBias, runningMean, runningInvVariance tensors indexed
according to spatial or per-activation mode.
|
static int |
JCudnn.cudnnBatchNormalizationForwardTraining(cudnnHandle handle,
int mode,
jcuda.Pointer alpha,
jcuda.Pointer beta,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor bnScaleBiasMeanVarDesc,
jcuda.Pointer bnScale,
jcuda.Pointer bnBias,
double exponentialAverageFactor,
jcuda.Pointer resultRunningMean,
jcuda.Pointer resultRunningInvVariance,
double epsilon,
jcuda.Pointer resultSaveMean,
jcuda.Pointer resultSaveInvVariance)
Computes y = BN(x).
|
static int |
JCudnn.cudnnConvolutionBackwardBias(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
jcuda.Pointer beta,
cudnnTensorDescriptor dbDesc,
jcuda.Pointer db)
Function to compute the bias gradient for batch convolution
|
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.cudnnCreate(cudnnHandle handle) |
static int |
JCudnn.cudnnDestroy(cudnnHandle handle) |
static int |
JCudnn.cudnnDivisiveNormalizationBackward(cudnnHandle handle,
cudnnLRNDescriptor normDesc,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer means,
jcuda.Pointer dy,
jcuda.Pointer temp,
jcuda.Pointer temp2,
jcuda.Pointer beta,
cudnnTensorDescriptor dXdMeansDesc,
jcuda.Pointer dx,
jcuda.Pointer dMeans) |
static int |
JCudnn.cudnnDivisiveNormalizationForward(cudnnHandle handle,
cudnnLRNDescriptor normDesc,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer means,
jcuda.Pointer temp,
jcuda.Pointer temp2,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
LCN/divisive normalization functions: y = alpha * normalize(x) + beta * y
|
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.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.cudnnGetStream(cudnnHandle handle,
jcuda.runtime.cudaStream_t streamId) |
static int |
JCudnn.cudnnIm2Col(cudnnHandle handle,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
cudnnFilterDescriptor wDesc,
cudnnConvolutionDescriptor convDesc,
jcuda.Pointer colBuffer) |
static int |
JCudnn.cudnnLRNCrossChannelBackward(cudnnHandle handle,
cudnnLRNDescriptor normDesc,
int lrnMode,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx)
LRN cross-channel backward computation.
|
static int |
JCudnn.cudnnLRNCrossChannelForward(cudnnHandle handle,
cudnnLRNDescriptor normDesc,
int lrnMode,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
LRN cross-channel forward computation.
|
static int |
JCudnn.cudnnPoolingBackward(cudnnHandle handle,
cudnnPoolingDescriptor poolingDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx)
Function to perform backward pooling
|
static int |
JCudnn.cudnnPoolingForward(cudnnHandle handle,
cudnnPoolingDescriptor poolingDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Function to perform forward pooling
|
static int |
JCudnn.cudnnScaleTensor(cudnnHandle handle,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
jcuda.Pointer alpha)
Scale all values of a tensor by a given factor : y[i] = alpha * y[i]
|
static int |
JCudnn.cudnnSetStream(cudnnHandle handle,
jcuda.runtime.cudaStream_t streamId) |
static int |
JCudnn.cudnnSetTensor(cudnnHandle handle,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
jcuda.Pointer valuePtr)
Set all values of a tensor to a given value : y[i] = value[0]
|
static int |
JCudnn.cudnnSoftmaxBackward(cudnnHandle handle,
int algo,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx)
Function to perform backward softmax
|
static int |
JCudnn.cudnnSoftmaxForward(cudnnHandle handle,
int algo,
int mode,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Function to perform forward softmax
|
static int |
JCudnn.cudnnTransformTensor(cudnnHandle handle,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y)
Tensor layout conversion helper (y = alpha * x + beta * y)
|
Copyright © 2016. All rights reserved.