Package | Description |
---|---|
org.bytedeco.cuda.global |
Modifier and Type | Method and Description |
---|---|
static int |
cudnn.cudnnActivationBackward(cudnnContext handle,
cudnnActivationStruct activationDesc,
Pointer alpha,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnTensorStruct xDesc,
Pointer x,
Pointer beta,
cudnnTensorStruct dxDesc,
Pointer dx) |
static int |
cudnn.cudnnActivationForward(cudnnContext handle,
cudnnActivationStruct activationDesc,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
Pointer beta,
cudnnTensorStruct yDesc,
Pointer y) |
static int |
cudnn.cudnnBatchNormalizationBackwardEx(cudnnContext handle,
int mode,
int bnOps,
Pointer alphaDataDiff,
Pointer betaDataDiff,
Pointer alphaParamDiff,
Pointer betaParamDiff,
cudnnTensorStruct xDesc,
Pointer xData,
cudnnTensorStruct yDesc,
Pointer yData,
cudnnTensorStruct dyDesc,
Pointer dyData,
cudnnTensorStruct dzDesc,
Pointer dzData,
cudnnTensorStruct dxDesc,
Pointer dxData,
cudnnTensorStruct dBnScaleBiasDesc,
Pointer bnScaleData,
Pointer bnBiasData,
Pointer dBnScaleData,
Pointer dBnBiasData,
double epsilon,
Pointer savedMean,
Pointer savedInvVariance,
cudnnActivationStruct activationDesc,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnn.cudnnBatchNormalizationForwardTrainingEx(cudnnContext handle,
int mode,
int bnOps,
Pointer alpha,
Pointer beta,
cudnnTensorStruct xDesc,
Pointer xData,
cudnnTensorStruct zDesc,
Pointer zData,
cudnnTensorStruct yDesc,
Pointer yData,
cudnnTensorStruct bnScaleBiasMeanVarDesc,
Pointer bnScale,
Pointer bnBias,
double exponentialAverageFactor,
Pointer resultRunningMean,
Pointer resultRunningVariance,
double epsilon,
Pointer resultSaveMean,
Pointer resultSaveInvVariance,
cudnnActivationStruct activationDesc,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnn.cudnnConvolutionBiasActivationForward(cudnnContext handle,
Pointer alpha1,
cudnnTensorStruct xDesc,
Pointer x,
cudnnFilterStruct wDesc,
Pointer w,
cudnnConvolutionStruct convDesc,
int algo,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer alpha2,
cudnnTensorStruct zDesc,
Pointer z,
cudnnTensorStruct biasDesc,
Pointer bias,
cudnnActivationStruct activationDesc,
cudnnTensorStruct yDesc,
Pointer y) |
static int |
cudnn.cudnnCreateActivationDescriptor(cudnnActivationStruct activationDesc) |
static int |
cudnn.cudnnDestroyActivationDescriptor(cudnnActivationStruct activationDesc) |
static int |
cudnn.cudnnGetActivationDescriptor(cudnnActivationStruct activationDesc,
int[] mode,
int[] reluNanOpt,
double[] coef) |
static int |
cudnn.cudnnGetActivationDescriptor(cudnnActivationStruct activationDesc,
IntBuffer mode,
IntBuffer reluNanOpt,
DoubleBuffer coef) |
static int |
cudnn.cudnnGetActivationDescriptor(cudnnActivationStruct activationDesc,
IntPointer mode,
IntPointer reluNanOpt,
DoublePointer coef) |
static int |
cudnn.cudnnGetActivationDescriptorSwishBeta(cudnnActivationStruct activationDesc,
double[] swish_beta) |
static int |
cudnn.cudnnGetActivationDescriptorSwishBeta(cudnnActivationStruct activationDesc,
DoubleBuffer swish_beta) |
static int |
cudnn.cudnnGetActivationDescriptorSwishBeta(cudnnActivationStruct activationDesc,
DoublePointer swish_beta) |
static int |
cudnn.cudnnGetBatchNormalizationBackwardExWorkspaceSize(cudnnContext handle,
int mode,
int bnOps,
cudnnTensorStruct xDesc,
cudnnTensorStruct yDesc,
cudnnTensorStruct dyDesc,
cudnnTensorStruct dzDesc,
cudnnTensorStruct dxDesc,
cudnnTensorStruct dBnScaleBiasDesc,
cudnnActivationStruct activationDesc,
SizeTPointer sizeInBytes) |
static int |
cudnn.cudnnGetBatchNormalizationForwardTrainingExWorkspaceSize(cudnnContext handle,
int mode,
int bnOps,
cudnnTensorStruct xDesc,
cudnnTensorStruct zDesc,
cudnnTensorStruct yDesc,
cudnnTensorStruct bnScaleBiasMeanVarDesc,
cudnnActivationStruct activationDesc,
SizeTPointer sizeInBytes) |
static int |
cudnn.cudnnGetBatchNormalizationTrainingExReserveSpaceSize(cudnnContext handle,
int mode,
int bnOps,
cudnnActivationStruct activationDesc,
cudnnTensorStruct xDesc,
SizeTPointer sizeInBytes) |
static int |
cudnn.cudnnGetNormalizationBackwardWorkspaceSize(cudnnContext handle,
int mode,
int normOps,
int algo,
cudnnTensorStruct xDesc,
cudnnTensorStruct yDesc,
cudnnTensorStruct dyDesc,
cudnnTensorStruct dzDesc,
cudnnTensorStruct dxDesc,
cudnnTensorStruct dNormScaleBiasDesc,
cudnnActivationStruct activationDesc,
cudnnTensorStruct normMeanVarDesc,
SizeTPointer sizeInBytes,
int groupCnt) |
static int |
cudnn.cudnnGetNormalizationForwardTrainingWorkspaceSize(cudnnContext handle,
int mode,
int normOps,
int algo,
cudnnTensorStruct xDesc,
cudnnTensorStruct zDesc,
cudnnTensorStruct yDesc,
cudnnTensorStruct normScaleBiasDesc,
cudnnActivationStruct activationDesc,
cudnnTensorStruct normMeanVarDesc,
SizeTPointer sizeInBytes,
int groupCnt) |
static int |
cudnn.cudnnGetNormalizationTrainingReserveSpaceSize(cudnnContext handle,
int mode,
int normOps,
int algo,
cudnnActivationStruct activationDesc,
cudnnTensorStruct xDesc,
SizeTPointer sizeInBytes,
int groupCnt) |
static int |
cudnn.cudnnNormalizationBackward(cudnnContext handle,
int mode,
int normOps,
int algo,
Pointer alphaDataDiff,
Pointer betaDataDiff,
Pointer alphaParamDiff,
Pointer betaParamDiff,
cudnnTensorStruct xDesc,
Pointer xData,
cudnnTensorStruct yDesc,
Pointer yData,
cudnnTensorStruct dyDesc,
Pointer dyData,
cudnnTensorStruct dzDesc,
Pointer dzData,
cudnnTensorStruct dxDesc,
Pointer dxData,
cudnnTensorStruct dNormScaleBiasDesc,
Pointer normScaleData,
Pointer normBiasData,
Pointer dNormScaleData,
Pointer dNormBiasData,
double epsilon,
cudnnTensorStruct normMeanVarDesc,
Pointer savedMean,
Pointer savedInvVariance,
cudnnActivationStruct activationDesc,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes,
int groupCnt) |
static int |
cudnn.cudnnNormalizationForwardInference(cudnnContext handle,
int mode,
int normOps,
int algo,
Pointer alpha,
Pointer beta,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct normScaleBiasDesc,
Pointer normScale,
Pointer normBias,
cudnnTensorStruct normMeanVarDesc,
Pointer estimatedMean,
Pointer estimatedVariance,
cudnnTensorStruct zDesc,
Pointer z,
cudnnActivationStruct activationDesc,
cudnnTensorStruct yDesc,
Pointer y,
double epsilon,
int groupCnt) |
static int |
cudnn.cudnnNormalizationForwardTraining(cudnnContext handle,
int mode,
int normOps,
int algo,
Pointer alpha,
Pointer beta,
cudnnTensorStruct xDesc,
Pointer xData,
cudnnTensorStruct normScaleBiasDesc,
Pointer normScale,
Pointer normBias,
double exponentialAverageFactor,
cudnnTensorStruct normMeanVarDesc,
Pointer resultRunningMean,
Pointer resultRunningVariance,
double epsilon,
Pointer resultSaveMean,
Pointer resultSaveInvVariance,
cudnnActivationStruct activationDesc,
cudnnTensorStruct zDesc,
Pointer zData,
cudnnTensorStruct yDesc,
Pointer yData,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes,
int groupCnt) |
static int |
cudnn.cudnnSetActivationDescriptor(cudnnActivationStruct activationDesc,
int mode,
int reluNanOpt,
double coef) |
static int |
cudnn.cudnnSetActivationDescriptorSwishBeta(cudnnActivationStruct activationDesc,
double swish_beta) |
Copyright © 2024. All rights reserved.