public class cudnn extends cudnn
| Modifier and Type | Field and Description |
|---|---|
static int |
CUDNN_16BIT_INDICES
enum cudnnIndicesType_t
|
static int |
CUDNN_32BIT_INDICES
enum cudnnIndicesType_t
|
static int |
CUDNN_64BIT_INDICES
enum cudnnIndicesType_t
|
static int |
CUDNN_8BIT_INDICES
enum cudnnIndicesType_t
|
static int |
CUDNN_ACTIVATION_CLIPPED_RELU
enum cudnnActivationMode_t
|
static int |
CUDNN_ACTIVATION_ELU
enum cudnnActivationMode_t
|
static int |
CUDNN_ACTIVATION_IDENTITY
enum cudnnActivationMode_t
|
static int |
CUDNN_ACTIVATION_RELU
enum cudnnActivationMode_t
|
static int |
CUDNN_ACTIVATION_SIGMOID
enum cudnnActivationMode_t
|
static int |
CUDNN_ACTIVATION_TANH
enum cudnnActivationMode_t
|
static int |
CUDNN_ATTN_QUERYMAP_ALL_TO_ONE
enum cudnnAttnQueryMap_t
|
static int |
CUDNN_ATTN_QUERYMAP_ONE_TO_ONE
enum cudnnAttnQueryMap_t
|
static int |
CUDNN_BATCHNORM_OPS_BN
enum cudnnBatchNormOps_t
|
static int |
CUDNN_BATCHNORM_OPS_BN_ACTIVATION
enum cudnnBatchNormOps_t
|
static int |
CUDNN_BATCHNORM_OPS_BN_ADD_ACTIVATION
enum cudnnBatchNormOps_t
|
static int |
CUDNN_BATCHNORM_PER_ACTIVATION
enum cudnnBatchNormMode_t
|
static int |
CUDNN_BATCHNORM_SPATIAL
enum cudnnBatchNormMode_t
|
static int |
CUDNN_BATCHNORM_SPATIAL_PERSISTENT
enum cudnnBatchNormMode_t
|
static int |
CUDNN_BIDIRECTIONAL
enum cudnnDirectionMode_t
|
static double |
CUDNN_BN_MIN_EPSILON |
static int |
CUDNN_CONVOLUTION
enum cudnnConvolutionMode_t
|
static int |
CUDNN_CONVOLUTION_BWD_DATA_ALGO_0
enum cudnnConvolutionBwdDataAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_DATA_ALGO_1
enum cudnnConvolutionBwdDataAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_DATA_ALGO_COUNT
enum cudnnConvolutionBwdDataAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_DATA_ALGO_FFT
enum cudnnConvolutionBwdDataAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_DATA_ALGO_FFT_TILING
enum cudnnConvolutionBwdDataAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_DATA_ALGO_WINOGRAD
enum cudnnConvolutionBwdDataAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_DATA_ALGO_WINOGRAD_NONFUSED
enum cudnnConvolutionBwdDataAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_DATA_NO_WORKSPACE
enum cudnnConvolutionBwdDataPreference_t
|
static int |
CUDNN_CONVOLUTION_BWD_DATA_PREFER_FASTEST
enum cudnnConvolutionBwdDataPreference_t
|
static int |
CUDNN_CONVOLUTION_BWD_DATA_SPECIFY_WORKSPACE_LIMIT
enum cudnnConvolutionBwdDataPreference_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_ALGO_0
enum cudnnConvolutionBwdFilterAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_ALGO_1
enum cudnnConvolutionBwdFilterAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_ALGO_3
enum cudnnConvolutionBwdFilterAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_ALGO_COUNT
enum cudnnConvolutionBwdFilterAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_ALGO_FFT
enum cudnnConvolutionBwdFilterAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_ALGO_FFT_TILING
enum cudnnConvolutionBwdFilterAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_ALGO_WINOGRAD
enum cudnnConvolutionBwdFilterAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_ALGO_WINOGRAD_NONFUSED
enum cudnnConvolutionBwdFilterAlgo_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_NO_WORKSPACE
enum cudnnConvolutionBwdFilterPreference_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_PREFER_FASTEST
enum cudnnConvolutionBwdFilterPreference_t
|
static int |
CUDNN_CONVOLUTION_BWD_FILTER_SPECIFY_WORKSPACE_LIMIT
enum cudnnConvolutionBwdFilterPreference_t
|
static int |
CUDNN_CONVOLUTION_FWD_ALGO_COUNT
enum cudnnConvolutionFwdAlgo_t
|
static int |
CUDNN_CONVOLUTION_FWD_ALGO_DIRECT
enum cudnnConvolutionFwdAlgo_t
|
static int |
CUDNN_CONVOLUTION_FWD_ALGO_FFT
enum cudnnConvolutionFwdAlgo_t
|
static int |
CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING
enum cudnnConvolutionFwdAlgo_t
|
static int |
CUDNN_CONVOLUTION_FWD_ALGO_GEMM
enum cudnnConvolutionFwdAlgo_t
|
static int |
CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_GEMM
enum cudnnConvolutionFwdAlgo_t
|
static int |
CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM
enum cudnnConvolutionFwdAlgo_t
|
static int |
CUDNN_CONVOLUTION_FWD_ALGO_WINOGRAD
enum cudnnConvolutionFwdAlgo_t
|
static int |
CUDNN_CONVOLUTION_FWD_ALGO_WINOGRAD_NONFUSED
enum cudnnConvolutionFwdAlgo_t
|
static int |
CUDNN_CONVOLUTION_FWD_NO_WORKSPACE
enum cudnnConvolutionFwdPreference_t
|
static int |
CUDNN_CONVOLUTION_FWD_PREFER_FASTEST
enum cudnnConvolutionFwdPreference_t
|
static int |
CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT
enum cudnnConvolutionFwdPreference_t
|
static int |
CUDNN_CROSS_CORRELATION
enum cudnnConvolutionMode_t
|
static int |
CUDNN_CTC_LOSS_ALGO_DETERMINISTIC
enum cudnnCTCLossAlgo_t
|
static int |
CUDNN_CTC_LOSS_ALGO_NON_DETERMINISTIC
enum cudnnCTCLossAlgo_t
|
static int |
CUDNN_DATA_DOUBLE
enum cudnnDataType_t
|
static int |
CUDNN_DATA_FLOAT
enum cudnnDataType_t
|
static int |
CUDNN_DATA_HALF
enum cudnnDataType_t
|
static int |
CUDNN_DATA_INT32
enum cudnnDataType_t
|
static int |
CUDNN_DATA_INT8
enum cudnnDataType_t
|
static int |
CUDNN_DATA_INT8x32
enum cudnnDataType_t
|
static int |
CUDNN_DATA_INT8x4
enum cudnnDataType_t
|
static int |
CUDNN_DATA_UINT8
enum cudnnDataType_t
|
static int |
CUDNN_DATA_UINT8x4
enum cudnnDataType_t
|
static int |
CUDNN_DEFAULT_MATH
enum cudnnMathType_t
|
static int |
CUDNN_DETERMINISTIC
enum cudnnDeterminism_t
|
static int |
CUDNN_DIM_MAX |
static int |
CUDNN_DIVNORM_PRECOMPUTED_MEANS
enum cudnnDivNormMode_t
|
static int |
CUDNN_ERRQUERY_BLOCKING
enum cudnnErrQueryMode_t
|
static int |
CUDNN_ERRQUERY_NONBLOCKING
enum cudnnErrQueryMode_t
|
static int |
CUDNN_ERRQUERY_RAWCODE
enum cudnnErrQueryMode_t
|
static int |
CUDNN_GRU
enum cudnnRNNMode_t
|
static int |
CUDNN_LINEAR_INPUT
enum cudnnRNNInputMode_t
|
static int |
CUDNN_LRN_CROSS_CHANNEL_DIM1
enum cudnnLRNMode_t
|
static int |
CUDNN_LRN_MAX_N |
static double |
CUDNN_LRN_MIN_BETA |
static double |
CUDNN_LRN_MIN_K |
static int |
CUDNN_LRN_MIN_N |
static int |
CUDNN_LSTM
enum cudnnRNNMode_t
|
static int |
CUDNN_MAJOR |
static int |
CUDNN_MH_ATTN_K_WEIGHTS
enum cudnnMultiHeadAttnWeightKind_t
|
static int |
CUDNN_MH_ATTN_O_WEIGHTS
enum cudnnMultiHeadAttnWeightKind_t
|
static int |
CUDNN_MH_ATTN_Q_WEIGHTS
enum cudnnMultiHeadAttnWeightKind_t
|
static int |
CUDNN_MH_ATTN_V_WEIGHTS
enum cudnnMultiHeadAttnWeightKind_t
|
static int |
CUDNN_MINOR |
static int |
CUDNN_NON_DETERMINISTIC
enum cudnnDeterminism_t
|
static int |
CUDNN_NOT_PROPAGATE_NAN
enum cudnnNanPropagation_t
|
static int |
CUDNN_OP_TENSOR_ADD
enum cudnnOpTensorOp_t
|
static int |
CUDNN_OP_TENSOR_MAX
enum cudnnOpTensorOp_t
|
static int |
CUDNN_OP_TENSOR_MIN
enum cudnnOpTensorOp_t
|
static int |
CUDNN_OP_TENSOR_MUL
enum cudnnOpTensorOp_t
|
static int |
CUDNN_OP_TENSOR_NOT
enum cudnnOpTensorOp_t
|
static int |
CUDNN_OP_TENSOR_SQRT
enum cudnnOpTensorOp_t
|
static int |
CUDNN_PATCHLEVEL |
static int |
CUDNN_POOLING_AVERAGE_COUNT_EXCLUDE_PADDING
enum cudnnPoolingMode_t
|
static int |
CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING
enum cudnnPoolingMode_t
|
static int |
CUDNN_POOLING_MAX
enum cudnnPoolingMode_t
|
static int |
CUDNN_POOLING_MAX_DETERMINISTIC
enum cudnnPoolingMode_t
|
static int |
CUDNN_PROPAGATE_NAN
enum cudnnNanPropagation_t
|
static int |
CUDNN_REDUCE_TENSOR_ADD
enum cudnnReduceTensorOp_t
|
static int |
CUDNN_REDUCE_TENSOR_AMAX
enum cudnnReduceTensorOp_t
|
static int |
CUDNN_REDUCE_TENSOR_AVG
enum cudnnReduceTensorOp_t
|
static int |
CUDNN_REDUCE_TENSOR_FLATTENED_INDICES
enum cudnnReduceTensorIndices_t
|
static int |
CUDNN_REDUCE_TENSOR_MAX
enum cudnnReduceTensorOp_t
|
static int |
CUDNN_REDUCE_TENSOR_MIN
enum cudnnReduceTensorOp_t
|
static int |
CUDNN_REDUCE_TENSOR_MUL
enum cudnnReduceTensorOp_t
|
static int |
CUDNN_REDUCE_TENSOR_MUL_NO_ZEROS
enum cudnnReduceTensorOp_t
|
static int |
CUDNN_REDUCE_TENSOR_NO_INDICES
enum cudnnReduceTensorIndices_t
|
static int |
CUDNN_REDUCE_TENSOR_NORM1
enum cudnnReduceTensorOp_t
|
static int |
CUDNN_REDUCE_TENSOR_NORM2
enum cudnnReduceTensorOp_t
|
static int |
CUDNN_RNN_ALGO_COUNT
enum cudnnRNNAlgo_t
|
static int |
CUDNN_RNN_ALGO_PERSIST_DYNAMIC
enum cudnnRNNAlgo_t
|
static int |
CUDNN_RNN_ALGO_PERSIST_STATIC
enum cudnnRNNAlgo_t
|
static int |
CUDNN_RNN_ALGO_STANDARD
enum cudnnRNNAlgo_t
|
static int |
CUDNN_RNN_CLIP_MINMAX
enum cudnnRNNClipMode_t
|
static int |
CUDNN_RNN_CLIP_NONE
enum cudnnRNNClipMode_t
|
static int |
CUDNN_RNN_DATA_LAYOUT_BATCH_MAJOR_UNPACKED
enum cudnnRNNDataLayout_t
|
static int |
CUDNN_RNN_DATA_LAYOUT_SEQ_MAJOR_PACKED
enum cudnnRNNDataLayout_t
|
static int |
CUDNN_RNN_DATA_LAYOUT_SEQ_MAJOR_UNPACKED
enum cudnnRNNDataLayout_t
|
static int |
CUDNN_RNN_DOUBLE_BIAS
enum cudnnRNNBiasMode_t
|
static int |
CUDNN_RNN_NO_BIAS
enum cudnnRNNBiasMode_t
|
static int |
CUDNN_RNN_PADDED_IO_DISABLED
enum cudnnRNNPaddingMode_t
|
static int |
CUDNN_RNN_PADDED_IO_ENABLED
enum cudnnRNNPaddingMode_t
|
static int |
CUDNN_RNN_RELU
enum cudnnRNNMode_t
|
static int |
CUDNN_RNN_SINGLE_INP_BIAS
enum cudnnRNNBiasMode_t
|
static int |
CUDNN_RNN_SINGLE_REC_BIAS
enum cudnnRNNBiasMode_t
|
static int |
CUDNN_RNN_TANH
enum cudnnRNNMode_t
|
static int |
CUDNN_SAMPLER_BILINEAR
enum cudnnSamplerType_t
|
static int |
CUDNN_SEQDATA_BATCH_DIM
enum cudnnSeqDataAxis_t
|
static int |
CUDNN_SEQDATA_BEAM_DIM
enum cudnnSeqDataAxis_t
|
static int |
CUDNN_SEQDATA_DIM_COUNT |
static int |
CUDNN_SEQDATA_TIME_DIM
enum cudnnSeqDataAxis_t
|
static int |
CUDNN_SEQDATA_VECT_DIM
enum cudnnSeqDataAxis_t
|
static int |
CUDNN_SEV_ERROR
enum cudnnSeverity_t
|
static long |
CUDNN_SEV_ERROR_EN |
static int |
CUDNN_SEV_FATAL
enum cudnnSeverity_t
|
static int |
CUDNN_SEV_INFO
enum cudnnSeverity_t
|
static long |
CUDNN_SEV_INFO_EN |
static int |
CUDNN_SEV_WARNING
enum cudnnSeverity_t
|
static long |
CUDNN_SEV_WARNING_EN |
static int |
CUDNN_SKIP_INPUT
enum cudnnRNNInputMode_t
|
static int |
CUDNN_SOFTMAX_ACCURATE
enum cudnnSoftmaxAlgorithm_t
|
static int |
CUDNN_SOFTMAX_FAST
enum cudnnSoftmaxAlgorithm_t
|
static int |
CUDNN_SOFTMAX_LOG
enum cudnnSoftmaxAlgorithm_t
|
static int |
CUDNN_SOFTMAX_MODE_CHANNEL
enum cudnnSoftmaxMode_t
|
static int |
CUDNN_SOFTMAX_MODE_INSTANCE
enum cudnnSoftmaxMode_t
|
static int |
CUDNN_STATUS_ALLOC_FAILED
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_ARCH_MISMATCH
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_BAD_PARAM
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_EXECUTION_FAILED
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_INTERNAL_ERROR
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_INVALID_VALUE
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_LICENSE_ERROR
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_MAPPING_ERROR
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_NOT_INITIALIZED
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_NOT_SUPPORTED
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_RUNTIME_FP_OVERFLOW
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_RUNTIME_IN_PROGRESS
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_RUNTIME_PREREQUISITE_MISSING
enum cudnnStatus_t
|
static int |
CUDNN_STATUS_SUCCESS
enum cudnnStatus_t
|
static int |
CUDNN_TENSOR_NCHW
enum cudnnTensorFormat_t
|
static int |
CUDNN_TENSOR_NCHW_VECT_C
enum cudnnTensorFormat_t
|
static int |
CUDNN_TENSOR_NHWC
enum cudnnTensorFormat_t
|
static int |
CUDNN_TENSOR_OP_MATH
enum cudnnMathType_t
|
static int |
CUDNN_TENSOR_OP_MATH_ALLOW_CONVERSION
enum cudnnMathType_t
|
static long |
CUDNN_TRANSFORM_FOLD
enum cudnnFoldingDirection_t
|
static long |
CUDNN_TRANSFORM_UNFOLD
enum cudnnFoldingDirection_t
|
static int |
CUDNN_UNIDIRECTIONAL
enum cudnnDirectionMode_t
|
static int |
CUDNN_VERSION |
static int |
CUDNN_WGRAD_MODE_ADD
enum cudnnWgradMode_t
|
static int |
CUDNN_WGRAD_MODE_SET
enum cudnnWgradMode_t
|
static int |
MAJOR_VERSION
enum libraryPropertyType_t
|
static int |
MINOR_VERSION
enum libraryPropertyType_t
|
static int |
PATCH_LEVEL
enum libraryPropertyType_t
|
| Constructor and Description |
|---|
cudnn() |
| Modifier and Type | Method and Description |
|---|---|
static int |
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 |
cudnnActivationForward(cudnnContext handle,
cudnnActivationStruct activationDesc,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
Pointer beta,
cudnnTensorStruct yDesc,
Pointer y) |
static int |
cudnnAddTensor(cudnnContext handle,
Pointer alpha,
cudnnTensorStruct aDesc,
Pointer A,
Pointer beta,
cudnnTensorStruct cDesc,
Pointer C) |
static int |
cudnnBatchNormalizationBackward(cudnnContext handle,
int mode,
Pointer alphaDataDiff,
Pointer betaDataDiff,
Pointer alphaParamDiff,
Pointer betaParamDiff,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnTensorStruct dxDesc,
Pointer dx,
cudnnTensorStruct dBnScaleBiasDesc,
Pointer bnScale,
Pointer dBnScaleResult,
Pointer dBnBiasResult,
double epsilon,
Pointer savedMean,
Pointer savedInvVariance) |
static int |
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 |
cudnnBatchNormalizationForwardInference(cudnnContext handle,
int mode,
Pointer alpha,
Pointer beta,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct bnScaleBiasMeanVarDesc,
Pointer bnScale,
Pointer bnBias,
Pointer estimatedMean,
Pointer estimatedVariance,
double epsilon) |
static int |
cudnnBatchNormalizationForwardTraining(cudnnContext handle,
int mode,
Pointer alpha,
Pointer beta,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct bnScaleBiasMeanVarDesc,
Pointer bnScale,
Pointer bnBias,
double exponentialAverageFactor,
Pointer resultRunningMean,
Pointer resultRunningVariance,
double epsilon,
Pointer resultSaveMean,
Pointer resultSaveInvVariance) |
static int |
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 |
cudnnConvolutionBackwardBias(cudnnContext handle,
Pointer alpha,
cudnnTensorStruct dyDesc,
Pointer dy,
Pointer beta,
cudnnTensorStruct dbDesc,
Pointer db) |
static int |
cudnnConvolutionBackwardData(cudnnContext handle,
Pointer alpha,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnConvolutionStruct convDesc,
int algo,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer beta,
cudnnTensorStruct dxDesc,
Pointer dx) |
static int |
cudnnConvolutionBackwardFilter(cudnnContext handle,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnConvolutionStruct convDesc,
int algo,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer beta,
cudnnFilterStruct dwDesc,
Pointer dw) |
static int |
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 |
cudnnConvolutionForward(cudnnContext handle,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
cudnnFilterStruct wDesc,
Pointer w,
cudnnConvolutionStruct convDesc,
int algo,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer beta,
cudnnTensorStruct yDesc,
Pointer y) |
static int |
cudnnCopyAlgorithmDescriptor(cudnnAlgorithmStruct src,
cudnnAlgorithmStruct dest) |
static int |
cudnnCreate(cudnnContext handle) |
static int |
cudnnCreateActivationDescriptor(cudnnActivationStruct activationDesc) |
static int |
cudnnCreateAlgorithmDescriptor(cudnnAlgorithmStruct algoDesc) |
static int |
cudnnCreateAlgorithmPerformance(cudnnAlgorithmPerformanceStruct algoPerf,
int numberToCreate) |
static int |
cudnnCreateAttnDescriptor(cudnnAttnStruct attnDesc) |
static int |
cudnnCreateConvolutionDescriptor(cudnnConvolutionStruct convDesc) |
static int |
cudnnCreateCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc) |
static int |
cudnnCreateDropoutDescriptor(cudnnDropoutStruct dropoutDesc) |
static int |
cudnnCreateFilterDescriptor(cudnnFilterStruct filterDesc) |
static int |
cudnnCreateLRNDescriptor(cudnnLRNStruct normDesc) |
static int |
cudnnCreateOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc) |
static int |
cudnnCreatePersistentRNNPlan(cudnnRNNStruct rnnDesc,
int minibatch,
int dataType,
cudnnPersistentRNNPlan plan) |
static int |
cudnnCreatePoolingDescriptor(cudnnPoolingStruct poolingDesc) |
static int |
cudnnCreateReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc) |
static int |
cudnnCreateRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc) |
static int |
cudnnCreateRNNDescriptor(cudnnRNNStruct rnnDesc) |
static int |
cudnnCreateSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc) |
static int |
cudnnCreateSpatialTransformerDescriptor(cudnnSpatialTransformerStruct stDesc) |
static int |
cudnnCreateTensorDescriptor(cudnnTensorStruct tensorDesc) |
static int |
cudnnCreateTensorDescriptor(PointerPointer tensorDesc) |
static int |
cudnnCreateTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc)
Create an empty tensor transform descriptor
|
static int |
cudnnCTCLoss(cudnnContext handle,
cudnnTensorStruct probsDesc,
Pointer probs,
int[] labels,
int[] labelLengths,
int[] inputLengths,
Pointer costs,
cudnnTensorStruct gradientsDesc,
Pointer gradients,
int algo,
cudnnCTCLossStruct ctcLossDesc,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnCTCLoss(cudnnContext handle,
cudnnTensorStruct probsDesc,
Pointer probs,
IntBuffer labels,
IntBuffer labelLengths,
IntBuffer inputLengths,
Pointer costs,
cudnnTensorStruct gradientsDesc,
Pointer gradients,
int algo,
cudnnCTCLossStruct ctcLossDesc,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnCTCLoss(cudnnContext handle,
cudnnTensorStruct probsDesc,
Pointer probs,
IntPointer labels,
IntPointer labelLengths,
IntPointer inputLengths,
Pointer costs,
cudnnTensorStruct gradientsDesc,
Pointer gradients,
int algo,
cudnnCTCLossStruct ctcLossDesc,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnDeriveBNTensorDescriptor(cudnnTensorStruct derivedBnDesc,
cudnnTensorStruct xDesc,
int mode) |
static int |
cudnnDestroy(cudnnContext handle) |
static int |
cudnnDestroyActivationDescriptor(cudnnActivationStruct activationDesc) |
static int |
cudnnDestroyAlgorithmDescriptor(cudnnAlgorithmStruct algoDesc) |
static int |
cudnnDestroyAlgorithmPerformance(cudnnAlgorithmPerformanceStruct algoPerf,
int numberToDestroy) |
static int |
cudnnDestroyAttnDescriptor(cudnnAttnStruct attnDesc) |
static int |
cudnnDestroyConvolutionDescriptor(cudnnConvolutionStruct convDesc) |
static int |
cudnnDestroyCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc) |
static int |
cudnnDestroyDropoutDescriptor(cudnnDropoutStruct dropoutDesc) |
static int |
cudnnDestroyFilterDescriptor(cudnnFilterStruct filterDesc) |
static int |
cudnnDestroyLRNDescriptor(cudnnLRNStruct lrnDesc) |
static int |
cudnnDestroyOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc) |
static int |
cudnnDestroyPersistentRNNPlan(cudnnPersistentRNNPlan plan) |
static int |
cudnnDestroyPoolingDescriptor(cudnnPoolingStruct poolingDesc) |
static int |
cudnnDestroyReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc) |
static int |
cudnnDestroyRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc) |
static int |
cudnnDestroyRNNDescriptor(cudnnRNNStruct rnnDesc) |
static int |
cudnnDestroySeqDataDescriptor(cudnnSeqDataStruct seqDataDesc) |
static int |
cudnnDestroySpatialTransformerDescriptor(cudnnSpatialTransformerStruct stDesc) |
static int |
cudnnDestroyTensorDescriptor(cudnnTensorStruct tensorDesc) |
static int |
cudnnDestroyTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc)
Destroys a previously created tensor transform descriptor.
|
static int |
cudnnDivisiveNormalizationBackward(cudnnContext handle,
cudnnLRNStruct normDesc,
int mode,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
Pointer means,
Pointer dy,
Pointer temp,
Pointer temp2,
Pointer beta,
cudnnTensorStruct dXdMeansDesc,
Pointer dx,
Pointer dMeans) |
static int |
cudnnDivisiveNormalizationForward(cudnnContext handle,
cudnnLRNStruct normDesc,
int mode,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
Pointer means,
Pointer temp,
Pointer temp2,
Pointer beta,
cudnnTensorStruct yDesc,
Pointer y) |
static int |
cudnnDropoutBackward(cudnnContext handle,
cudnnDropoutStruct dropoutDesc,
cudnnTensorStruct dydesc,
Pointer dy,
cudnnTensorStruct dxdesc,
Pointer dx,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnDropoutForward(cudnnContext handle,
cudnnDropoutStruct dropoutDesc,
cudnnTensorStruct xdesc,
Pointer x,
cudnnTensorStruct ydesc,
Pointer y,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnDropoutGetReserveSpaceSize(cudnnTensorStruct xdesc,
SizeTPointer sizeInBytes) |
static int |
cudnnDropoutGetStatesSize(cudnnContext handle,
SizeTPointer sizeInBytes) |
static int |
cudnnFindConvolutionBackwardDataAlgorithm(cudnnContext handle,
cudnnFilterStruct wDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct dxDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf_t perfResults) |
static int |
cudnnFindConvolutionBackwardDataAlgorithm(cudnnContext handle,
cudnnFilterStruct wDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct dxDesc,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf_t perfResults) |
static int |
cudnnFindConvolutionBackwardDataAlgorithm(cudnnContext handle,
cudnnFilterStruct wDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct dxDesc,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf_t perfResults) |
static int |
cudnnFindConvolutionBackwardDataAlgorithmEx(cudnnContext handle,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct dxDesc,
Pointer dx,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf_t perfResults,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindConvolutionBackwardDataAlgorithmEx(cudnnContext handle,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct dxDesc,
Pointer dx,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf_t perfResults,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindConvolutionBackwardDataAlgorithmEx(cudnnContext handle,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct dxDesc,
Pointer dx,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf_t perfResults,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindConvolutionBackwardFilterAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct dwDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf_t perfResults) |
static int |
cudnnFindConvolutionBackwardFilterAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct dwDesc,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf_t perfResults) |
static int |
cudnnFindConvolutionBackwardFilterAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct dwDesc,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf_t perfResults) |
static int |
cudnnFindConvolutionBackwardFilterAlgorithmEx(cudnnContext handle,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct dyDesc,
Pointer y,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct dwDesc,
Pointer dw,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf_t perfResults,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindConvolutionBackwardFilterAlgorithmEx(cudnnContext handle,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct dyDesc,
Pointer y,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct dwDesc,
Pointer dw,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf_t perfResults,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindConvolutionBackwardFilterAlgorithmEx(cudnnContext handle,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct dyDesc,
Pointer y,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct dwDesc,
Pointer dw,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf_t perfResults,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindConvolutionForwardAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct yDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf_t perfResults) |
static int |
cudnnFindConvolutionForwardAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct yDesc,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf_t perfResults) |
static int |
cudnnFindConvolutionForwardAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct yDesc,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf_t perfResults) |
static int |
cudnnFindConvolutionForwardAlgorithmEx(cudnnContext handle,
cudnnTensorStruct xDesc,
Pointer x,
cudnnFilterStruct wDesc,
Pointer w,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct yDesc,
Pointer y,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf_t perfResults,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindConvolutionForwardAlgorithmEx(cudnnContext handle,
cudnnTensorStruct xDesc,
Pointer x,
cudnnFilterStruct wDesc,
Pointer w,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct yDesc,
Pointer y,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf_t perfResults,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindConvolutionForwardAlgorithmEx(cudnnContext handle,
cudnnTensorStruct xDesc,
Pointer x,
cudnnFilterStruct wDesc,
Pointer w,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct yDesc,
Pointer y,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf_t perfResults,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnTensorStruct dhyDesc,
Pointer dhy,
cudnnTensorStruct dcyDesc,
Pointer dcy,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnTensorStruct dxDesc,
Pointer dx,
cudnnTensorStruct dhxDesc,
Pointer dhx,
cudnnTensorStruct dcxDesc,
Pointer dcx,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnTensorStruct dhyDesc,
Pointer dhy,
cudnnTensorStruct dcyDesc,
Pointer dcy,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnTensorStruct dxDesc,
Pointer dx,
cudnnTensorStruct dhxDesc,
Pointer dhx,
cudnnTensorStruct dcxDesc,
Pointer dcx,
float findIntensity,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnTensorStruct dhyDesc,
Pointer dhy,
cudnnTensorStruct dcyDesc,
Pointer dcy,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnTensorStruct dxDesc,
Pointer dx,
cudnnTensorStruct dhxDesc,
Pointer dhx,
cudnnTensorStruct dcxDesc,
Pointer dcx,
float findIntensity,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer yDesc,
Pointer y,
PointerPointer dyDesc,
Pointer dy,
cudnnTensorStruct dhyDesc,
Pointer dhy,
cudnnTensorStruct dcyDesc,
Pointer dcy,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
PointerPointer dxDesc,
Pointer dx,
cudnnTensorStruct dhxDesc,
Pointer dhx,
cudnnTensorStruct dcxDesc,
Pointer dcx,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer yDesc,
Pointer y,
PointerPointer dyDesc,
Pointer dy,
cudnnTensorStruct dhyDesc,
Pointer dhy,
cudnnTensorStruct dcyDesc,
Pointer dcy,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
PointerPointer dxDesc,
Pointer dx,
cudnnTensorStruct dhxDesc,
Pointer dhx,
cudnnTensorStruct dcxDesc,
Pointer dcx,
float findIntensity,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer yDesc,
Pointer y,
PointerPointer dyDesc,
Pointer dy,
cudnnTensorStruct dhyDesc,
Pointer dhy,
cudnnTensorStruct dcyDesc,
Pointer dcy,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
PointerPointer dxDesc,
Pointer dx,
cudnnTensorStruct dhxDesc,
Pointer dhx,
cudnnTensorStruct dcxDesc,
Pointer dcx,
float findIntensity,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct yDesc,
Pointer y,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterStruct dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct yDesc,
Pointer y,
float findIntensity,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterStruct dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct yDesc,
Pointer y,
float findIntensity,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterStruct dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
PointerPointer yDesc,
Pointer y,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterStruct dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
PointerPointer yDesc,
Pointer y,
float findIntensity,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterStruct dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
PointerPointer yDesc,
Pointer y,
float findIntensity,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterStruct dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
PointerPointer yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
PointerPointer yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
PointerPointer yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
PointerPointer yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
PointerPointer yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
PointerPointer yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
float findIntensity,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnAlgorithmPerformanceStruct perfResults,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnGetActivationDescriptor(cudnnActivationStruct activationDesc,
int[] mode,
int[] reluNanOpt,
double[] coef) |
static int |
cudnnGetActivationDescriptor(cudnnActivationStruct activationDesc,
IntBuffer mode,
IntBuffer reluNanOpt,
DoubleBuffer coef) |
static int |
cudnnGetActivationDescriptor(cudnnActivationStruct activationDesc,
IntPointer mode,
IntPointer reluNanOpt,
DoublePointer coef) |
static int |
cudnnGetAlgorithmDescriptor(cudnnAlgorithmStruct algoDesc,
cudnnAlgorithm_t algorithm) |
static int |
cudnnGetAlgorithmPerformance(cudnnAlgorithmPerformanceStruct algoPerf,
cudnnAlgorithmStruct algoDesc,
int[] status,
float[] time,
SizeTPointer memory) |
static int |
cudnnGetAlgorithmPerformance(cudnnAlgorithmPerformanceStruct algoPerf,
cudnnAlgorithmStruct algoDesc,
IntBuffer status,
FloatBuffer time,
SizeTPointer memory) |
static int |
cudnnGetAlgorithmPerformance(cudnnAlgorithmPerformanceStruct algoPerf,
cudnnAlgorithmStruct algoDesc,
IntPointer status,
FloatPointer time,
SizeTPointer memory) |
static int |
cudnnGetAlgorithmSpaceSize(cudnnContext handle,
cudnnAlgorithmStruct algoDesc,
SizeTPointer algoSpaceSizeInBytes) |
static int |
cudnnGetAttnDescriptor(cudnnAttnStruct attnDesc,
int[] queryMap,
int[] nHeads,
double[] smScaler,
int[] dataType,
int[] computePrec,
int[] mathType,
cudnnDropoutStruct attnDropoutDesc,
cudnnDropoutStruct postDropoutDesc,
int[] qSize,
int[] kSize,
int[] vSize,
int[] qProjSize,
int[] kProjSize,
int[] vProjSize,
int[] oProjSize,
int[] qoMaxSeqLength,
int[] kvMaxSeqLength,
int[] maxBatchSize,
int[] maxBeamSize) |
static int |
cudnnGetAttnDescriptor(cudnnAttnStruct attnDesc,
IntBuffer queryMap,
IntBuffer nHeads,
DoubleBuffer smScaler,
IntBuffer dataType,
IntBuffer computePrec,
IntBuffer mathType,
cudnnDropoutStruct attnDropoutDesc,
cudnnDropoutStruct postDropoutDesc,
IntBuffer qSize,
IntBuffer kSize,
IntBuffer vSize,
IntBuffer qProjSize,
IntBuffer kProjSize,
IntBuffer vProjSize,
IntBuffer oProjSize,
IntBuffer qoMaxSeqLength,
IntBuffer kvMaxSeqLength,
IntBuffer maxBatchSize,
IntBuffer maxBeamSize) |
static int |
cudnnGetAttnDescriptor(cudnnAttnStruct attnDesc,
IntPointer queryMap,
IntPointer nHeads,
DoublePointer smScaler,
IntPointer dataType,
IntPointer computePrec,
IntPointer mathType,
cudnnDropoutStruct attnDropoutDesc,
cudnnDropoutStruct postDropoutDesc,
IntPointer qSize,
IntPointer kSize,
IntPointer vSize,
IntPointer qProjSize,
IntPointer kProjSize,
IntPointer vProjSize,
IntPointer oProjSize,
IntPointer qoMaxSeqLength,
IntPointer kvMaxSeqLength,
IntPointer maxBatchSize,
IntPointer maxBeamSize) |
static int |
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 |
cudnnGetBatchNormalizationForwardTrainingExWorkspaceSize(cudnnContext handle,
int mode,
int bnOps,
cudnnTensorStruct xDesc,
cudnnTensorStruct zDesc,
cudnnTensorStruct yDesc,
cudnnTensorStruct bnScaleBiasMeanVarDesc,
cudnnActivationStruct activationDesc,
SizeTPointer sizeInBytes) |
static int |
cudnnGetBatchNormalizationTrainingExReserveSpaceSize(cudnnContext handle,
int mode,
int bnOps,
cudnnActivationStruct activationDesc,
cudnnTensorStruct xDesc,
SizeTPointer sizeInBytes) |
static int |
cudnnGetCallback(int[] mask,
Pointer udata,
cudnnCallback_t fptr) |
static int |
cudnnGetCallback(IntBuffer mask,
Pointer udata,
cudnnCallback_t fptr) |
static int |
cudnnGetCallback(IntPointer mask,
Pointer udata,
cudnnCallback_t fptr) |
static int |
cudnnGetCallback(IntPointer mask,
PointerPointer udata,
cudnnCallback_t fptr) |
static int |
cudnnGetConvolution2dDescriptor(cudnnConvolutionStruct convDesc,
int[] pad_h,
int[] pad_w,
int[] u,
int[] v,
int[] dilation_h,
int[] dilation_w,
int[] mode,
int[] computeType) |
static int |
cudnnGetConvolution2dDescriptor(cudnnConvolutionStruct convDesc,
IntBuffer pad_h,
IntBuffer pad_w,
IntBuffer u,
IntBuffer v,
IntBuffer dilation_h,
IntBuffer dilation_w,
IntBuffer mode,
IntBuffer computeType) |
static int |
cudnnGetConvolution2dDescriptor(cudnnConvolutionStruct convDesc,
IntPointer pad_h,
IntPointer pad_w,
IntPointer u,
IntPointer v,
IntPointer dilation_h,
IntPointer dilation_w,
IntPointer mode,
IntPointer computeType) |
static int |
cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionStruct convDesc,
cudnnTensorStruct inputTensorDesc,
cudnnFilterStruct filterDesc,
int[] n,
int[] c,
int[] h,
int[] w) |
static int |
cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionStruct convDesc,
cudnnTensorStruct inputTensorDesc,
cudnnFilterStruct filterDesc,
IntBuffer n,
IntBuffer c,
IntBuffer h,
IntBuffer w) |
static int |
cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionStruct convDesc,
cudnnTensorStruct inputTensorDesc,
cudnnFilterStruct filterDesc,
IntPointer n,
IntPointer c,
IntPointer h,
IntPointer w) |
static int |
cudnnGetConvolutionBackwardDataAlgorithm_v7(cudnnContext handle,
cudnnFilterStruct filterDesc,
cudnnTensorStruct diffDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct gradDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf_t perfResults) |
static int |
cudnnGetConvolutionBackwardDataAlgorithm_v7(cudnnContext handle,
cudnnFilterStruct filterDesc,
cudnnTensorStruct diffDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct gradDesc,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf_t perfResults) |
static int |
cudnnGetConvolutionBackwardDataAlgorithm_v7(cudnnContext handle,
cudnnFilterStruct filterDesc,
cudnnTensorStruct diffDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct gradDesc,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf_t perfResults) |
static int |
cudnnGetConvolutionBackwardDataAlgorithm(cudnnContext handle,
cudnnFilterStruct wDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct dxDesc,
int preference,
long memoryLimitInBytes,
int[] algo) |
static int |
cudnnGetConvolutionBackwardDataAlgorithm(cudnnContext handle,
cudnnFilterStruct wDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct dxDesc,
int preference,
long memoryLimitInBytes,
IntBuffer algo) |
static int |
cudnnGetConvolutionBackwardDataAlgorithm(cudnnContext handle,
cudnnFilterStruct wDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct dxDesc,
int preference,
long memoryLimitInBytes,
IntPointer algo) |
static int |
cudnnGetConvolutionBackwardDataAlgorithmMaxCount(cudnnContext handle,
int[] count) |
static int |
cudnnGetConvolutionBackwardDataAlgorithmMaxCount(cudnnContext handle,
IntBuffer count) |
static int |
cudnnGetConvolutionBackwardDataAlgorithmMaxCount(cudnnContext handle,
IntPointer count) |
static int |
cudnnGetConvolutionBackwardDataWorkspaceSize(cudnnContext handle,
cudnnFilterStruct wDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct dxDesc,
int algo,
SizeTPointer sizeInBytes) |
static int |
cudnnGetConvolutionBackwardFilterAlgorithm_v7(cudnnContext handle,
cudnnTensorStruct srcDesc,
cudnnTensorStruct diffDesc,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct gradDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf_t perfResults) |
static int |
cudnnGetConvolutionBackwardFilterAlgorithm_v7(cudnnContext handle,
cudnnTensorStruct srcDesc,
cudnnTensorStruct diffDesc,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct gradDesc,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf_t perfResults) |
static int |
cudnnGetConvolutionBackwardFilterAlgorithm_v7(cudnnContext handle,
cudnnTensorStruct srcDesc,
cudnnTensorStruct diffDesc,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct gradDesc,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnConvolutionBwdFilterAlgoPerf_t perfResults) |
static int |
cudnnGetConvolutionBackwardFilterAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct dwDesc,
int preference,
long memoryLimitInBytes,
int[] algo) |
static int |
cudnnGetConvolutionBackwardFilterAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct dwDesc,
int preference,
long memoryLimitInBytes,
IntBuffer algo) |
static int |
cudnnGetConvolutionBackwardFilterAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct dwDesc,
int preference,
long memoryLimitInBytes,
IntPointer algo) |
static int |
cudnnGetConvolutionBackwardFilterAlgorithmMaxCount(cudnnContext handle,
int[] count) |
static int |
cudnnGetConvolutionBackwardFilterAlgorithmMaxCount(cudnnContext handle,
IntBuffer count) |
static int |
cudnnGetConvolutionBackwardFilterAlgorithmMaxCount(cudnnContext handle,
IntPointer count) |
static int |
cudnnGetConvolutionBackwardFilterWorkspaceSize(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnTensorStruct dyDesc,
cudnnConvolutionStruct convDesc,
cudnnFilterStruct gradDesc,
int algo,
SizeTPointer sizeInBytes) |
static int |
cudnnGetConvolutionForwardAlgorithm_v7(cudnnContext handle,
cudnnTensorStruct srcDesc,
cudnnFilterStruct filterDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct destDesc,
int requestedAlgoCount,
int[] returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf_t perfResults) |
static int |
cudnnGetConvolutionForwardAlgorithm_v7(cudnnContext handle,
cudnnTensorStruct srcDesc,
cudnnFilterStruct filterDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct destDesc,
int requestedAlgoCount,
IntBuffer returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf_t perfResults) |
static int |
cudnnGetConvolutionForwardAlgorithm_v7(cudnnContext handle,
cudnnTensorStruct srcDesc,
cudnnFilterStruct filterDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct destDesc,
int requestedAlgoCount,
IntPointer returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf_t perfResults) |
static int |
cudnnGetConvolutionForwardAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct yDesc,
int preference,
long memoryLimitInBytes,
int[] algo) |
static int |
cudnnGetConvolutionForwardAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct yDesc,
int preference,
long memoryLimitInBytes,
IntBuffer algo) |
static int |
cudnnGetConvolutionForwardAlgorithm(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct yDesc,
int preference,
long memoryLimitInBytes,
IntPointer algo) |
static int |
cudnnGetConvolutionForwardAlgorithmMaxCount(cudnnContext handle,
int[] count) |
static int |
cudnnGetConvolutionForwardAlgorithmMaxCount(cudnnContext handle,
IntBuffer count) |
static int |
cudnnGetConvolutionForwardAlgorithmMaxCount(cudnnContext handle,
IntPointer count) |
static int |
cudnnGetConvolutionForwardWorkspaceSize(cudnnContext handle,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct yDesc,
int algo,
SizeTPointer sizeInBytes) |
static int |
cudnnGetConvolutionGroupCount(cudnnConvolutionStruct convDesc,
int[] groupCount) |
static int |
cudnnGetConvolutionGroupCount(cudnnConvolutionStruct convDesc,
IntBuffer groupCount) |
static int |
cudnnGetConvolutionGroupCount(cudnnConvolutionStruct convDesc,
IntPointer groupCount) |
static int |
cudnnGetConvolutionMathType(cudnnConvolutionStruct convDesc,
int[] mathType) |
static int |
cudnnGetConvolutionMathType(cudnnConvolutionStruct convDesc,
IntBuffer mathType) |
static int |
cudnnGetConvolutionMathType(cudnnConvolutionStruct convDesc,
IntPointer mathType) |
static int |
cudnnGetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc,
int arrayLengthRequested,
int[] arrayLength,
int[] padA,
int[] strideA,
int[] dilationA,
int[] mode,
int[] computeType) |
static int |
cudnnGetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc,
int arrayLengthRequested,
IntBuffer arrayLength,
IntBuffer padA,
IntBuffer strideA,
IntBuffer dilationA,
IntBuffer mode,
IntBuffer computeType) |
static int |
cudnnGetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc,
int arrayLengthRequested,
IntPointer arrayLength,
IntPointer padA,
IntPointer strideA,
IntPointer dilationA,
IntPointer mode,
IntPointer computeType) |
static int |
cudnnGetConvolutionNdForwardOutputDim(cudnnConvolutionStruct convDesc,
cudnnTensorStruct inputTensorDesc,
cudnnFilterStruct filterDesc,
int nbDims,
int[] tensorOuputDimA) |
static int |
cudnnGetConvolutionNdForwardOutputDim(cudnnConvolutionStruct convDesc,
cudnnTensorStruct inputTensorDesc,
cudnnFilterStruct filterDesc,
int nbDims,
IntBuffer tensorOuputDimA) |
static int |
cudnnGetConvolutionNdForwardOutputDim(cudnnConvolutionStruct convDesc,
cudnnTensorStruct inputTensorDesc,
cudnnFilterStruct filterDesc,
int nbDims,
IntPointer tensorOuputDimA) |
static int |
cudnnGetCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc,
int[] compType) |
static int |
cudnnGetCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc,
IntBuffer compType) |
static int |
cudnnGetCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc,
IntPointer compType) |
static int |
cudnnGetCTCLossWorkspaceSize(cudnnContext handle,
cudnnTensorStruct probsDesc,
cudnnTensorStruct gradientsDesc,
int[] labels,
int[] labelLengths,
int[] inputLengths,
int algo,
cudnnCTCLossStruct ctcLossDesc,
SizeTPointer sizeInBytes) |
static int |
cudnnGetCTCLossWorkspaceSize(cudnnContext handle,
cudnnTensorStruct probsDesc,
cudnnTensorStruct gradientsDesc,
IntBuffer labels,
IntBuffer labelLengths,
IntBuffer inputLengths,
int algo,
cudnnCTCLossStruct ctcLossDesc,
SizeTPointer sizeInBytes) |
static int |
cudnnGetCTCLossWorkspaceSize(cudnnContext handle,
cudnnTensorStruct probsDesc,
cudnnTensorStruct gradientsDesc,
IntPointer labels,
IntPointer labelLengths,
IntPointer inputLengths,
int algo,
cudnnCTCLossStruct ctcLossDesc,
SizeTPointer sizeInBytes) |
static long |
cudnnGetCudartVersion() |
static int |
cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
float[] dropout,
Pointer states,
long[] seed) |
static int |
cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
FloatBuffer dropout,
Pointer states,
LongBuffer seed) |
static int |
cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
FloatPointer dropout,
Pointer states,
LongPointer seed) |
static int |
cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
FloatPointer dropout,
PointerPointer states,
LongPointer seed) |
static BytePointer |
cudnnGetErrorString(int status) |
static int |
cudnnGetFilter4dDescriptor(cudnnFilterStruct filterDesc,
int[] dataType,
int[] format,
int[] k,
int[] c,
int[] h,
int[] w) |
static int |
cudnnGetFilter4dDescriptor(cudnnFilterStruct filterDesc,
IntBuffer dataType,
IntBuffer format,
IntBuffer k,
IntBuffer c,
IntBuffer h,
IntBuffer w) |
static int |
cudnnGetFilter4dDescriptor(cudnnFilterStruct filterDesc,
IntPointer dataType,
IntPointer format,
IntPointer k,
IntPointer c,
IntPointer h,
IntPointer w) |
static int |
cudnnGetFilterNdDescriptor(cudnnFilterStruct filterDesc,
int nbDimsRequested,
int[] dataType,
int[] format,
int[] nbDims,
int[] filterDimA) |
static int |
cudnnGetFilterNdDescriptor(cudnnFilterStruct filterDesc,
int nbDimsRequested,
IntBuffer dataType,
IntBuffer format,
IntBuffer nbDims,
IntBuffer filterDimA) |
static int |
cudnnGetFilterNdDescriptor(cudnnFilterStruct filterDesc,
int nbDimsRequested,
IntPointer dataType,
IntPointer format,
IntPointer nbDims,
IntPointer filterDimA) |
static int |
cudnnGetLRNDescriptor(cudnnLRNStruct normDesc,
int[] lrnN,
double[] lrnAlpha,
double[] lrnBeta,
double[] lrnK) |
static int |
cudnnGetLRNDescriptor(cudnnLRNStruct normDesc,
IntBuffer lrnN,
DoubleBuffer lrnAlpha,
DoubleBuffer lrnBeta,
DoubleBuffer lrnK) |
static int |
cudnnGetLRNDescriptor(cudnnLRNStruct normDesc,
IntPointer lrnN,
DoublePointer lrnAlpha,
DoublePointer lrnBeta,
DoublePointer lrnK) |
static int |
cudnnGetMultiHeadAttnBuffers(cudnnContext handle,
cudnnAttnStruct attnDesc,
SizeTPointer weightSizeInBytes,
SizeTPointer workSpaceSizeInBytes,
SizeTPointer reserveSpaceSizeInBytes) |
static int |
cudnnGetMultiHeadAttnWeights(cudnnContext handle,
cudnnAttnStruct attnDesc,
int wKind,
long weightSizeInBytes,
Pointer w,
cudnnTensorStruct wDesc,
Pointer wAddr) |
static int |
cudnnGetMultiHeadAttnWeights(cudnnContext handle,
cudnnAttnStruct attnDesc,
int wKind,
long weightSizeInBytes,
Pointer w,
cudnnTensorStruct wDesc,
PointerPointer wAddr) |
static int |
cudnnGetOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc,
int[] opTensorOp,
int[] opTensorCompType,
int[] opTensorNanOpt) |
static int |
cudnnGetOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc,
IntBuffer opTensorOp,
IntBuffer opTensorCompType,
IntBuffer opTensorNanOpt) |
static int |
cudnnGetOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc,
IntPointer opTensorOp,
IntPointer opTensorCompType,
IntPointer opTensorNanOpt) |
static int |
cudnnGetPooling2dDescriptor(cudnnPoolingStruct poolingDesc,
int[] mode,
int[] maxpoolingNanOpt,
int[] windowHeight,
int[] windowWidth,
int[] verticalPadding,
int[] horizontalPadding,
int[] verticalStride,
int[] horizontalStride) |
static int |
cudnnGetPooling2dDescriptor(cudnnPoolingStruct poolingDesc,
IntBuffer mode,
IntBuffer maxpoolingNanOpt,
IntBuffer windowHeight,
IntBuffer windowWidth,
IntBuffer verticalPadding,
IntBuffer horizontalPadding,
IntBuffer verticalStride,
IntBuffer horizontalStride) |
static int |
cudnnGetPooling2dDescriptor(cudnnPoolingStruct poolingDesc,
IntPointer mode,
IntPointer maxpoolingNanOpt,
IntPointer windowHeight,
IntPointer windowWidth,
IntPointer verticalPadding,
IntPointer horizontalPadding,
IntPointer verticalStride,
IntPointer horizontalStride) |
static int |
cudnnGetPooling2dForwardOutputDim(cudnnPoolingStruct poolingDesc,
cudnnTensorStruct inputTensorDesc,
int[] n,
int[] c,
int[] h,
int[] w) |
static int |
cudnnGetPooling2dForwardOutputDim(cudnnPoolingStruct poolingDesc,
cudnnTensorStruct inputTensorDesc,
IntBuffer n,
IntBuffer c,
IntBuffer h,
IntBuffer w) |
static int |
cudnnGetPooling2dForwardOutputDim(cudnnPoolingStruct poolingDesc,
cudnnTensorStruct inputTensorDesc,
IntPointer n,
IntPointer c,
IntPointer h,
IntPointer w) |
static int |
cudnnGetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc,
int nbDimsRequested,
int[] mode,
int[] maxpoolingNanOpt,
int[] nbDims,
int[] windowDimA,
int[] paddingA,
int[] strideA) |
static int |
cudnnGetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc,
int nbDimsRequested,
IntBuffer mode,
IntBuffer maxpoolingNanOpt,
IntBuffer nbDims,
IntBuffer windowDimA,
IntBuffer paddingA,
IntBuffer strideA) |
static int |
cudnnGetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc,
int nbDimsRequested,
IntPointer mode,
IntPointer maxpoolingNanOpt,
IntPointer nbDims,
IntPointer windowDimA,
IntPointer paddingA,
IntPointer strideA) |
static int |
cudnnGetPoolingNdForwardOutputDim(cudnnPoolingStruct poolingDesc,
cudnnTensorStruct inputTensorDesc,
int nbDims,
int[] outputTensorDimA) |
static int |
cudnnGetPoolingNdForwardOutputDim(cudnnPoolingStruct poolingDesc,
cudnnTensorStruct inputTensorDesc,
int nbDims,
IntBuffer outputTensorDimA) |
static int |
cudnnGetPoolingNdForwardOutputDim(cudnnPoolingStruct poolingDesc,
cudnnTensorStruct inputTensorDesc,
int nbDims,
IntPointer outputTensorDimA) |
static int |
cudnnGetProperty(int type,
int[] value) |
static int |
cudnnGetProperty(int type,
IntBuffer value) |
static int |
cudnnGetProperty(int type,
IntPointer value) |
static int |
cudnnGetReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc,
int[] reduceTensorOp,
int[] reduceTensorCompType,
int[] reduceTensorNanOpt,
int[] reduceTensorIndices,
int[] reduceTensorIndicesType) |
static int |
cudnnGetReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc,
IntBuffer reduceTensorOp,
IntBuffer reduceTensorCompType,
IntBuffer reduceTensorNanOpt,
IntBuffer reduceTensorIndices,
IntBuffer reduceTensorIndicesType) |
static int |
cudnnGetReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc,
IntPointer reduceTensorOp,
IntPointer reduceTensorCompType,
IntPointer reduceTensorNanOpt,
IntPointer reduceTensorIndices,
IntPointer reduceTensorIndicesType) |
static int |
cudnnGetReductionIndicesSize(cudnnContext handle,
cudnnReduceTensorStruct reduceTensorDesc,
cudnnTensorStruct aDesc,
cudnnTensorStruct cDesc,
SizeTPointer sizeInBytes) |
static int |
cudnnGetReductionWorkspaceSize(cudnnContext handle,
cudnnReduceTensorStruct reduceTensorDesc,
cudnnTensorStruct aDesc,
cudnnTensorStruct cDesc,
SizeTPointer sizeInBytes) |
static int |
cudnnGetRNNBackwardDataAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] count) |
static int |
cudnnGetRNNBackwardDataAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer count) |
static int |
cudnnGetRNNBackwardDataAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer count) |
static int |
cudnnGetRNNBackwardWeightsAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] count) |
static int |
cudnnGetRNNBackwardWeightsAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer count) |
static int |
cudnnGetRNNBackwardWeightsAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer count) |
static int |
cudnnGetRNNBiasMode(cudnnRNNStruct rnnDesc,
int[] biasMode) |
static int |
cudnnGetRNNBiasMode(cudnnRNNStruct rnnDesc,
IntBuffer biasMode) |
static int |
cudnnGetRNNBiasMode(cudnnRNNStruct rnnDesc,
IntPointer biasMode) |
static int |
cudnnGetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
int[] dataType,
int[] layout,
int[] maxSeqLength,
int[] batchSize,
int[] vectorSize,
int arrayLengthRequested,
int[] seqLengthArray,
Pointer paddingFill) |
static int |
cudnnGetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
IntBuffer dataType,
IntBuffer layout,
IntBuffer maxSeqLength,
IntBuffer batchSize,
IntBuffer vectorSize,
int arrayLengthRequested,
IntBuffer seqLengthArray,
Pointer paddingFill) |
static int |
cudnnGetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
IntPointer dataType,
IntPointer layout,
IntPointer maxSeqLength,
IntPointer batchSize,
IntPointer vectorSize,
int arrayLengthRequested,
IntPointer seqLengthArray,
Pointer paddingFill) |
static int |
cudnnGetRNNDescriptor(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] hiddenSize,
int[] numLayers,
cudnnDropoutStruct dropoutDesc,
int[] inputMode,
int[] direction,
int[] mode,
int[] algo,
int[] mathPrec) |
static int |
cudnnGetRNNDescriptor(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer hiddenSize,
IntBuffer numLayers,
cudnnDropoutStruct dropoutDesc,
IntBuffer inputMode,
IntBuffer direction,
IntBuffer mode,
IntBuffer algo,
IntBuffer mathPrec) |
static int |
cudnnGetRNNDescriptor(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer hiddenSize,
IntPointer numLayers,
cudnnDropoutStruct dropoutDesc,
IntPointer inputMode,
IntPointer direction,
IntPointer mode,
IntPointer algo,
IntPointer mathPrec) |
static int |
cudnnGetRNNForwardInferenceAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] count) |
static int |
cudnnGetRNNForwardInferenceAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer count) |
static int |
cudnnGetRNNForwardInferenceAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer count) |
static int |
cudnnGetRNNForwardTrainingAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] count) |
static int |
cudnnGetRNNForwardTrainingAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer count) |
static int |
cudnnGetRNNForwardTrainingAlgorithmMaxCount(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer count) |
static int |
cudnnGetRNNLinLayerBiasParams(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int pseudoLayer,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
Pointer w,
int linLayerID,
cudnnFilterStruct linLayerBiasDesc,
Pointer linLayerBias) |
static int |
cudnnGetRNNLinLayerBiasParams(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int pseudoLayer,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
Pointer w,
int linLayerID,
cudnnFilterStruct linLayerBiasDesc,
PointerPointer linLayerBias) |
static int |
cudnnGetRNNLinLayerMatrixParams(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int pseudoLayer,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
Pointer w,
int linLayerID,
cudnnFilterStruct linLayerMatDesc,
Pointer linLayerMat) |
static int |
cudnnGetRNNLinLayerMatrixParams(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int pseudoLayer,
cudnnTensorStruct xDesc,
cudnnFilterStruct wDesc,
Pointer w,
int linLayerID,
cudnnFilterStruct linLayerMatDesc,
PointerPointer linLayerMat) |
static int |
cudnnGetRNNMatrixMathType(cudnnRNNStruct rnnDesc,
int[] mType) |
static int |
cudnnGetRNNMatrixMathType(cudnnRNNStruct rnnDesc,
IntBuffer mType) |
static int |
cudnnGetRNNMatrixMathType(cudnnRNNStruct rnnDesc,
IntPointer mType) |
static int |
cudnnGetRNNPaddingMode(cudnnRNNStruct rnnDesc,
int[] paddingMode) |
static int |
cudnnGetRNNPaddingMode(cudnnRNNStruct rnnDesc,
IntBuffer paddingMode) |
static int |
cudnnGetRNNPaddingMode(cudnnRNNStruct rnnDesc,
IntPointer paddingMode) |
static int |
cudnnGetRNNParamsSize(cudnnContext handle,
cudnnRNNStruct rnnDesc,
cudnnTensorStruct xDesc,
SizeTPointer sizeInBytes,
int dataType) |
static int |
cudnnGetRNNProjectionLayers(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] recProjSize,
int[] outProjSize) |
static int |
cudnnGetRNNProjectionLayers(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer recProjSize,
IntBuffer outProjSize) |
static int |
cudnnGetRNNProjectionLayers(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer recProjSize,
IntPointer outProjSize) |
static int |
cudnnGetRNNTrainingReserveSize(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
SizeTPointer sizeInBytes) |
static int |
cudnnGetRNNTrainingReserveSize(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
SizeTPointer sizeInBytes) |
static int |
cudnnGetRNNWorkspaceSize(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
SizeTPointer sizeInBytes) |
static int |
cudnnGetRNNWorkspaceSize(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
SizeTPointer sizeInBytes) |
static int |
cudnnGetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
int[] dataType,
int[] nbDims,
int nbDimsRequested,
int[] dimA,
int[] axes,
SizeTPointer seqLengthArraySize,
long seqLengthSizeRequsted,
int[] seqLengthArray,
Pointer paddingFill) |
static int |
cudnnGetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
IntBuffer dataType,
IntBuffer nbDims,
int nbDimsRequested,
IntBuffer dimA,
IntBuffer axes,
SizeTPointer seqLengthArraySize,
long seqLengthSizeRequsted,
IntBuffer seqLengthArray,
Pointer paddingFill) |
static int |
cudnnGetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
IntPointer dataType,
IntPointer nbDims,
int nbDimsRequested,
IntPointer dimA,
IntPointer axes,
SizeTPointer seqLengthArraySize,
long seqLengthSizeRequsted,
IntPointer seqLengthArray,
Pointer paddingFill) |
static int |
cudnnGetStream(cudnnContext handle,
CUstream_st streamId) |
static int |
cudnnGetTensor4dDescriptor(cudnnTensorStruct tensorDesc,
int[] dataType,
int[] n,
int[] c,
int[] h,
int[] w,
int[] nStride,
int[] cStride,
int[] hStride,
int[] wStride) |
static int |
cudnnGetTensor4dDescriptor(cudnnTensorStruct tensorDesc,
IntBuffer dataType,
IntBuffer n,
IntBuffer c,
IntBuffer h,
IntBuffer w,
IntBuffer nStride,
IntBuffer cStride,
IntBuffer hStride,
IntBuffer wStride) |
static int |
cudnnGetTensor4dDescriptor(cudnnTensorStruct tensorDesc,
IntPointer dataType,
IntPointer n,
IntPointer c,
IntPointer h,
IntPointer w,
IntPointer nStride,
IntPointer cStride,
IntPointer hStride,
IntPointer wStride) |
static int |
cudnnGetTensorNdDescriptor(cudnnTensorStruct tensorDesc,
int nbDimsRequested,
int[] dataType,
int[] nbDims,
int[] dimA,
int[] strideA) |
static int |
cudnnGetTensorNdDescriptor(cudnnTensorStruct tensorDesc,
int nbDimsRequested,
IntBuffer dataType,
IntBuffer nbDims,
IntBuffer dimA,
IntBuffer strideA) |
static int |
cudnnGetTensorNdDescriptor(cudnnTensorStruct tensorDesc,
int nbDimsRequested,
IntPointer dataType,
IntPointer nbDims,
IntPointer dimA,
IntPointer strideA) |
static int |
cudnnGetTensorSizeInBytes(cudnnTensorStruct tensorDesc,
SizeTPointer size) |
static int |
cudnnGetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDimsRequested,
int[] destFormat,
int[] padBeforeA,
int[] padAfterA,
int[] foldA,
long[] direction) |
static int |
cudnnGetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDimsRequested,
IntBuffer destFormat,
IntBuffer padBeforeA,
IntBuffer padAfterA,
IntBuffer foldA,
LongBuffer direction) |
static int |
cudnnGetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDimsRequested,
IntPointer destFormat,
IntPointer padBeforeA,
IntPointer padAfterA,
IntPointer foldA,
LongPointer direction)
Retrieves the values stored in a previously initialized tensor transform
descriptor.
|
static long |
cudnnGetVersion() |
static int |
cudnnIm2Col(cudnnContext handle,
cudnnTensorStruct xDesc,
Pointer x,
cudnnFilterStruct wDesc,
cudnnConvolutionStruct convDesc,
Pointer colBuffer) |
static int |
cudnnInitTransformDest(cudnnTensorTransformStruct transformDesc,
cudnnTensorStruct srcDesc,
cudnnTensorStruct destDesc,
SizeTPointer destSizeInBytes)
Create a destination descriptor for cudnnTransformTensor
|
static int |
cudnnLRNCrossChannelBackward(cudnnContext handle,
cudnnLRNStruct normDesc,
int lrnMode,
Pointer alpha,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnTensorStruct xDesc,
Pointer x,
Pointer beta,
cudnnTensorStruct dxDesc,
Pointer dx) |
static int |
cudnnLRNCrossChannelForward(cudnnContext handle,
cudnnLRNStruct normDesc,
int lrnMode,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
Pointer beta,
cudnnTensorStruct yDesc,
Pointer y) |
static int |
cudnnMultiHeadAttnBackwardData(cudnnContext handle,
cudnnAttnStruct attnDesc,
int[] loWinIdx,
int[] hiWinIdx,
int[] seqLengthArrayDQDO,
int[] seqLengthArrayDKDV,
cudnnSeqDataStruct doDesc,
Pointer dout,
cudnnSeqDataStruct dqDesc,
Pointer dqueries,
Pointer queries,
cudnnSeqDataStruct dkDesc,
Pointer dkeys,
Pointer keys,
cudnnSeqDataStruct dvDesc,
Pointer dvalues,
Pointer values,
long weightSizeInBytes,
Pointer w,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnnMultiHeadAttnBackwardData(cudnnContext handle,
cudnnAttnStruct attnDesc,
IntBuffer loWinIdx,
IntBuffer hiWinIdx,
IntBuffer seqLengthArrayDQDO,
IntBuffer seqLengthArrayDKDV,
cudnnSeqDataStruct doDesc,
Pointer dout,
cudnnSeqDataStruct dqDesc,
Pointer dqueries,
Pointer queries,
cudnnSeqDataStruct dkDesc,
Pointer dkeys,
Pointer keys,
cudnnSeqDataStruct dvDesc,
Pointer dvalues,
Pointer values,
long weightSizeInBytes,
Pointer w,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnnMultiHeadAttnBackwardData(cudnnContext handle,
cudnnAttnStruct attnDesc,
IntPointer loWinIdx,
IntPointer hiWinIdx,
IntPointer seqLengthArrayDQDO,
IntPointer seqLengthArrayDKDV,
cudnnSeqDataStruct doDesc,
Pointer dout,
cudnnSeqDataStruct dqDesc,
Pointer dqueries,
Pointer queries,
cudnnSeqDataStruct dkDesc,
Pointer dkeys,
Pointer keys,
cudnnSeqDataStruct dvDesc,
Pointer dvalues,
Pointer values,
long weightSizeInBytes,
Pointer w,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnnMultiHeadAttnBackwardWeights(cudnnContext handle,
cudnnAttnStruct attnDesc,
int addGrad,
cudnnSeqDataStruct qDesc,
Pointer queries,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct doDesc,
Pointer dout,
long weightSizeInBytes,
Pointer w,
Pointer dw,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnnMultiHeadAttnForward(cudnnContext handle,
cudnnAttnStruct attnDesc,
int currIdx,
int[] loWinIdx,
int[] hiWinIdx,
int[] seqLengthArrayQRO,
int[] seqLengthArrayKV,
cudnnSeqDataStruct qDesc,
Pointer queries,
Pointer residuals,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct oDesc,
Pointer out,
long weightSizeInBytes,
Pointer w,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnnMultiHeadAttnForward(cudnnContext handle,
cudnnAttnStruct attnDesc,
int currIdx,
IntBuffer loWinIdx,
IntBuffer hiWinIdx,
IntBuffer seqLengthArrayQRO,
IntBuffer seqLengthArrayKV,
cudnnSeqDataStruct qDesc,
Pointer queries,
Pointer residuals,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct oDesc,
Pointer out,
long weightSizeInBytes,
Pointer w,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnnMultiHeadAttnForward(cudnnContext handle,
cudnnAttnStruct attnDesc,
int currIdx,
IntPointer loWinIdx,
IntPointer hiWinIdx,
IntPointer seqLengthArrayQRO,
IntPointer seqLengthArrayKV,
cudnnSeqDataStruct qDesc,
Pointer queries,
Pointer residuals,
cudnnSeqDataStruct kDesc,
Pointer keys,
cudnnSeqDataStruct vDesc,
Pointer values,
cudnnSeqDataStruct oDesc,
Pointer out,
long weightSizeInBytes,
Pointer w,
long workSpaceSizeInBytes,
Pointer workSpace,
long reserveSpaceSizeInBytes,
Pointer reserveSpace) |
static int |
cudnnOpTensor(cudnnContext handle,
cudnnOpTensorStruct opTensorDesc,
Pointer alpha1,
cudnnTensorStruct aDesc,
Pointer A,
Pointer alpha2,
cudnnTensorStruct bDesc,
Pointer B,
Pointer beta,
cudnnTensorStruct cDesc,
Pointer C) |
static int |
cudnnPoolingBackward(cudnnContext handle,
cudnnPoolingStruct poolingDesc,
Pointer alpha,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnTensorStruct xDesc,
Pointer x,
Pointer beta,
cudnnTensorStruct dxDesc,
Pointer dx) |
static int |
cudnnPoolingForward(cudnnContext handle,
cudnnPoolingStruct poolingDesc,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
Pointer beta,
cudnnTensorStruct yDesc,
Pointer y) |
static int |
cudnnQueryRuntimeError(cudnnContext handle,
int[] rstatus,
int mode,
cudnnRuntimeTag_t tag) |
static int |
cudnnQueryRuntimeError(cudnnContext handle,
IntBuffer rstatus,
int mode,
cudnnRuntimeTag_t tag) |
static int |
cudnnQueryRuntimeError(cudnnContext handle,
IntPointer rstatus,
int mode,
cudnnRuntimeTag_t tag) |
static int |
cudnnReduceTensor(cudnnContext handle,
cudnnReduceTensorStruct reduceTensorDesc,
Pointer indices,
long indicesSizeInBytes,
Pointer workspace,
long workspaceSizeInBytes,
Pointer alpha,
cudnnTensorStruct aDesc,
Pointer A,
Pointer beta,
cudnnTensorStruct cDesc,
Pointer C) |
static int |
cudnnRestoreAlgorithm(cudnnContext handle,
Pointer algoSpace,
long algoSpaceSizeInBytes,
cudnnAlgorithmStruct algoDesc) |
static int |
cudnnRestoreDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
float dropout,
Pointer states,
long stateSizeInBytes,
long seed) |
static int |
cudnnRNNBackwardData(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct dyDesc,
Pointer dy,
cudnnTensorStruct dhyDesc,
Pointer dhy,
cudnnTensorStruct dcyDesc,
Pointer dcy,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnTensorStruct dxDesc,
Pointer dx,
cudnnTensorStruct dhxDesc,
Pointer dhx,
cudnnTensorStruct dcxDesc,
Pointer dcx,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNBackwardData(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer yDesc,
Pointer y,
PointerPointer dyDesc,
Pointer dy,
cudnnTensorStruct dhyDesc,
Pointer dhy,
cudnnTensorStruct dcyDesc,
Pointer dcy,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
PointerPointer dxDesc,
Pointer dx,
cudnnTensorStruct dhxDesc,
Pointer dhx,
cudnnTensorStruct dcxDesc,
Pointer dcx,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNBackwardDataEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnRNNDataStruct dyDesc,
Pointer dy,
cudnnRNNDataStruct dcDesc,
Pointer dcAttn,
cudnnTensorStruct dhyDesc,
Pointer dhy,
cudnnTensorStruct dcyDesc,
Pointer dcy,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnRNNDataStruct dxDesc,
Pointer dx,
cudnnTensorStruct dhxDesc,
Pointer dhx,
cudnnTensorStruct dcxDesc,
Pointer dcx,
cudnnRNNDataStruct dkDesc,
Pointer dkeys,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNBackwardWeights(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct yDesc,
Pointer y,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterStruct dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNBackwardWeights(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
PointerPointer yDesc,
Pointer y,
Pointer workspace,
long workSpaceSizeInBytes,
cudnnFilterStruct dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNBackwardWeightsEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnRNNDataStruct yDesc,
Pointer y,
Pointer workSpace,
long workSpaceSizeInBytes,
cudnnFilterStruct dwDesc,
Pointer dw,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNForwardInference(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnRNNForwardInference(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
PointerPointer yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
Pointer workspace,
long workSpaceSizeInBytes) |
static int |
cudnnRNNForwardInferenceEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
cudnnRNNDataStruct kDesc,
Pointer keys,
cudnnRNNDataStruct cDesc,
Pointer cAttn,
cudnnRNNDataStruct iDesc,
Pointer iAttn,
cudnnRNNDataStruct qDesc,
Pointer queries,
Pointer workSpace,
long workSpaceSizeInBytes) |
static int |
cudnnRNNForwardTraining(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
cudnnTensorStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNForwardTraining(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int seqLength,
PointerPointer xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
PointerPointer yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
Pointer workspace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNForwardTrainingEx(cudnnContext handle,
cudnnRNNStruct rnnDesc,
cudnnRNNDataStruct xDesc,
Pointer x,
cudnnTensorStruct hxDesc,
Pointer hx,
cudnnTensorStruct cxDesc,
Pointer cx,
cudnnFilterStruct wDesc,
Pointer w,
cudnnRNNDataStruct yDesc,
Pointer y,
cudnnTensorStruct hyDesc,
Pointer hy,
cudnnTensorStruct cyDesc,
Pointer cy,
cudnnRNNDataStruct kDesc,
Pointer keys,
cudnnRNNDataStruct cDesc,
Pointer cAttn,
cudnnRNNDataStruct iDesc,
Pointer iAttn,
cudnnRNNDataStruct qDesc,
Pointer queries,
Pointer workSpace,
long workSpaceSizeInBytes,
Pointer reserveSpace,
long reserveSpaceSizeInBytes) |
static int |
cudnnRNNGetClip(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int[] clipMode,
int[] clipNanOpt,
double[] lclip,
double[] rclip) |
static int |
cudnnRNNGetClip(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntBuffer clipMode,
IntBuffer clipNanOpt,
DoubleBuffer lclip,
DoubleBuffer rclip) |
static int |
cudnnRNNGetClip(cudnnContext handle,
cudnnRNNStruct rnnDesc,
IntPointer clipMode,
IntPointer clipNanOpt,
DoublePointer lclip,
DoublePointer rclip) |
static int |
cudnnRNNSetClip(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int clipMode,
int clipNanOpt,
double lclip,
double rclip) |
static int |
cudnnSaveAlgorithm(cudnnContext handle,
cudnnAlgorithmStruct algoDesc,
Pointer algoSpace,
long algoSpaceSizeInBytes) |
static int |
cudnnScaleTensor(cudnnContext handle,
cudnnTensorStruct yDesc,
Pointer y,
Pointer alpha) |
static int |
cudnnSetActivationDescriptor(cudnnActivationStruct activationDesc,
int mode,
int reluNanOpt,
double coef) |
static int |
cudnnSetAlgorithmDescriptor(cudnnAlgorithmStruct algoDesc,
cudnnAlgorithm_t algorithm) |
static int |
cudnnSetAlgorithmPerformance(cudnnAlgorithmPerformanceStruct algoPerf,
cudnnAlgorithmStruct algoDesc,
int status,
float time,
long memory) |
static int |
cudnnSetAttnDescriptor(cudnnAttnStruct attnDesc,
int queryMap,
int nHeads,
double smScaler,
int dataType,
int computePrec,
int mathType,
cudnnDropoutStruct attnDropoutDesc,
cudnnDropoutStruct postDropoutDesc,
int qSize,
int kSize,
int vSize,
int qProjSize,
int kProjSize,
int vProjSize,
int oProjSize,
int qoMaxSeqLength,
int kvMaxSeqLength,
int maxBatchSize,
int maxBeamSize) |
static int |
cudnnSetCallback(int mask,
Pointer udata,
cudnnCallback_t fptr) |
static int |
cudnnSetConvolution2dDescriptor(cudnnConvolutionStruct convDesc,
int pad_h,
int pad_w,
int u,
int v,
int dilation_h,
int dilation_w,
int mode,
int computeType) |
static int |
cudnnSetConvolutionGroupCount(cudnnConvolutionStruct convDesc,
int groupCount) |
static int |
cudnnSetConvolutionMathType(cudnnConvolutionStruct convDesc,
int mathType) |
static int |
cudnnSetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc,
int arrayLength,
int[] padA,
int[] filterStrideA,
int[] dilationA,
int mode,
int computeType) |
static int |
cudnnSetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc,
int arrayLength,
IntBuffer padA,
IntBuffer filterStrideA,
IntBuffer dilationA,
int mode,
int computeType) |
static int |
cudnnSetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc,
int arrayLength,
IntPointer padA,
IntPointer filterStrideA,
IntPointer dilationA,
int mode,
int computeType) |
static int |
cudnnSetCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc,
int compType) |
static int |
cudnnSetDropoutDescriptor(cudnnDropoutStruct dropoutDesc,
cudnnContext handle,
float dropout,
Pointer states,
long stateSizeInBytes,
long seed) |
static int |
cudnnSetFilter4dDescriptor(cudnnFilterStruct filterDesc,
int dataType,
int format,
int k,
int c,
int h,
int w) |
static int |
cudnnSetFilterNdDescriptor(cudnnFilterStruct filterDesc,
int dataType,
int format,
int nbDims,
int[] filterDimA) |
static int |
cudnnSetFilterNdDescriptor(cudnnFilterStruct filterDesc,
int dataType,
int format,
int nbDims,
IntBuffer filterDimA) |
static int |
cudnnSetFilterNdDescriptor(cudnnFilterStruct filterDesc,
int dataType,
int format,
int nbDims,
IntPointer filterDimA) |
static int |
cudnnSetLRNDescriptor(cudnnLRNStruct normDesc,
int lrnN,
double lrnAlpha,
double lrnBeta,
double lrnK) |
static int |
cudnnSetOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc,
int opTensorOp,
int opTensorCompType,
int opTensorNanOpt) |
static int |
cudnnSetPersistentRNNPlan(cudnnRNNStruct rnnDesc,
cudnnPersistentRNNPlan plan) |
static int |
cudnnSetPooling2dDescriptor(cudnnPoolingStruct poolingDesc,
int mode,
int maxpoolingNanOpt,
int windowHeight,
int windowWidth,
int verticalPadding,
int horizontalPadding,
int verticalStride,
int horizontalStride) |
static int |
cudnnSetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc,
int mode,
int maxpoolingNanOpt,
int nbDims,
int[] windowDimA,
int[] paddingA,
int[] strideA) |
static int |
cudnnSetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc,
int mode,
int maxpoolingNanOpt,
int nbDims,
IntBuffer windowDimA,
IntBuffer paddingA,
IntBuffer strideA) |
static int |
cudnnSetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc,
int mode,
int maxpoolingNanOpt,
int nbDims,
IntPointer windowDimA,
IntPointer paddingA,
IntPointer strideA) |
static int |
cudnnSetReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc,
int reduceTensorOp,
int reduceTensorCompType,
int reduceTensorNanOpt,
int reduceTensorIndices,
int reduceTensorIndicesType) |
static int |
cudnnSetRNNAlgorithmDescriptor(cudnnContext handle,
cudnnRNNStruct rnnDesc,
cudnnAlgorithmStruct algoDesc) |
static int |
cudnnSetRNNBiasMode(cudnnRNNStruct rnnDesc,
int biasMode) |
static int |
cudnnSetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
int dataType,
int layout,
int maxSeqLength,
int batchSize,
int vectorSize,
int[] seqLengthArray,
Pointer paddingFill) |
static int |
cudnnSetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
int dataType,
int layout,
int maxSeqLength,
int batchSize,
int vectorSize,
IntBuffer seqLengthArray,
Pointer paddingFill) |
static int |
cudnnSetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc,
int dataType,
int layout,
int maxSeqLength,
int batchSize,
int vectorSize,
IntPointer seqLengthArray,
Pointer paddingFill) |
static int |
cudnnSetRNNDescriptor_v5(cudnnRNNStruct rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutStruct dropoutDesc,
int inputMode,
int direction,
int mode,
int mathPrec) |
static int |
cudnnSetRNNDescriptor_v6(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutStruct dropoutDesc,
int inputMode,
int direction,
int mode,
int algo,
int mathPrec) |
static int |
cudnnSetRNNDescriptor(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int hiddenSize,
int numLayers,
cudnnDropoutStruct dropoutDesc,
int inputMode,
int direction,
int mode,
int algo,
int mathPrec) |
static int |
cudnnSetRNNMatrixMathType(cudnnRNNStruct rnnDesc,
int mType) |
static int |
cudnnSetRNNPaddingMode(cudnnRNNStruct rnnDesc,
int paddingMode) |
static int |
cudnnSetRNNProjectionLayers(cudnnContext handle,
cudnnRNNStruct rnnDesc,
int recProjSize,
int outProjSize) |
static int |
cudnnSetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
int dataType,
int nbDims,
int[] dimA,
int[] axes,
long seqLengthArraySize,
int[] seqLengthArray,
Pointer paddingFill) |
static int |
cudnnSetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
int dataType,
int nbDims,
IntBuffer dimA,
IntBuffer axes,
long seqLengthArraySize,
IntBuffer seqLengthArray,
Pointer paddingFill) |
static int |
cudnnSetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc,
int dataType,
int nbDims,
IntPointer dimA,
IntPointer axes,
long seqLengthArraySize,
IntPointer seqLengthArray,
Pointer paddingFill) |
static int |
cudnnSetSpatialTransformerNdDescriptor(cudnnSpatialTransformerStruct stDesc,
int samplerType,
int dataType,
int nbDims,
int[] dimA) |
static int |
cudnnSetSpatialTransformerNdDescriptor(cudnnSpatialTransformerStruct stDesc,
int samplerType,
int dataType,
int nbDims,
IntBuffer dimA) |
static int |
cudnnSetSpatialTransformerNdDescriptor(cudnnSpatialTransformerStruct stDesc,
int samplerType,
int dataType,
int nbDims,
IntPointer dimA) |
static int |
cudnnSetStream(cudnnContext handle,
CUstream_st streamId) |
static int |
cudnnSetTensor(cudnnContext handle,
cudnnTensorStruct yDesc,
Pointer y,
Pointer valuePtr) |
static int |
cudnnSetTensor4dDescriptor(cudnnTensorStruct tensorDesc,
int format,
int dataType,
int n,
int c,
int h,
int w) |
static int |
cudnnSetTensor4dDescriptorEx(cudnnTensorStruct tensorDesc,
int dataType,
int n,
int c,
int h,
int w,
int nStride,
int cStride,
int hStride,
int wStride) |
static int |
cudnnSetTensorNdDescriptor(cudnnTensorStruct tensorDesc,
int dataType,
int nbDims,
int[] dimA,
int[] strideA) |
static int |
cudnnSetTensorNdDescriptor(cudnnTensorStruct tensorDesc,
int dataType,
int nbDims,
IntBuffer dimA,
IntBuffer strideA) |
static int |
cudnnSetTensorNdDescriptor(cudnnTensorStruct tensorDesc,
int dataType,
int nbDims,
IntPointer dimA,
IntPointer strideA) |
static int |
cudnnSetTensorNdDescriptorEx(cudnnTensorStruct tensorDesc,
int format,
int dataType,
int nbDims,
int[] dimA) |
static int |
cudnnSetTensorNdDescriptorEx(cudnnTensorStruct tensorDesc,
int format,
int dataType,
int nbDims,
IntBuffer dimA) |
static int |
cudnnSetTensorNdDescriptorEx(cudnnTensorStruct tensorDesc,
int format,
int dataType,
int nbDims,
IntPointer dimA) |
static int |
cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDims,
int destFormat,
int[] padBeforeA,
int[] padAfterA,
int[] foldA,
long direction) |
static int |
cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDims,
int destFormat,
IntBuffer padBeforeA,
IntBuffer padAfterA,
IntBuffer foldA,
long direction) |
static int |
cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDims,
int destFormat,
IntPointer padBeforeA,
IntPointer padAfterA,
IntPointer foldA,
long direction)
Initialize a previously created tensor transform descriptor.
|
static int |
cudnnSoftmaxBackward(cudnnContext handle,
int algo,
int mode,
Pointer alpha,
cudnnTensorStruct yDesc,
Pointer y,
cudnnTensorStruct dyDesc,
Pointer dy,
Pointer beta,
cudnnTensorStruct dxDesc,
Pointer dx) |
static int |
cudnnSoftmaxForward(cudnnContext handle,
int algo,
int mode,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
Pointer beta,
cudnnTensorStruct yDesc,
Pointer y) |
static int |
cudnnSpatialTfGridGeneratorBackward(cudnnContext handle,
cudnnSpatialTransformerStruct stDesc,
Pointer dgrid,
Pointer dtheta) |
static int |
cudnnSpatialTfGridGeneratorForward(cudnnContext handle,
cudnnSpatialTransformerStruct stDesc,
Pointer theta,
Pointer grid) |
static int |
cudnnSpatialTfSamplerBackward(cudnnContext handle,
cudnnSpatialTransformerStruct stDesc,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
Pointer beta,
cudnnTensorStruct dxDesc,
Pointer dx,
Pointer alphaDgrid,
cudnnTensorStruct dyDesc,
Pointer dy,
Pointer grid,
Pointer betaDgrid,
Pointer dgrid) |
static int |
cudnnSpatialTfSamplerForward(cudnnContext handle,
cudnnSpatialTransformerStruct stDesc,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
Pointer grid,
Pointer beta,
cudnnTensorStruct yDesc,
Pointer y) |
static int |
cudnnTransformTensor(cudnnContext handle,
Pointer alpha,
cudnnTensorStruct xDesc,
Pointer x,
Pointer beta,
cudnnTensorStruct yDesc,
Pointer y) |
static int |
cudnnTransformTensorEx(cudnnContext handle,
cudnnTensorTransformStruct transDesc,
Pointer alpha,
cudnnTensorStruct srcDesc,
Pointer srcData,
Pointer beta,
cudnnTensorStruct destDesc,
Pointer destData) |
public static final int CUDNN_MAJOR
public static final int CUDNN_MINOR
public static final int CUDNN_PATCHLEVEL
public static final int CUDNN_VERSION
public static final int CUDNN_STATUS_SUCCESS
public static final int CUDNN_STATUS_NOT_INITIALIZED
public static final int CUDNN_STATUS_ALLOC_FAILED
public static final int CUDNN_STATUS_BAD_PARAM
public static final int CUDNN_STATUS_INTERNAL_ERROR
public static final int CUDNN_STATUS_INVALID_VALUE
public static final int CUDNN_STATUS_ARCH_MISMATCH
public static final int CUDNN_STATUS_MAPPING_ERROR
public static final int CUDNN_STATUS_EXECUTION_FAILED
public static final int CUDNN_STATUS_NOT_SUPPORTED
public static final int CUDNN_STATUS_LICENSE_ERROR
public static final int CUDNN_STATUS_RUNTIME_PREREQUISITE_MISSING
public static final int CUDNN_STATUS_RUNTIME_IN_PROGRESS
public static final int CUDNN_STATUS_RUNTIME_FP_OVERFLOW
public static final int CUDNN_ERRQUERY_RAWCODE
public static final int CUDNN_ERRQUERY_NONBLOCKING
public static final int CUDNN_ERRQUERY_BLOCKING
public static final int MAJOR_VERSION
public static final int MINOR_VERSION
public static final int PATCH_LEVEL
public static final int CUDNN_DATA_FLOAT
public static final int CUDNN_DATA_DOUBLE
public static final int CUDNN_DATA_HALF
public static final int CUDNN_DATA_INT8
public static final int CUDNN_DATA_INT32
public static final int CUDNN_DATA_INT8x4
public static final int CUDNN_DATA_UINT8
public static final int CUDNN_DATA_UINT8x4
public static final int CUDNN_DATA_INT8x32
public static final int CUDNN_DEFAULT_MATH
public static final int CUDNN_TENSOR_OP_MATH
public static final int CUDNN_TENSOR_OP_MATH_ALLOW_CONVERSION
public static final int CUDNN_NOT_PROPAGATE_NAN
public static final int CUDNN_PROPAGATE_NAN
public static final int CUDNN_NON_DETERMINISTIC
public static final int CUDNN_DETERMINISTIC
public static final int CUDNN_DIM_MAX
public static final int CUDNN_TENSOR_NCHW
public static final int CUDNN_TENSOR_NHWC
public static final int CUDNN_TENSOR_NCHW_VECT_C
public static final long CUDNN_TRANSFORM_FOLD
public static final long CUDNN_TRANSFORM_UNFOLD
public static final int CUDNN_OP_TENSOR_ADD
public static final int CUDNN_OP_TENSOR_MUL
public static final int CUDNN_OP_TENSOR_MIN
public static final int CUDNN_OP_TENSOR_MAX
public static final int CUDNN_OP_TENSOR_SQRT
public static final int CUDNN_OP_TENSOR_NOT
public static final int CUDNN_REDUCE_TENSOR_ADD
public static final int CUDNN_REDUCE_TENSOR_MUL
public static final int CUDNN_REDUCE_TENSOR_MIN
public static final int CUDNN_REDUCE_TENSOR_MAX
public static final int CUDNN_REDUCE_TENSOR_AMAX
public static final int CUDNN_REDUCE_TENSOR_AVG
public static final int CUDNN_REDUCE_TENSOR_NORM1
public static final int CUDNN_REDUCE_TENSOR_NORM2
public static final int CUDNN_REDUCE_TENSOR_MUL_NO_ZEROS
public static final int CUDNN_REDUCE_TENSOR_NO_INDICES
public static final int CUDNN_REDUCE_TENSOR_FLATTENED_INDICES
public static final int CUDNN_32BIT_INDICES
public static final int CUDNN_64BIT_INDICES
public static final int CUDNN_16BIT_INDICES
public static final int CUDNN_8BIT_INDICES
public static final int CUDNN_CONVOLUTION
public static final int CUDNN_CROSS_CORRELATION
public static final int CUDNN_CONVOLUTION_FWD_NO_WORKSPACE
public static final int CUDNN_CONVOLUTION_FWD_PREFER_FASTEST
public static final int CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT
public static final int CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_GEMM
public static final int CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM
public static final int CUDNN_CONVOLUTION_FWD_ALGO_GEMM
public static final int CUDNN_CONVOLUTION_FWD_ALGO_DIRECT
public static final int CUDNN_CONVOLUTION_FWD_ALGO_FFT
public static final int CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING
public static final int CUDNN_CONVOLUTION_FWD_ALGO_WINOGRAD
public static final int CUDNN_CONVOLUTION_FWD_ALGO_WINOGRAD_NONFUSED
public static final int CUDNN_CONVOLUTION_FWD_ALGO_COUNT
public static final int CUDNN_CONVOLUTION_BWD_FILTER_NO_WORKSPACE
public static final int CUDNN_CONVOLUTION_BWD_FILTER_PREFER_FASTEST
public static final int CUDNN_CONVOLUTION_BWD_FILTER_SPECIFY_WORKSPACE_LIMIT
public static final int CUDNN_CONVOLUTION_BWD_FILTER_ALGO_0
public static final int CUDNN_CONVOLUTION_BWD_FILTER_ALGO_1
public static final int CUDNN_CONVOLUTION_BWD_FILTER_ALGO_FFT
public static final int CUDNN_CONVOLUTION_BWD_FILTER_ALGO_3
public static final int CUDNN_CONVOLUTION_BWD_FILTER_ALGO_WINOGRAD
public static final int CUDNN_CONVOLUTION_BWD_FILTER_ALGO_WINOGRAD_NONFUSED
public static final int CUDNN_CONVOLUTION_BWD_FILTER_ALGO_FFT_TILING
public static final int CUDNN_CONVOLUTION_BWD_FILTER_ALGO_COUNT
public static final int CUDNN_CONVOLUTION_BWD_DATA_NO_WORKSPACE
public static final int CUDNN_CONVOLUTION_BWD_DATA_PREFER_FASTEST
public static final int CUDNN_CONVOLUTION_BWD_DATA_SPECIFY_WORKSPACE_LIMIT
public static final int CUDNN_CONVOLUTION_BWD_DATA_ALGO_0
public static final int CUDNN_CONVOLUTION_BWD_DATA_ALGO_1
public static final int CUDNN_CONVOLUTION_BWD_DATA_ALGO_FFT
public static final int CUDNN_CONVOLUTION_BWD_DATA_ALGO_FFT_TILING
public static final int CUDNN_CONVOLUTION_BWD_DATA_ALGO_WINOGRAD
public static final int CUDNN_CONVOLUTION_BWD_DATA_ALGO_WINOGRAD_NONFUSED
public static final int CUDNN_CONVOLUTION_BWD_DATA_ALGO_COUNT
public static final int CUDNN_SOFTMAX_FAST
public static final int CUDNN_SOFTMAX_ACCURATE
public static final int CUDNN_SOFTMAX_LOG
public static final int CUDNN_SOFTMAX_MODE_INSTANCE
public static final int CUDNN_SOFTMAX_MODE_CHANNEL
public static final int CUDNN_POOLING_MAX
public static final int CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING
public static final int CUDNN_POOLING_AVERAGE_COUNT_EXCLUDE_PADDING
public static final int CUDNN_POOLING_MAX_DETERMINISTIC
public static final int CUDNN_ACTIVATION_SIGMOID
public static final int CUDNN_ACTIVATION_RELU
public static final int CUDNN_ACTIVATION_TANH
public static final int CUDNN_ACTIVATION_CLIPPED_RELU
public static final int CUDNN_ACTIVATION_ELU
public static final int CUDNN_ACTIVATION_IDENTITY
public static final int CUDNN_LRN_MIN_N
public static final int CUDNN_LRN_MAX_N
public static final double CUDNN_LRN_MIN_K
public static final double CUDNN_LRN_MIN_BETA
public static final int CUDNN_LRN_CROSS_CHANNEL_DIM1
public static final int CUDNN_DIVNORM_PRECOMPUTED_MEANS
public static final int CUDNN_BATCHNORM_PER_ACTIVATION
public static final int CUDNN_BATCHNORM_SPATIAL
public static final int CUDNN_BATCHNORM_SPATIAL_PERSISTENT
public static final double CUDNN_BN_MIN_EPSILON
public static final int CUDNN_BATCHNORM_OPS_BN
public static final int CUDNN_BATCHNORM_OPS_BN_ACTIVATION
public static final int CUDNN_BATCHNORM_OPS_BN_ADD_ACTIVATION
public static final int CUDNN_SAMPLER_BILINEAR
public static final int CUDNN_RNN_ALGO_STANDARD
public static final int CUDNN_RNN_ALGO_PERSIST_STATIC
public static final int CUDNN_RNN_ALGO_PERSIST_DYNAMIC
public static final int CUDNN_RNN_ALGO_COUNT
public static final int CUDNN_RNN_RELU
public static final int CUDNN_RNN_TANH
public static final int CUDNN_LSTM
public static final int CUDNN_GRU
public static final int CUDNN_RNN_NO_BIAS
public static final int CUDNN_RNN_SINGLE_INP_BIAS
public static final int CUDNN_RNN_DOUBLE_BIAS
public static final int CUDNN_RNN_SINGLE_REC_BIAS
public static final int CUDNN_UNIDIRECTIONAL
public static final int CUDNN_BIDIRECTIONAL
public static final int CUDNN_LINEAR_INPUT
public static final int CUDNN_SKIP_INPUT
public static final int CUDNN_RNN_CLIP_NONE
public static final int CUDNN_RNN_CLIP_MINMAX
public static final int CUDNN_RNN_DATA_LAYOUT_SEQ_MAJOR_UNPACKED
public static final int CUDNN_RNN_DATA_LAYOUT_SEQ_MAJOR_PACKED
public static final int CUDNN_RNN_DATA_LAYOUT_BATCH_MAJOR_UNPACKED
public static final int CUDNN_RNN_PADDED_IO_DISABLED
public static final int CUDNN_RNN_PADDED_IO_ENABLED
public static final int CUDNN_SEQDATA_TIME_DIM
public static final int CUDNN_SEQDATA_BATCH_DIM
public static final int CUDNN_SEQDATA_BEAM_DIM
public static final int CUDNN_SEQDATA_VECT_DIM
public static final int CUDNN_SEQDATA_DIM_COUNT
public static final int CUDNN_ATTN_QUERYMAP_ALL_TO_ONE
public static final int CUDNN_ATTN_QUERYMAP_ONE_TO_ONE
public static final int CUDNN_MH_ATTN_Q_WEIGHTS
public static final int CUDNN_MH_ATTN_K_WEIGHTS
public static final int CUDNN_MH_ATTN_V_WEIGHTS
public static final int CUDNN_MH_ATTN_O_WEIGHTS
public static final int CUDNN_WGRAD_MODE_ADD
public static final int CUDNN_WGRAD_MODE_SET
public static final int CUDNN_CTC_LOSS_ALGO_DETERMINISTIC
public static final int CUDNN_CTC_LOSS_ALGO_NON_DETERMINISTIC
public static final int CUDNN_SEV_FATAL
public static final int CUDNN_SEV_ERROR
public static final int CUDNN_SEV_WARNING
public static final int CUDNN_SEV_INFO
public static final long CUDNN_SEV_ERROR_EN
public static final long CUDNN_SEV_WARNING_EN
public static final long CUDNN_SEV_INFO_EN
@Cast(value="const char*") public static BytePointer cudnnGetErrorString(@Cast(value="cudnnStatus_t") int status)
@Cast(value="cudnnStatus_t") public static int cudnnQueryRuntimeError(cudnnContext handle, @Cast(value="cudnnStatus_t*") IntPointer rstatus, @Cast(value="cudnnErrQueryMode_t") int mode, cudnnRuntimeTag_t tag)
@Cast(value="cudnnStatus_t") public static int cudnnQueryRuntimeError(cudnnContext handle, @Cast(value="cudnnStatus_t*") IntBuffer rstatus, @Cast(value="cudnnErrQueryMode_t") int mode, cudnnRuntimeTag_t tag)
@Cast(value="cudnnStatus_t") public static int cudnnQueryRuntimeError(cudnnContext handle, @Cast(value="cudnnStatus_t*") int[] rstatus, @Cast(value="cudnnErrQueryMode_t") int mode, cudnnRuntimeTag_t tag)
@Cast(value="cudnnStatus_t") public static int cudnnGetProperty(@Cast(value="libraryPropertyType") int type, IntPointer value)
@Cast(value="cudnnStatus_t") public static int cudnnGetProperty(@Cast(value="libraryPropertyType") int type, IntBuffer value)
@Cast(value="cudnnStatus_t") public static int cudnnGetProperty(@Cast(value="libraryPropertyType") int type, int[] value)
@Cast(value="cudnnStatus_t") public static int cudnnCreate(@ByPtrPtr cudnnContext handle)
@Cast(value="cudnnStatus_t") public static int cudnnDestroy(cudnnContext handle)
@Cast(value="cudnnStatus_t") public static int cudnnSetStream(cudnnContext handle, CUstream_st streamId)
@Cast(value="cudnnStatus_t") public static int cudnnGetStream(cudnnContext handle, @ByPtrPtr CUstream_st streamId)
@Cast(value="cudnnStatus_t") public static int cudnnCreateTensorDescriptor(@Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct tensorDesc)
@Cast(value="cudnnStatus_t") public static int cudnnCreateTensorDescriptor(@Cast(value="cudnnTensorStruct**") PointerPointer tensorDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensor4dDescriptor(cudnnTensorStruct tensorDesc, @Cast(value="cudnnTensorFormat_t") int format, @Cast(value="cudnnDataType_t") int dataType, int n, int c, int h, int w)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensor4dDescriptorEx(cudnnTensorStruct tensorDesc, @Cast(value="cudnnDataType_t") int dataType, int n, int c, int h, int w, int nStride, int cStride, int hStride, int wStride)
@Cast(value="cudnnStatus_t") public static int cudnnGetTensor4dDescriptor(cudnnTensorStruct tensorDesc, @Cast(value="cudnnDataType_t*") IntPointer dataType, IntPointer n, IntPointer c, IntPointer h, IntPointer w, IntPointer nStride, IntPointer cStride, IntPointer hStride, IntPointer wStride)
@Cast(value="cudnnStatus_t") public static int cudnnGetTensor4dDescriptor(cudnnTensorStruct tensorDesc, @Cast(value="cudnnDataType_t*") IntBuffer dataType, IntBuffer n, IntBuffer c, IntBuffer h, IntBuffer w, IntBuffer nStride, IntBuffer cStride, IntBuffer hStride, IntBuffer wStride)
@Cast(value="cudnnStatus_t") public static int cudnnGetTensor4dDescriptor(cudnnTensorStruct tensorDesc, @Cast(value="cudnnDataType_t*") int[] dataType, int[] n, int[] c, int[] h, int[] w, int[] nStride, int[] cStride, int[] hStride, int[] wStride)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensorNdDescriptor(cudnnTensorStruct tensorDesc, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const IntPointer dimA, @Const IntPointer strideA)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensorNdDescriptor(cudnnTensorStruct tensorDesc, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const IntBuffer dimA, @Const IntBuffer strideA)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensorNdDescriptor(cudnnTensorStruct tensorDesc, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const int[] dimA, @Const int[] strideA)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensorNdDescriptorEx(cudnnTensorStruct tensorDesc, @Cast(value="cudnnTensorFormat_t") int format, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const IntPointer dimA)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensorNdDescriptorEx(cudnnTensorStruct tensorDesc, @Cast(value="cudnnTensorFormat_t") int format, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const IntBuffer dimA)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensorNdDescriptorEx(cudnnTensorStruct tensorDesc, @Cast(value="cudnnTensorFormat_t") int format, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const int[] dimA)
@Cast(value="cudnnStatus_t") public static int cudnnGetTensorNdDescriptor(cudnnTensorStruct tensorDesc, int nbDimsRequested, @Cast(value="cudnnDataType_t*") IntPointer dataType, IntPointer nbDims, IntPointer dimA, IntPointer strideA)
@Cast(value="cudnnStatus_t") public static int cudnnGetTensorNdDescriptor(cudnnTensorStruct tensorDesc, int nbDimsRequested, @Cast(value="cudnnDataType_t*") IntBuffer dataType, IntBuffer nbDims, IntBuffer dimA, IntBuffer strideA)
@Cast(value="cudnnStatus_t") public static int cudnnGetTensorNdDescriptor(cudnnTensorStruct tensorDesc, int nbDimsRequested, @Cast(value="cudnnDataType_t*") int[] dataType, int[] nbDims, int[] dimA, int[] strideA)
@Cast(value="cudnnStatus_t") public static int cudnnGetTensorSizeInBytes(cudnnTensorStruct tensorDesc, @Cast(value="size_t*") SizeTPointer size)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyTensorDescriptor(cudnnTensorStruct tensorDesc)
@Cast(value="cudnnStatus_t") public static int cudnnInitTransformDest(cudnnTensorTransformStruct transformDesc, cudnnTensorStruct srcDesc, cudnnTensorStruct destDesc, @Cast(value="size_t*") SizeTPointer destSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnCreateTensorTransformDescriptor(@ByPtrPtr cudnnTensorTransformStruct transformDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc, @Cast(value="const uint32_t") int nbDims, @Cast(value="const cudnnTensorFormat_t") int destFormat, @Const IntPointer padBeforeA, @Const IntPointer padAfterA, @Cast(value="const uint32_t*") IntPointer foldA, @Cast(value="const cudnnFoldingDirection_t") long direction)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc, @Cast(value="const uint32_t") int nbDims, @Cast(value="const cudnnTensorFormat_t") int destFormat, @Const IntBuffer padBeforeA, @Const IntBuffer padAfterA, @Cast(value="const uint32_t*") IntBuffer foldA, @Cast(value="const cudnnFoldingDirection_t") long direction)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc, @Cast(value="const uint32_t") int nbDims, @Cast(value="const cudnnTensorFormat_t") int destFormat, @Const int[] padBeforeA, @Const int[] padAfterA, @Cast(value="const uint32_t*") int[] foldA, @Cast(value="const cudnnFoldingDirection_t") long direction)
@Cast(value="cudnnStatus_t") public static int cudnnGetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc, @Cast(value="uint32_t") int nbDimsRequested, @Cast(value="cudnnTensorFormat_t*") IntPointer destFormat, IntPointer padBeforeA, IntPointer padAfterA, @Cast(value="uint32_t*") IntPointer foldA, @Cast(value="cudnnFoldingDirection_t*") LongPointer direction)
@Cast(value="cudnnStatus_t") public static int cudnnGetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc, @Cast(value="uint32_t") int nbDimsRequested, @Cast(value="cudnnTensorFormat_t*") IntBuffer destFormat, IntBuffer padBeforeA, IntBuffer padAfterA, @Cast(value="uint32_t*") IntBuffer foldA, @Cast(value="cudnnFoldingDirection_t*") LongBuffer direction)
@Cast(value="cudnnStatus_t") public static int cudnnGetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc, @Cast(value="uint32_t") int nbDimsRequested, @Cast(value="cudnnTensorFormat_t*") int[] destFormat, int[] padBeforeA, int[] padAfterA, @Cast(value="uint32_t*") int[] foldA, @Cast(value="cudnnFoldingDirection_t*") long[] direction)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc)
@Cast(value="cudnnStatus_t") public static int cudnnTransformTensor(cudnnContext handle, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer beta, cudnnTensorStruct yDesc, Pointer y)
@Cast(value="cudnnStatus_t") public static int cudnnTransformTensorEx(cudnnContext handle, cudnnTensorTransformStruct transDesc, @Const Pointer alpha, cudnnTensorStruct srcDesc, @Const Pointer srcData, @Const Pointer beta, cudnnTensorStruct destDesc, Pointer destData)
@Cast(value="cudnnStatus_t") public static int cudnnAddTensor(cudnnContext handle, @Const Pointer alpha, cudnnTensorStruct aDesc, @Const Pointer A, @Const Pointer beta, cudnnTensorStruct cDesc, Pointer C)
@Cast(value="cudnnStatus_t") public static int cudnnCreateOpTensorDescriptor(@ByPtrPtr cudnnOpTensorStruct opTensorDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc, @Cast(value="cudnnOpTensorOp_t") int opTensorOp, @Cast(value="cudnnDataType_t") int opTensorCompType, @Cast(value="cudnnNanPropagation_t") int opTensorNanOpt)
@Cast(value="cudnnStatus_t") public static int cudnnGetOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc, @Cast(value="cudnnOpTensorOp_t*") IntPointer opTensorOp, @Cast(value="cudnnDataType_t*") IntPointer opTensorCompType, @Cast(value="cudnnNanPropagation_t*") IntPointer opTensorNanOpt)
@Cast(value="cudnnStatus_t") public static int cudnnGetOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc, @Cast(value="cudnnOpTensorOp_t*") IntBuffer opTensorOp, @Cast(value="cudnnDataType_t*") IntBuffer opTensorCompType, @Cast(value="cudnnNanPropagation_t*") IntBuffer opTensorNanOpt)
@Cast(value="cudnnStatus_t") public static int cudnnGetOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc, @Cast(value="cudnnOpTensorOp_t*") int[] opTensorOp, @Cast(value="cudnnDataType_t*") int[] opTensorCompType, @Cast(value="cudnnNanPropagation_t*") int[] opTensorNanOpt)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyOpTensorDescriptor(cudnnOpTensorStruct opTensorDesc)
@Cast(value="cudnnStatus_t") public static int cudnnOpTensor(cudnnContext handle, cudnnOpTensorStruct opTensorDesc, @Const Pointer alpha1, cudnnTensorStruct aDesc, @Const Pointer A, @Const Pointer alpha2, cudnnTensorStruct bDesc, @Const Pointer B, @Const Pointer beta, cudnnTensorStruct cDesc, Pointer C)
@Cast(value="cudnnStatus_t") public static int cudnnCreateReduceTensorDescriptor(@ByPtrPtr cudnnReduceTensorStruct reduceTensorDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc, @Cast(value="cudnnReduceTensorOp_t") int reduceTensorOp, @Cast(value="cudnnDataType_t") int reduceTensorCompType, @Cast(value="cudnnNanPropagation_t") int reduceTensorNanOpt, @Cast(value="cudnnReduceTensorIndices_t") int reduceTensorIndices, @Cast(value="cudnnIndicesType_t") int reduceTensorIndicesType)
@Cast(value="cudnnStatus_t") public static int cudnnGetReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc, @Cast(value="cudnnReduceTensorOp_t*") IntPointer reduceTensorOp, @Cast(value="cudnnDataType_t*") IntPointer reduceTensorCompType, @Cast(value="cudnnNanPropagation_t*") IntPointer reduceTensorNanOpt, @Cast(value="cudnnReduceTensorIndices_t*") IntPointer reduceTensorIndices, @Cast(value="cudnnIndicesType_t*") IntPointer reduceTensorIndicesType)
@Cast(value="cudnnStatus_t") public static int cudnnGetReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc, @Cast(value="cudnnReduceTensorOp_t*") IntBuffer reduceTensorOp, @Cast(value="cudnnDataType_t*") IntBuffer reduceTensorCompType, @Cast(value="cudnnNanPropagation_t*") IntBuffer reduceTensorNanOpt, @Cast(value="cudnnReduceTensorIndices_t*") IntBuffer reduceTensorIndices, @Cast(value="cudnnIndicesType_t*") IntBuffer reduceTensorIndicesType)
@Cast(value="cudnnStatus_t") public static int cudnnGetReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc, @Cast(value="cudnnReduceTensorOp_t*") int[] reduceTensorOp, @Cast(value="cudnnDataType_t*") int[] reduceTensorCompType, @Cast(value="cudnnNanPropagation_t*") int[] reduceTensorNanOpt, @Cast(value="cudnnReduceTensorIndices_t*") int[] reduceTensorIndices, @Cast(value="cudnnIndicesType_t*") int[] reduceTensorIndicesType)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyReduceTensorDescriptor(cudnnReduceTensorStruct reduceTensorDesc)
@Cast(value="cudnnStatus_t") public static int cudnnGetReductionIndicesSize(cudnnContext handle, cudnnReduceTensorStruct reduceTensorDesc, cudnnTensorStruct aDesc, cudnnTensorStruct cDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetReductionWorkspaceSize(cudnnContext handle, cudnnReduceTensorStruct reduceTensorDesc, cudnnTensorStruct aDesc, cudnnTensorStruct cDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnReduceTensor(cudnnContext handle, cudnnReduceTensorStruct reduceTensorDesc, Pointer indices, @Cast(value="size_t") long indicesSizeInBytes, Pointer workspace, @Cast(value="size_t") long workspaceSizeInBytes, @Const Pointer alpha, cudnnTensorStruct aDesc, @Const Pointer A, @Const Pointer beta, cudnnTensorStruct cDesc, Pointer C)
@Cast(value="cudnnStatus_t") public static int cudnnSetTensor(cudnnContext handle, cudnnTensorStruct yDesc, Pointer y, @Const Pointer valuePtr)
@Cast(value="cudnnStatus_t") public static int cudnnScaleTensor(cudnnContext handle, cudnnTensorStruct yDesc, Pointer y, @Const Pointer alpha)
@Cast(value="cudnnStatus_t") public static int cudnnCreateFilterDescriptor(@ByPtrPtr cudnnFilterStruct filterDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetFilter4dDescriptor(cudnnFilterStruct filterDesc, @Cast(value="cudnnDataType_t") int dataType, @Cast(value="cudnnTensorFormat_t") int format, int k, int c, int h, int w)
@Cast(value="cudnnStatus_t") public static int cudnnGetFilter4dDescriptor(cudnnFilterStruct filterDesc, @Cast(value="cudnnDataType_t*") IntPointer dataType, @Cast(value="cudnnTensorFormat_t*") IntPointer format, IntPointer k, IntPointer c, IntPointer h, IntPointer w)
@Cast(value="cudnnStatus_t") public static int cudnnGetFilter4dDescriptor(cudnnFilterStruct filterDesc, @Cast(value="cudnnDataType_t*") IntBuffer dataType, @Cast(value="cudnnTensorFormat_t*") IntBuffer format, IntBuffer k, IntBuffer c, IntBuffer h, IntBuffer w)
@Cast(value="cudnnStatus_t") public static int cudnnGetFilter4dDescriptor(cudnnFilterStruct filterDesc, @Cast(value="cudnnDataType_t*") int[] dataType, @Cast(value="cudnnTensorFormat_t*") int[] format, int[] k, int[] c, int[] h, int[] w)
@Cast(value="cudnnStatus_t") public static int cudnnSetFilterNdDescriptor(cudnnFilterStruct filterDesc, @Cast(value="cudnnDataType_t") int dataType, @Cast(value="cudnnTensorFormat_t") int format, int nbDims, @Const IntPointer filterDimA)
@Cast(value="cudnnStatus_t") public static int cudnnSetFilterNdDescriptor(cudnnFilterStruct filterDesc, @Cast(value="cudnnDataType_t") int dataType, @Cast(value="cudnnTensorFormat_t") int format, int nbDims, @Const IntBuffer filterDimA)
@Cast(value="cudnnStatus_t") public static int cudnnSetFilterNdDescriptor(cudnnFilterStruct filterDesc, @Cast(value="cudnnDataType_t") int dataType, @Cast(value="cudnnTensorFormat_t") int format, int nbDims, @Const int[] filterDimA)
@Cast(value="cudnnStatus_t") public static int cudnnGetFilterNdDescriptor(cudnnFilterStruct filterDesc, int nbDimsRequested, @Cast(value="cudnnDataType_t*") IntPointer dataType, @Cast(value="cudnnTensorFormat_t*") IntPointer format, IntPointer nbDims, IntPointer filterDimA)
@Cast(value="cudnnStatus_t") public static int cudnnGetFilterNdDescriptor(cudnnFilterStruct filterDesc, int nbDimsRequested, @Cast(value="cudnnDataType_t*") IntBuffer dataType, @Cast(value="cudnnTensorFormat_t*") IntBuffer format, IntBuffer nbDims, IntBuffer filterDimA)
@Cast(value="cudnnStatus_t") public static int cudnnGetFilterNdDescriptor(cudnnFilterStruct filterDesc, int nbDimsRequested, @Cast(value="cudnnDataType_t*") int[] dataType, @Cast(value="cudnnTensorFormat_t*") int[] format, int[] nbDims, int[] filterDimA)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyFilterDescriptor(cudnnFilterStruct filterDesc)
@Cast(value="cudnnStatus_t") public static int cudnnCreateConvolutionDescriptor(@ByPtrPtr cudnnConvolutionStruct convDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetConvolutionMathType(cudnnConvolutionStruct convDesc, @Cast(value="cudnnMathType_t") int mathType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionMathType(cudnnConvolutionStruct convDesc, @Cast(value="cudnnMathType_t*") IntPointer mathType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionMathType(cudnnConvolutionStruct convDesc, @Cast(value="cudnnMathType_t*") IntBuffer mathType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionMathType(cudnnConvolutionStruct convDesc, @Cast(value="cudnnMathType_t*") int[] mathType)
@Cast(value="cudnnStatus_t") public static int cudnnSetConvolutionGroupCount(cudnnConvolutionStruct convDesc, int groupCount)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionGroupCount(cudnnConvolutionStruct convDesc, IntPointer groupCount)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionGroupCount(cudnnConvolutionStruct convDesc, IntBuffer groupCount)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionGroupCount(cudnnConvolutionStruct convDesc, int[] groupCount)
@Cast(value="cudnnStatus_t") public static int cudnnSetConvolution2dDescriptor(cudnnConvolutionStruct convDesc, int pad_h, int pad_w, int u, int v, int dilation_h, int dilation_w, @Cast(value="cudnnConvolutionMode_t") int mode, @Cast(value="cudnnDataType_t") int computeType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolution2dDescriptor(cudnnConvolutionStruct convDesc, IntPointer pad_h, IntPointer pad_w, IntPointer u, IntPointer v, IntPointer dilation_h, IntPointer dilation_w, @Cast(value="cudnnConvolutionMode_t*") IntPointer mode, @Cast(value="cudnnDataType_t*") IntPointer computeType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolution2dDescriptor(cudnnConvolutionStruct convDesc, IntBuffer pad_h, IntBuffer pad_w, IntBuffer u, IntBuffer v, IntBuffer dilation_h, IntBuffer dilation_w, @Cast(value="cudnnConvolutionMode_t*") IntBuffer mode, @Cast(value="cudnnDataType_t*") IntBuffer computeType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolution2dDescriptor(cudnnConvolutionStruct convDesc, int[] pad_h, int[] pad_w, int[] u, int[] v, int[] dilation_h, int[] dilation_w, @Cast(value="cudnnConvolutionMode_t*") int[] mode, @Cast(value="cudnnDataType_t*") int[] computeType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionStruct convDesc, cudnnTensorStruct inputTensorDesc, cudnnFilterStruct filterDesc, IntPointer n, IntPointer c, IntPointer h, IntPointer w)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionStruct convDesc, cudnnTensorStruct inputTensorDesc, cudnnFilterStruct filterDesc, IntBuffer n, IntBuffer c, IntBuffer h, IntBuffer w)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolution2dForwardOutputDim(cudnnConvolutionStruct convDesc, cudnnTensorStruct inputTensorDesc, cudnnFilterStruct filterDesc, int[] n, int[] c, int[] h, int[] w)
@Cast(value="cudnnStatus_t") public static int cudnnSetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc, int arrayLength, @Const IntPointer padA, @Const IntPointer filterStrideA, @Const IntPointer dilationA, @Cast(value="cudnnConvolutionMode_t") int mode, @Cast(value="cudnnDataType_t") int computeType)
@Cast(value="cudnnStatus_t") public static int cudnnSetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc, int arrayLength, @Const IntBuffer padA, @Const IntBuffer filterStrideA, @Const IntBuffer dilationA, @Cast(value="cudnnConvolutionMode_t") int mode, @Cast(value="cudnnDataType_t") int computeType)
@Cast(value="cudnnStatus_t") public static int cudnnSetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc, int arrayLength, @Const int[] padA, @Const int[] filterStrideA, @Const int[] dilationA, @Cast(value="cudnnConvolutionMode_t") int mode, @Cast(value="cudnnDataType_t") int computeType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc, int arrayLengthRequested, IntPointer arrayLength, IntPointer padA, IntPointer strideA, IntPointer dilationA, @Cast(value="cudnnConvolutionMode_t*") IntPointer mode, @Cast(value="cudnnDataType_t*") IntPointer computeType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc, int arrayLengthRequested, IntBuffer arrayLength, IntBuffer padA, IntBuffer strideA, IntBuffer dilationA, @Cast(value="cudnnConvolutionMode_t*") IntBuffer mode, @Cast(value="cudnnDataType_t*") IntBuffer computeType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionNdDescriptor(cudnnConvolutionStruct convDesc, int arrayLengthRequested, int[] arrayLength, int[] padA, int[] strideA, int[] dilationA, @Cast(value="cudnnConvolutionMode_t*") int[] mode, @Cast(value="cudnnDataType_t*") int[] computeType)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionNdForwardOutputDim(cudnnConvolutionStruct convDesc, cudnnTensorStruct inputTensorDesc, cudnnFilterStruct filterDesc, int nbDims, IntPointer tensorOuputDimA)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionNdForwardOutputDim(cudnnConvolutionStruct convDesc, cudnnTensorStruct inputTensorDesc, cudnnFilterStruct filterDesc, int nbDims, IntBuffer tensorOuputDimA)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionNdForwardOutputDim(cudnnConvolutionStruct convDesc, cudnnTensorStruct inputTensorDesc, cudnnFilterStruct filterDesc, int nbDims, int[] tensorOuputDimA)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyConvolutionDescriptor(cudnnConvolutionStruct convDesc)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionForwardAlgorithmMaxCount(cudnnContext handle, IntPointer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionForwardAlgorithmMaxCount(cudnnContext handle, IntBuffer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionForwardAlgorithmMaxCount(cudnnContext handle, int[] count)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionForwardAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct yDesc, int requestedAlgoCount, IntPointer returnedAlgoCount, cudnnConvolutionFwdAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionForwardAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct yDesc, int requestedAlgoCount, IntBuffer returnedAlgoCount, cudnnConvolutionFwdAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionForwardAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct yDesc, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionFwdAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionForwardAlgorithmEx(cudnnContext handle, cudnnTensorStruct xDesc, @Const Pointer x, cudnnFilterStruct wDesc, @Const Pointer w, cudnnConvolutionStruct convDesc, cudnnTensorStruct yDesc, Pointer y, int requestedAlgoCount, IntPointer returnedAlgoCount, cudnnConvolutionFwdAlgoPerf_t perfResults, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionForwardAlgorithmEx(cudnnContext handle, cudnnTensorStruct xDesc, @Const Pointer x, cudnnFilterStruct wDesc, @Const Pointer w, cudnnConvolutionStruct convDesc, cudnnTensorStruct yDesc, Pointer y, int requestedAlgoCount, IntBuffer returnedAlgoCount, cudnnConvolutionFwdAlgoPerf_t perfResults, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionForwardAlgorithmEx(cudnnContext handle, cudnnTensorStruct xDesc, @Const Pointer x, cudnnFilterStruct wDesc, @Const Pointer w, cudnnConvolutionStruct convDesc, cudnnTensorStruct yDesc, Pointer y, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionFwdAlgoPerf_t perfResults, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionForwardAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct yDesc, @Cast(value="cudnnConvolutionFwdPreference_t") int preference, @Cast(value="size_t") long memoryLimitInBytes, @Cast(value="cudnnConvolutionFwdAlgo_t*") IntPointer algo)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionForwardAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct yDesc, @Cast(value="cudnnConvolutionFwdPreference_t") int preference, @Cast(value="size_t") long memoryLimitInBytes, @Cast(value="cudnnConvolutionFwdAlgo_t*") IntBuffer algo)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionForwardAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct yDesc, @Cast(value="cudnnConvolutionFwdPreference_t") int preference, @Cast(value="size_t") long memoryLimitInBytes, @Cast(value="cudnnConvolutionFwdAlgo_t*") int[] algo)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionForwardAlgorithm_v7(cudnnContext handle, cudnnTensorStruct srcDesc, cudnnFilterStruct filterDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct destDesc, int requestedAlgoCount, IntPointer returnedAlgoCount, cudnnConvolutionFwdAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionForwardAlgorithm_v7(cudnnContext handle, cudnnTensorStruct srcDesc, cudnnFilterStruct filterDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct destDesc, int requestedAlgoCount, IntBuffer returnedAlgoCount, cudnnConvolutionFwdAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionForwardAlgorithm_v7(cudnnContext handle, cudnnTensorStruct srcDesc, cudnnFilterStruct filterDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct destDesc, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionFwdAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionForwardWorkspaceSize(cudnnContext handle, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct yDesc, @Cast(value="cudnnConvolutionFwdAlgo_t") int algo, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnConvolutionForward(cudnnContext handle, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, cudnnFilterStruct wDesc, @Const Pointer w, cudnnConvolutionStruct convDesc, @Cast(value="cudnnConvolutionFwdAlgo_t") int algo, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes, @Const Pointer beta, cudnnTensorStruct yDesc, Pointer y)
@Cast(value="cudnnStatus_t") public static int cudnnConvolutionBiasActivationForward(cudnnContext handle, @Const Pointer alpha1, cudnnTensorStruct xDesc, @Const Pointer x, cudnnFilterStruct wDesc, @Const Pointer w, cudnnConvolutionStruct convDesc, @Cast(value="cudnnConvolutionFwdAlgo_t") int algo, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes, @Const Pointer alpha2, cudnnTensorStruct zDesc, @Const Pointer z, cudnnTensorStruct biasDesc, @Const Pointer bias, cudnnActivationStruct activationDesc, cudnnTensorStruct yDesc, Pointer y)
@Cast(value="cudnnStatus_t") public static int cudnnConvolutionBackwardBias(cudnnContext handle, @Const Pointer alpha, cudnnTensorStruct dyDesc, @Const Pointer dy, @Const Pointer beta, cudnnTensorStruct dbDesc, Pointer db)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardFilterAlgorithmMaxCount(cudnnContext handle, IntPointer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardFilterAlgorithmMaxCount(cudnnContext handle, IntBuffer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardFilterAlgorithmMaxCount(cudnnContext handle, int[] count)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardFilterAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnFilterStruct dwDesc, int requestedAlgoCount, IntPointer returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardFilterAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnFilterStruct dwDesc, int requestedAlgoCount, IntBuffer returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardFilterAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnFilterStruct dwDesc, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardFilterAlgorithmEx(cudnnContext handle, cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct dyDesc, @Const Pointer y, cudnnConvolutionStruct convDesc, cudnnFilterStruct dwDesc, Pointer dw, int requestedAlgoCount, IntPointer returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf_t perfResults, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardFilterAlgorithmEx(cudnnContext handle, cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct dyDesc, @Const Pointer y, cudnnConvolutionStruct convDesc, cudnnFilterStruct dwDesc, Pointer dw, int requestedAlgoCount, IntBuffer returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf_t perfResults, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardFilterAlgorithmEx(cudnnContext handle, cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct dyDesc, @Const Pointer y, cudnnConvolutionStruct convDesc, cudnnFilterStruct dwDesc, Pointer dw, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf_t perfResults, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardFilterAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnFilterStruct dwDesc, @Cast(value="cudnnConvolutionBwdFilterPreference_t") int preference, @Cast(value="size_t") long memoryLimitInBytes, @Cast(value="cudnnConvolutionBwdFilterAlgo_t*") IntPointer algo)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardFilterAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnFilterStruct dwDesc, @Cast(value="cudnnConvolutionBwdFilterPreference_t") int preference, @Cast(value="size_t") long memoryLimitInBytes, @Cast(value="cudnnConvolutionBwdFilterAlgo_t*") IntBuffer algo)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardFilterAlgorithm(cudnnContext handle, cudnnTensorStruct xDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnFilterStruct dwDesc, @Cast(value="cudnnConvolutionBwdFilterPreference_t") int preference, @Cast(value="size_t") long memoryLimitInBytes, @Cast(value="cudnnConvolutionBwdFilterAlgo_t*") int[] algo)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardFilterAlgorithm_v7(cudnnContext handle, cudnnTensorStruct srcDesc, cudnnTensorStruct diffDesc, cudnnConvolutionStruct convDesc, cudnnFilterStruct gradDesc, int requestedAlgoCount, IntPointer returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardFilterAlgorithm_v7(cudnnContext handle, cudnnTensorStruct srcDesc, cudnnTensorStruct diffDesc, cudnnConvolutionStruct convDesc, cudnnFilterStruct gradDesc, int requestedAlgoCount, IntBuffer returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardFilterAlgorithm_v7(cudnnContext handle, cudnnTensorStruct srcDesc, cudnnTensorStruct diffDesc, cudnnConvolutionStruct convDesc, cudnnFilterStruct gradDesc, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdFilterAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardFilterWorkspaceSize(cudnnContext handle, cudnnTensorStruct xDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnFilterStruct gradDesc, @Cast(value="cudnnConvolutionBwdFilterAlgo_t") int algo, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnConvolutionBackwardFilter(cudnnContext handle, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnConvolutionStruct convDesc, @Cast(value="cudnnConvolutionBwdFilterAlgo_t") int algo, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes, @Const Pointer beta, cudnnFilterStruct dwDesc, Pointer dw)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardDataAlgorithmMaxCount(cudnnContext handle, IntPointer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardDataAlgorithmMaxCount(cudnnContext handle, IntBuffer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardDataAlgorithmMaxCount(cudnnContext handle, int[] count)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardDataAlgorithm(cudnnContext handle, cudnnFilterStruct wDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct dxDesc, int requestedAlgoCount, IntPointer returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardDataAlgorithm(cudnnContext handle, cudnnFilterStruct wDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct dxDesc, int requestedAlgoCount, IntBuffer returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardDataAlgorithm(cudnnContext handle, cudnnFilterStruct wDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct dxDesc, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardDataAlgorithmEx(cudnnContext handle, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnConvolutionStruct convDesc, cudnnTensorStruct dxDesc, Pointer dx, int requestedAlgoCount, IntPointer returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf_t perfResults, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardDataAlgorithmEx(cudnnContext handle, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnConvolutionStruct convDesc, cudnnTensorStruct dxDesc, Pointer dx, int requestedAlgoCount, IntBuffer returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf_t perfResults, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindConvolutionBackwardDataAlgorithmEx(cudnnContext handle, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnConvolutionStruct convDesc, cudnnTensorStruct dxDesc, Pointer dx, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf_t perfResults, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardDataAlgorithm(cudnnContext handle, cudnnFilterStruct wDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct dxDesc, @Cast(value="cudnnConvolutionBwdDataPreference_t") int preference, @Cast(value="size_t") long memoryLimitInBytes, @Cast(value="cudnnConvolutionBwdDataAlgo_t*") IntPointer algo)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardDataAlgorithm(cudnnContext handle, cudnnFilterStruct wDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct dxDesc, @Cast(value="cudnnConvolutionBwdDataPreference_t") int preference, @Cast(value="size_t") long memoryLimitInBytes, @Cast(value="cudnnConvolutionBwdDataAlgo_t*") IntBuffer algo)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardDataAlgorithm(cudnnContext handle, cudnnFilterStruct wDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct dxDesc, @Cast(value="cudnnConvolutionBwdDataPreference_t") int preference, @Cast(value="size_t") long memoryLimitInBytes, @Cast(value="cudnnConvolutionBwdDataAlgo_t*") int[] algo)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardDataAlgorithm_v7(cudnnContext handle, cudnnFilterStruct filterDesc, cudnnTensorStruct diffDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct gradDesc, int requestedAlgoCount, IntPointer returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardDataAlgorithm_v7(cudnnContext handle, cudnnFilterStruct filterDesc, cudnnTensorStruct diffDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct gradDesc, int requestedAlgoCount, IntBuffer returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardDataAlgorithm_v7(cudnnContext handle, cudnnFilterStruct filterDesc, cudnnTensorStruct diffDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct gradDesc, int requestedAlgoCount, int[] returnedAlgoCount, cudnnConvolutionBwdDataAlgoPerf_t perfResults)
@Cast(value="cudnnStatus_t") public static int cudnnGetConvolutionBackwardDataWorkspaceSize(cudnnContext handle, cudnnFilterStruct wDesc, cudnnTensorStruct dyDesc, cudnnConvolutionStruct convDesc, cudnnTensorStruct dxDesc, @Cast(value="cudnnConvolutionBwdDataAlgo_t") int algo, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnConvolutionBackwardData(cudnnContext handle, @Const Pointer alpha, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnConvolutionStruct convDesc, @Cast(value="cudnnConvolutionBwdDataAlgo_t") int algo, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes, @Const Pointer beta, cudnnTensorStruct dxDesc, Pointer dx)
@Cast(value="cudnnStatus_t") public static int cudnnIm2Col(cudnnContext handle, cudnnTensorStruct xDesc, @Const Pointer x, cudnnFilterStruct wDesc, cudnnConvolutionStruct convDesc, Pointer colBuffer)
@Cast(value="cudnnStatus_t") public static int cudnnSoftmaxForward(cudnnContext handle, @Cast(value="cudnnSoftmaxAlgorithm_t") int algo, @Cast(value="cudnnSoftmaxMode_t") int mode, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer beta, cudnnTensorStruct yDesc, Pointer y)
@Cast(value="cudnnStatus_t") public static int cudnnSoftmaxBackward(cudnnContext handle, @Cast(value="cudnnSoftmaxAlgorithm_t") int algo, @Cast(value="cudnnSoftmaxMode_t") int mode, @Const Pointer alpha, cudnnTensorStruct yDesc, @Const Pointer y, cudnnTensorStruct dyDesc, @Const Pointer dy, @Const Pointer beta, cudnnTensorStruct dxDesc, Pointer dx)
@Cast(value="cudnnStatus_t") public static int cudnnCreatePoolingDescriptor(@ByPtrPtr cudnnPoolingStruct poolingDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetPooling2dDescriptor(cudnnPoolingStruct poolingDesc, @Cast(value="cudnnPoolingMode_t") int mode, @Cast(value="cudnnNanPropagation_t") int maxpoolingNanOpt, int windowHeight, int windowWidth, int verticalPadding, int horizontalPadding, int verticalStride, int horizontalStride)
@Cast(value="cudnnStatus_t") public static int cudnnGetPooling2dDescriptor(cudnnPoolingStruct poolingDesc, @Cast(value="cudnnPoolingMode_t*") IntPointer mode, @Cast(value="cudnnNanPropagation_t*") IntPointer maxpoolingNanOpt, IntPointer windowHeight, IntPointer windowWidth, IntPointer verticalPadding, IntPointer horizontalPadding, IntPointer verticalStride, IntPointer horizontalStride)
@Cast(value="cudnnStatus_t") public static int cudnnGetPooling2dDescriptor(cudnnPoolingStruct poolingDesc, @Cast(value="cudnnPoolingMode_t*") IntBuffer mode, @Cast(value="cudnnNanPropagation_t*") IntBuffer maxpoolingNanOpt, IntBuffer windowHeight, IntBuffer windowWidth, IntBuffer verticalPadding, IntBuffer horizontalPadding, IntBuffer verticalStride, IntBuffer horizontalStride)
@Cast(value="cudnnStatus_t") public static int cudnnGetPooling2dDescriptor(cudnnPoolingStruct poolingDesc, @Cast(value="cudnnPoolingMode_t*") int[] mode, @Cast(value="cudnnNanPropagation_t*") int[] maxpoolingNanOpt, int[] windowHeight, int[] windowWidth, int[] verticalPadding, int[] horizontalPadding, int[] verticalStride, int[] horizontalStride)
@Cast(value="cudnnStatus_t") public static int cudnnSetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc, @Cast(value="const cudnnPoolingMode_t") int mode, @Cast(value="const cudnnNanPropagation_t") int maxpoolingNanOpt, int nbDims, @Const IntPointer windowDimA, @Const IntPointer paddingA, @Const IntPointer strideA)
@Cast(value="cudnnStatus_t") public static int cudnnSetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc, @Cast(value="const cudnnPoolingMode_t") int mode, @Cast(value="const cudnnNanPropagation_t") int maxpoolingNanOpt, int nbDims, @Const IntBuffer windowDimA, @Const IntBuffer paddingA, @Const IntBuffer strideA)
@Cast(value="cudnnStatus_t") public static int cudnnSetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc, @Cast(value="const cudnnPoolingMode_t") int mode, @Cast(value="const cudnnNanPropagation_t") int maxpoolingNanOpt, int nbDims, @Const int[] windowDimA, @Const int[] paddingA, @Const int[] strideA)
@Cast(value="cudnnStatus_t") public static int cudnnGetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc, int nbDimsRequested, @Cast(value="cudnnPoolingMode_t*") IntPointer mode, @Cast(value="cudnnNanPropagation_t*") IntPointer maxpoolingNanOpt, IntPointer nbDims, IntPointer windowDimA, IntPointer paddingA, IntPointer strideA)
@Cast(value="cudnnStatus_t") public static int cudnnGetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc, int nbDimsRequested, @Cast(value="cudnnPoolingMode_t*") IntBuffer mode, @Cast(value="cudnnNanPropagation_t*") IntBuffer maxpoolingNanOpt, IntBuffer nbDims, IntBuffer windowDimA, IntBuffer paddingA, IntBuffer strideA)
@Cast(value="cudnnStatus_t") public static int cudnnGetPoolingNdDescriptor(cudnnPoolingStruct poolingDesc, int nbDimsRequested, @Cast(value="cudnnPoolingMode_t*") int[] mode, @Cast(value="cudnnNanPropagation_t*") int[] maxpoolingNanOpt, int[] nbDims, int[] windowDimA, int[] paddingA, int[] strideA)
@Cast(value="cudnnStatus_t") public static int cudnnGetPoolingNdForwardOutputDim(cudnnPoolingStruct poolingDesc, cudnnTensorStruct inputTensorDesc, int nbDims, IntPointer outputTensorDimA)
@Cast(value="cudnnStatus_t") public static int cudnnGetPoolingNdForwardOutputDim(cudnnPoolingStruct poolingDesc, cudnnTensorStruct inputTensorDesc, int nbDims, IntBuffer outputTensorDimA)
@Cast(value="cudnnStatus_t") public static int cudnnGetPoolingNdForwardOutputDim(cudnnPoolingStruct poolingDesc, cudnnTensorStruct inputTensorDesc, int nbDims, int[] outputTensorDimA)
@Cast(value="cudnnStatus_t") public static int cudnnGetPooling2dForwardOutputDim(cudnnPoolingStruct poolingDesc, cudnnTensorStruct inputTensorDesc, IntPointer n, IntPointer c, IntPointer h, IntPointer w)
@Cast(value="cudnnStatus_t") public static int cudnnGetPooling2dForwardOutputDim(cudnnPoolingStruct poolingDesc, cudnnTensorStruct inputTensorDesc, IntBuffer n, IntBuffer c, IntBuffer h, IntBuffer w)
@Cast(value="cudnnStatus_t") public static int cudnnGetPooling2dForwardOutputDim(cudnnPoolingStruct poolingDesc, cudnnTensorStruct inputTensorDesc, int[] n, int[] c, int[] h, int[] w)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyPoolingDescriptor(cudnnPoolingStruct poolingDesc)
@Cast(value="cudnnStatus_t") public static int cudnnPoolingForward(cudnnContext handle, cudnnPoolingStruct poolingDesc, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer beta, cudnnTensorStruct yDesc, Pointer y)
@Cast(value="cudnnStatus_t") public static int cudnnPoolingBackward(cudnnContext handle, cudnnPoolingStruct poolingDesc, @Const Pointer alpha, cudnnTensorStruct yDesc, @Const Pointer y, cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer beta, cudnnTensorStruct dxDesc, Pointer dx)
@Cast(value="cudnnStatus_t") public static int cudnnCreateActivationDescriptor(@ByPtrPtr cudnnActivationStruct activationDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetActivationDescriptor(cudnnActivationStruct activationDesc, @Cast(value="cudnnActivationMode_t") int mode, @Cast(value="cudnnNanPropagation_t") int reluNanOpt, double coef)
@Cast(value="cudnnStatus_t") public static int cudnnGetActivationDescriptor(cudnnActivationStruct activationDesc, @Cast(value="cudnnActivationMode_t*") IntPointer mode, @Cast(value="cudnnNanPropagation_t*") IntPointer reluNanOpt, DoublePointer coef)
@Cast(value="cudnnStatus_t") public static int cudnnGetActivationDescriptor(cudnnActivationStruct activationDesc, @Cast(value="cudnnActivationMode_t*") IntBuffer mode, @Cast(value="cudnnNanPropagation_t*") IntBuffer reluNanOpt, DoubleBuffer coef)
@Cast(value="cudnnStatus_t") public static int cudnnGetActivationDescriptor(cudnnActivationStruct activationDesc, @Cast(value="cudnnActivationMode_t*") int[] mode, @Cast(value="cudnnNanPropagation_t*") int[] reluNanOpt, double[] coef)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyActivationDescriptor(cudnnActivationStruct activationDesc)
@Cast(value="cudnnStatus_t") public static int cudnnActivationForward(cudnnContext handle, cudnnActivationStruct activationDesc, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer beta, cudnnTensorStruct yDesc, Pointer y)
@Cast(value="cudnnStatus_t") public static int cudnnActivationBackward(cudnnContext handle, cudnnActivationStruct activationDesc, @Const Pointer alpha, cudnnTensorStruct yDesc, @Const Pointer y, cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer beta, cudnnTensorStruct dxDesc, Pointer dx)
@Cast(value="cudnnStatus_t") public static int cudnnCreateLRNDescriptor(@ByPtrPtr cudnnLRNStruct normDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetLRNDescriptor(cudnnLRNStruct normDesc, @Cast(value="unsigned") int lrnN, double lrnAlpha, double lrnBeta, double lrnK)
@Cast(value="cudnnStatus_t") public static int cudnnGetLRNDescriptor(cudnnLRNStruct normDesc, @Cast(value="unsigned*") IntPointer lrnN, DoublePointer lrnAlpha, DoublePointer lrnBeta, DoublePointer lrnK)
@Cast(value="cudnnStatus_t") public static int cudnnGetLRNDescriptor(cudnnLRNStruct normDesc, @Cast(value="unsigned*") IntBuffer lrnN, DoubleBuffer lrnAlpha, DoubleBuffer lrnBeta, DoubleBuffer lrnK)
@Cast(value="cudnnStatus_t") public static int cudnnGetLRNDescriptor(cudnnLRNStruct normDesc, @Cast(value="unsigned*") int[] lrnN, double[] lrnAlpha, double[] lrnBeta, double[] lrnK)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyLRNDescriptor(cudnnLRNStruct lrnDesc)
@Cast(value="cudnnStatus_t") public static int cudnnLRNCrossChannelForward(cudnnContext handle, cudnnLRNStruct normDesc, @Cast(value="cudnnLRNMode_t") int lrnMode, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer beta, cudnnTensorStruct yDesc, Pointer y)
@Cast(value="cudnnStatus_t") public static int cudnnLRNCrossChannelBackward(cudnnContext handle, cudnnLRNStruct normDesc, @Cast(value="cudnnLRNMode_t") int lrnMode, @Const Pointer alpha, cudnnTensorStruct yDesc, @Const Pointer y, cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer beta, cudnnTensorStruct dxDesc, Pointer dx)
@Cast(value="cudnnStatus_t") public static int cudnnDivisiveNormalizationForward(cudnnContext handle, cudnnLRNStruct normDesc, @Cast(value="cudnnDivNormMode_t") int mode, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer means, Pointer temp, Pointer temp2, @Const Pointer beta, cudnnTensorStruct yDesc, Pointer y)
@Cast(value="cudnnStatus_t") public static int cudnnDivisiveNormalizationBackward(cudnnContext handle, cudnnLRNStruct normDesc, @Cast(value="cudnnDivNormMode_t") int mode, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer means, @Const Pointer dy, Pointer temp, Pointer temp2, @Const Pointer beta, cudnnTensorStruct dXdMeansDesc, Pointer dx, Pointer dMeans)
@Cast(value="cudnnStatus_t") public static int cudnnDeriveBNTensorDescriptor(cudnnTensorStruct derivedBnDesc, cudnnTensorStruct xDesc, @Cast(value="cudnnBatchNormMode_t") int mode)
@Cast(value="cudnnStatus_t") public static int cudnnGetBatchNormalizationForwardTrainingExWorkspaceSize(cudnnContext handle, @Cast(value="cudnnBatchNormMode_t") int mode, @Cast(value="cudnnBatchNormOps_t") int bnOps, cudnnTensorStruct xDesc, cudnnTensorStruct zDesc, cudnnTensorStruct yDesc, cudnnTensorStruct bnScaleBiasMeanVarDesc, cudnnActivationStruct activationDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetBatchNormalizationBackwardExWorkspaceSize(cudnnContext handle, @Cast(value="cudnnBatchNormMode_t") int mode, @Cast(value="cudnnBatchNormOps_t") int bnOps, cudnnTensorStruct xDesc, cudnnTensorStruct yDesc, cudnnTensorStruct dyDesc, cudnnTensorStruct dzDesc, cudnnTensorStruct dxDesc, cudnnTensorStruct dBnScaleBiasDesc, cudnnActivationStruct activationDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetBatchNormalizationTrainingExReserveSpaceSize(cudnnContext handle, @Cast(value="cudnnBatchNormMode_t") int mode, @Cast(value="cudnnBatchNormOps_t") int bnOps, cudnnActivationStruct activationDesc, cudnnTensorStruct xDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnBatchNormalizationForwardTraining(cudnnContext handle, @Cast(value="cudnnBatchNormMode_t") int mode, @Const Pointer alpha, @Const Pointer beta, cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct yDesc, Pointer y, cudnnTensorStruct bnScaleBiasMeanVarDesc, @Const Pointer bnScale, @Const Pointer bnBias, double exponentialAverageFactor, Pointer resultRunningMean, Pointer resultRunningVariance, double epsilon, Pointer resultSaveMean, Pointer resultSaveInvVariance)
@Cast(value="cudnnStatus_t") public static int cudnnBatchNormalizationForwardTrainingEx(cudnnContext handle, @Cast(value="cudnnBatchNormMode_t") int mode, @Cast(value="cudnnBatchNormOps_t") int bnOps, @Const Pointer alpha, @Const Pointer beta, cudnnTensorStruct xDesc, @Const Pointer xData, cudnnTensorStruct zDesc, @Const Pointer zData, cudnnTensorStruct yDesc, Pointer yData, cudnnTensorStruct bnScaleBiasMeanVarDesc, @Const Pointer bnScale, @Const Pointer bnBias, double exponentialAverageFactor, Pointer resultRunningMean, Pointer resultRunningVariance, double epsilon, Pointer resultSaveMean, Pointer resultSaveInvVariance, cudnnActivationStruct activationDesc, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnBatchNormalizationForwardInference(cudnnContext handle, @Cast(value="cudnnBatchNormMode_t") int mode, @Const Pointer alpha, @Const Pointer beta, cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct yDesc, Pointer y, cudnnTensorStruct bnScaleBiasMeanVarDesc, @Const Pointer bnScale, @Const Pointer bnBias, @Const Pointer estimatedMean, @Const Pointer estimatedVariance, double epsilon)
@Cast(value="cudnnStatus_t") public static int cudnnBatchNormalizationBackward(cudnnContext handle, @Cast(value="cudnnBatchNormMode_t") int mode, @Const Pointer alphaDataDiff, @Const Pointer betaDataDiff, @Const Pointer alphaParamDiff, @Const Pointer betaParamDiff, cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnTensorStruct dxDesc, Pointer dx, cudnnTensorStruct dBnScaleBiasDesc, @Const Pointer bnScale, Pointer dBnScaleResult, Pointer dBnBiasResult, double epsilon, @Const Pointer savedMean, @Const Pointer savedInvVariance)
@Cast(value="cudnnStatus_t") public static int cudnnBatchNormalizationBackwardEx(cudnnContext handle, @Cast(value="cudnnBatchNormMode_t") int mode, @Cast(value="cudnnBatchNormOps_t") int bnOps, @Const Pointer alphaDataDiff, @Const Pointer betaDataDiff, @Const Pointer alphaParamDiff, @Const Pointer betaParamDiff, cudnnTensorStruct xDesc, @Const Pointer xData, cudnnTensorStruct yDesc, @Const Pointer yData, cudnnTensorStruct dyDesc, @Const Pointer dyData, cudnnTensorStruct dzDesc, Pointer dzData, cudnnTensorStruct dxDesc, Pointer dxData, cudnnTensorStruct dBnScaleBiasDesc, @Const Pointer bnScaleData, @Const Pointer bnBiasData, Pointer dBnScaleData, Pointer dBnBiasData, double epsilon, @Const Pointer savedMean, @Const Pointer savedInvVariance, cudnnActivationStruct activationDesc, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnCreateSpatialTransformerDescriptor(@ByPtrPtr cudnnSpatialTransformerStruct stDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetSpatialTransformerNdDescriptor(cudnnSpatialTransformerStruct stDesc, @Cast(value="cudnnSamplerType_t") int samplerType, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const IntPointer dimA)
@Cast(value="cudnnStatus_t") public static int cudnnSetSpatialTransformerNdDescriptor(cudnnSpatialTransformerStruct stDesc, @Cast(value="cudnnSamplerType_t") int samplerType, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const IntBuffer dimA)
@Cast(value="cudnnStatus_t") public static int cudnnSetSpatialTransformerNdDescriptor(cudnnSpatialTransformerStruct stDesc, @Cast(value="cudnnSamplerType_t") int samplerType, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const int[] dimA)
@Cast(value="cudnnStatus_t") public static int cudnnDestroySpatialTransformerDescriptor(cudnnSpatialTransformerStruct stDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSpatialTfGridGeneratorForward(cudnnContext handle, cudnnSpatialTransformerStruct stDesc, @Const Pointer theta, Pointer grid)
@Cast(value="cudnnStatus_t") public static int cudnnSpatialTfGridGeneratorBackward(cudnnContext handle, cudnnSpatialTransformerStruct stDesc, @Const Pointer dgrid, Pointer dtheta)
@Cast(value="cudnnStatus_t") public static int cudnnSpatialTfSamplerForward(cudnnContext handle, cudnnSpatialTransformerStruct stDesc, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer grid, @Const Pointer beta, cudnnTensorStruct yDesc, Pointer y)
@Cast(value="cudnnStatus_t") public static int cudnnSpatialTfSamplerBackward(cudnnContext handle, cudnnSpatialTransformerStruct stDesc, @Const Pointer alpha, cudnnTensorStruct xDesc, @Const Pointer x, @Const Pointer beta, cudnnTensorStruct dxDesc, Pointer dx, @Const Pointer alphaDgrid, cudnnTensorStruct dyDesc, @Const Pointer dy, @Const Pointer grid, @Const Pointer betaDgrid, Pointer dgrid)
@Cast(value="cudnnStatus_t") public static int cudnnCreateDropoutDescriptor(@ByPtrPtr cudnnDropoutStruct dropoutDesc)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyDropoutDescriptor(cudnnDropoutStruct dropoutDesc)
@Cast(value="cudnnStatus_t") public static int cudnnDropoutGetStatesSize(cudnnContext handle, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnDropoutGetReserveSpaceSize(cudnnTensorStruct xdesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnSetDropoutDescriptor(cudnnDropoutStruct dropoutDesc, cudnnContext handle, float dropout, Pointer states, @Cast(value="size_t") long stateSizeInBytes, @Cast(value="unsigned long long") long seed)
@Cast(value="cudnnStatus_t") public static int cudnnRestoreDropoutDescriptor(cudnnDropoutStruct dropoutDesc, cudnnContext handle, float dropout, Pointer states, @Cast(value="size_t") long stateSizeInBytes, @Cast(value="unsigned long long") long seed)
@Cast(value="cudnnStatus_t") public static int cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc, cudnnContext handle, FloatPointer dropout, @Cast(value="void**") PointerPointer states, @Cast(value="unsigned long long*") LongPointer seed)
@Cast(value="cudnnStatus_t") public static int cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc, cudnnContext handle, FloatPointer dropout, @Cast(value="void**") @ByPtrPtr Pointer states, @Cast(value="unsigned long long*") LongPointer seed)
@Cast(value="cudnnStatus_t") public static int cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc, cudnnContext handle, FloatBuffer dropout, @Cast(value="void**") @ByPtrPtr Pointer states, @Cast(value="unsigned long long*") LongBuffer seed)
@Cast(value="cudnnStatus_t") public static int cudnnGetDropoutDescriptor(cudnnDropoutStruct dropoutDesc, cudnnContext handle, float[] dropout, @Cast(value="void**") @ByPtrPtr Pointer states, @Cast(value="unsigned long long*") long[] seed)
@Cast(value="cudnnStatus_t") public static int cudnnDropoutForward(cudnnContext handle, cudnnDropoutStruct dropoutDesc, cudnnTensorStruct xdesc, @Const Pointer x, cudnnTensorStruct ydesc, Pointer y, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnDropoutBackward(cudnnContext handle, cudnnDropoutStruct dropoutDesc, cudnnTensorStruct dydesc, @Const Pointer dy, cudnnTensorStruct dxdesc, Pointer dx, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnCreateRNNDescriptor(@ByPtrPtr cudnnRNNStruct rnnDesc)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyRNNDescriptor(cudnnRNNStruct rnnDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNDescriptor(cudnnContext handle, cudnnRNNStruct rnnDesc, int hiddenSize, int numLayers, cudnnDropoutStruct dropoutDesc, @Cast(value="cudnnRNNInputMode_t") int inputMode, @Cast(value="cudnnDirectionMode_t") int direction, @Cast(value="cudnnRNNMode_t") int mode, @Cast(value="cudnnRNNAlgo_t") int algo, @Cast(value="cudnnDataType_t") int mathPrec)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNDescriptor(cudnnContext handle, cudnnRNNStruct rnnDesc, IntPointer hiddenSize, IntPointer numLayers, @ByPtrPtr cudnnDropoutStruct dropoutDesc, @Cast(value="cudnnRNNInputMode_t*") IntPointer inputMode, @Cast(value="cudnnDirectionMode_t*") IntPointer direction, @Cast(value="cudnnRNNMode_t*") IntPointer mode, @Cast(value="cudnnRNNAlgo_t*") IntPointer algo, @Cast(value="cudnnDataType_t*") IntPointer mathPrec)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNDescriptor(cudnnContext handle, cudnnRNNStruct rnnDesc, IntBuffer hiddenSize, IntBuffer numLayers, @ByPtrPtr cudnnDropoutStruct dropoutDesc, @Cast(value="cudnnRNNInputMode_t*") IntBuffer inputMode, @Cast(value="cudnnDirectionMode_t*") IntBuffer direction, @Cast(value="cudnnRNNMode_t*") IntBuffer mode, @Cast(value="cudnnRNNAlgo_t*") IntBuffer algo, @Cast(value="cudnnDataType_t*") IntBuffer mathPrec)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNDescriptor(cudnnContext handle, cudnnRNNStruct rnnDesc, int[] hiddenSize, int[] numLayers, @ByPtrPtr cudnnDropoutStruct dropoutDesc, @Cast(value="cudnnRNNInputMode_t*") int[] inputMode, @Cast(value="cudnnDirectionMode_t*") int[] direction, @Cast(value="cudnnRNNMode_t*") int[] mode, @Cast(value="cudnnRNNAlgo_t*") int[] algo, @Cast(value="cudnnDataType_t*") int[] mathPrec)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNMatrixMathType(cudnnRNNStruct rnnDesc, @Cast(value="cudnnMathType_t") int mType)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNMatrixMathType(cudnnRNNStruct rnnDesc, @Cast(value="cudnnMathType_t*") IntPointer mType)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNMatrixMathType(cudnnRNNStruct rnnDesc, @Cast(value="cudnnMathType_t*") IntBuffer mType)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNMatrixMathType(cudnnRNNStruct rnnDesc, @Cast(value="cudnnMathType_t*") int[] mType)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNBiasMode(cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNBiasMode_t") int biasMode)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNBiasMode(cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNBiasMode_t*") IntPointer biasMode)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNBiasMode(cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNBiasMode_t*") IntBuffer biasMode)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNBiasMode(cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNBiasMode_t*") int[] biasMode)
@Cast(value="cudnnStatus_t") public static int cudnnRNNSetClip(cudnnContext handle, cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNClipMode_t") int clipMode, @Cast(value="cudnnNanPropagation_t") int clipNanOpt, double lclip, double rclip)
@Cast(value="cudnnStatus_t") public static int cudnnRNNGetClip(cudnnContext handle, cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNClipMode_t*") IntPointer clipMode, @Cast(value="cudnnNanPropagation_t*") IntPointer clipNanOpt, DoublePointer lclip, DoublePointer rclip)
@Cast(value="cudnnStatus_t") public static int cudnnRNNGetClip(cudnnContext handle, cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNClipMode_t*") IntBuffer clipMode, @Cast(value="cudnnNanPropagation_t*") IntBuffer clipNanOpt, DoubleBuffer lclip, DoubleBuffer rclip)
@Cast(value="cudnnStatus_t") public static int cudnnRNNGetClip(cudnnContext handle, cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNClipMode_t*") int[] clipMode, @Cast(value="cudnnNanPropagation_t*") int[] clipNanOpt, double[] lclip, double[] rclip)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNProjectionLayers(cudnnContext handle, cudnnRNNStruct rnnDesc, int recProjSize, int outProjSize)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNProjectionLayers(cudnnContext handle, cudnnRNNStruct rnnDesc, IntPointer recProjSize, IntPointer outProjSize)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNProjectionLayers(cudnnContext handle, cudnnRNNStruct rnnDesc, IntBuffer recProjSize, IntBuffer outProjSize)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNProjectionLayers(cudnnContext handle, cudnnRNNStruct rnnDesc, int[] recProjSize, int[] outProjSize)
@Cast(value="cudnnStatus_t") public static int cudnnCreatePersistentRNNPlan(cudnnRNNStruct rnnDesc, int minibatch, @Cast(value="const cudnnDataType_t") int dataType, @ByPtrPtr cudnnPersistentRNNPlan plan)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyPersistentRNNPlan(cudnnPersistentRNNPlan plan)
@Cast(value="cudnnStatus_t") public static int cudnnSetPersistentRNNPlan(cudnnRNNStruct rnnDesc, cudnnPersistentRNNPlan plan)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNWorkspaceSize(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNWorkspaceSize(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNTrainingReserveSize(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNTrainingReserveSize(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNParamsSize(cudnnContext handle, cudnnRNNStruct rnnDesc, cudnnTensorStruct xDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes, @Cast(value="cudnnDataType_t") int dataType)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNLinLayerMatrixParams(cudnnContext handle, cudnnRNNStruct rnnDesc, int pseudoLayer, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, @Const Pointer w, int linLayerID, cudnnFilterStruct linLayerMatDesc, @Cast(value="void**") PointerPointer linLayerMat)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNLinLayerMatrixParams(cudnnContext handle, cudnnRNNStruct rnnDesc, int pseudoLayer, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, @Const Pointer w, int linLayerID, cudnnFilterStruct linLayerMatDesc, @Cast(value="void**") @ByPtrPtr Pointer linLayerMat)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNLinLayerBiasParams(cudnnContext handle, cudnnRNNStruct rnnDesc, int pseudoLayer, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, @Const Pointer w, int linLayerID, cudnnFilterStruct linLayerBiasDesc, @Cast(value="void**") PointerPointer linLayerBias)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNLinLayerBiasParams(cudnnContext handle, cudnnRNNStruct rnnDesc, int pseudoLayer, cudnnTensorStruct xDesc, cudnnFilterStruct wDesc, @Const Pointer w, int linLayerID, cudnnFilterStruct linLayerBiasDesc, @Cast(value="void**") @ByPtrPtr Pointer linLayerBias)
@Cast(value="cudnnStatus_t") public static int cudnnRNNForwardInference(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRNNForwardInference(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRNNForwardTraining(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRNNForwardTraining(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRNNBackwardData(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, @Const Pointer y, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnTensorStruct dhyDesc, @Const Pointer dhy, cudnnTensorStruct dcyDesc, @Const Pointer dcy, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct dxDesc, Pointer dx, cudnnTensorStruct dhxDesc, Pointer dhx, cudnnTensorStruct dcxDesc, Pointer dcx, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRNNBackwardData(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, @Const Pointer y, @Cast(value="cudnnTensorStruct**") PointerPointer dyDesc, @Const Pointer dy, cudnnTensorStruct dhyDesc, @Const Pointer dhy, cudnnTensorStruct dcyDesc, @Const Pointer dcy, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, @Cast(value="cudnnTensorStruct**") PointerPointer dxDesc, Pointer dx, cudnnTensorStruct dhxDesc, Pointer dhx, cudnnTensorStruct dcxDesc, Pointer dcx, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRNNBackwardWeights(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, @Const Pointer y, @Const Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, cudnnFilterStruct dwDesc, Pointer dw, @Const Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRNNBackwardWeights(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, @Const Pointer y, @Const Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, cudnnFilterStruct dwDesc, Pointer dw, @Const Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNPaddingMode(cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNPaddingMode_t") int paddingMode)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNPaddingMode(cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNPaddingMode_t*") IntPointer paddingMode)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNPaddingMode(cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNPaddingMode_t*") IntBuffer paddingMode)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNPaddingMode(cudnnRNNStruct rnnDesc, @Cast(value="cudnnRNNPaddingMode_t*") int[] paddingMode)
@Cast(value="cudnnStatus_t") public static int cudnnCreateRNNDataDescriptor(@ByPtrPtr cudnnRNNDataStruct rnnDataDesc)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc, @Cast(value="cudnnDataType_t") int dataType, @Cast(value="cudnnRNNDataLayout_t") int layout, int maxSeqLength, int batchSize, int vectorSize, @Const IntPointer seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc, @Cast(value="cudnnDataType_t") int dataType, @Cast(value="cudnnRNNDataLayout_t") int layout, int maxSeqLength, int batchSize, int vectorSize, @Const IntBuffer seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc, @Cast(value="cudnnDataType_t") int dataType, @Cast(value="cudnnRNNDataLayout_t") int layout, int maxSeqLength, int batchSize, int vectorSize, @Const int[] seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc, @Cast(value="cudnnDataType_t*") IntPointer dataType, @Cast(value="cudnnRNNDataLayout_t*") IntPointer layout, IntPointer maxSeqLength, IntPointer batchSize, IntPointer vectorSize, int arrayLengthRequested, IntPointer seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc, @Cast(value="cudnnDataType_t*") IntBuffer dataType, @Cast(value="cudnnRNNDataLayout_t*") IntBuffer layout, IntBuffer maxSeqLength, IntBuffer batchSize, IntBuffer vectorSize, int arrayLengthRequested, IntBuffer seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNDataDescriptor(cudnnRNNDataStruct rnnDataDesc, @Cast(value="cudnnDataType_t*") int[] dataType, @Cast(value="cudnnRNNDataLayout_t*") int[] layout, int[] maxSeqLength, int[] batchSize, int[] vectorSize, int arrayLengthRequested, int[] seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnRNNForwardTrainingEx(cudnnContext handle, cudnnRNNStruct rnnDesc, cudnnRNNDataStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, cudnnRNNDataStruct yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, cudnnRNNDataStruct kDesc, @Const Pointer keys, cudnnRNNDataStruct cDesc, Pointer cAttn, cudnnRNNDataStruct iDesc, Pointer iAttn, cudnnRNNDataStruct qDesc, Pointer queries, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRNNForwardInferenceEx(cudnnContext handle, cudnnRNNStruct rnnDesc, cudnnRNNDataStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, cudnnRNNDataStruct yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, cudnnRNNDataStruct kDesc, @Const Pointer keys, cudnnRNNDataStruct cDesc, Pointer cAttn, cudnnRNNDataStruct iDesc, Pointer iAttn, cudnnRNNDataStruct qDesc, Pointer queries, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRNNBackwardDataEx(cudnnContext handle, cudnnRNNStruct rnnDesc, cudnnRNNDataStruct yDesc, @Const Pointer y, cudnnRNNDataStruct dyDesc, @Const Pointer dy, cudnnRNNDataStruct dcDesc, @Const Pointer dcAttn, cudnnTensorStruct dhyDesc, @Const Pointer dhy, cudnnTensorStruct dcyDesc, @Const Pointer dcy, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnRNNDataStruct dxDesc, Pointer dx, cudnnTensorStruct dhxDesc, Pointer dhx, cudnnTensorStruct dcxDesc, Pointer dcx, cudnnRNNDataStruct dkDesc, Pointer dkeys, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRNNBackwardWeightsEx(cudnnContext handle, cudnnRNNStruct rnnDesc, cudnnRNNDataStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnRNNDataStruct yDesc, @Const Pointer y, Pointer workSpace, @Cast(value="size_t") long workSpaceSizeInBytes, cudnnFilterStruct dwDesc, Pointer dw, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNAlgorithmDescriptor(cudnnContext handle, cudnnRNNStruct rnnDesc, cudnnAlgorithmStruct algoDesc)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNForwardInferenceAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, IntPointer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNForwardInferenceAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, IntBuffer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNForwardInferenceAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, int[] count)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, IntPointer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, IntBuffer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, int[] returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, IntPointer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, IntBuffer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardInferenceAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, int[] returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNForwardTrainingAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, IntPointer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNForwardTrainingAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, IntBuffer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNForwardTrainingAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, int[] count)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, IntPointer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, IntBuffer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, int[] returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, IntPointer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, IntBuffer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNForwardTrainingAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, cudnnFilterStruct wDesc, @Const Pointer w, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, Pointer y, cudnnTensorStruct hyDesc, Pointer hy, cudnnTensorStruct cyDesc, Pointer cy, float findIntensity, int requestedAlgoCount, int[] returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNBackwardDataAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, IntPointer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNBackwardDataAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, IntBuffer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNBackwardDataAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, int[] count)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, @Const Pointer y, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnTensorStruct dhyDesc, @Const Pointer dhy, cudnnTensorStruct dcyDesc, @Const Pointer dcy, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct dxDesc, Pointer dx, cudnnTensorStruct dhxDesc, Pointer dhx, cudnnTensorStruct dcxDesc, Pointer dcx, float findIntensity, int requestedAlgoCount, IntPointer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, @Const Pointer y, @Cast(value="cudnnTensorStruct**") PointerPointer dyDesc, @Const Pointer dy, cudnnTensorStruct dhyDesc, @Const Pointer dhy, cudnnTensorStruct dcyDesc, @Const Pointer dcy, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, @Cast(value="cudnnTensorStruct**") PointerPointer dxDesc, Pointer dx, cudnnTensorStruct dhxDesc, Pointer dhx, cudnnTensorStruct dcxDesc, Pointer dcx, float findIntensity, int requestedAlgoCount, IntBuffer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, @Const Pointer y, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnTensorStruct dhyDesc, @Const Pointer dhy, cudnnTensorStruct dcyDesc, @Const Pointer dcy, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct dxDesc, Pointer dx, cudnnTensorStruct dhxDesc, Pointer dhx, cudnnTensorStruct dcxDesc, Pointer dcx, float findIntensity, int requestedAlgoCount, int[] returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, @Const Pointer y, @Cast(value="cudnnTensorStruct**") PointerPointer dyDesc, @Const Pointer dy, cudnnTensorStruct dhyDesc, @Const Pointer dhy, cudnnTensorStruct dcyDesc, @Const Pointer dcy, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, @Cast(value="cudnnTensorStruct**") PointerPointer dxDesc, Pointer dx, cudnnTensorStruct dhxDesc, Pointer dhx, cudnnTensorStruct dcxDesc, Pointer dcx, float findIntensity, int requestedAlgoCount, IntPointer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, @Const Pointer y, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct dyDesc, @Const Pointer dy, cudnnTensorStruct dhyDesc, @Const Pointer dhy, cudnnTensorStruct dcyDesc, @Const Pointer dcy, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct dxDesc, Pointer dx, cudnnTensorStruct dhxDesc, Pointer dhx, cudnnTensorStruct dcxDesc, Pointer dcx, float findIntensity, int requestedAlgoCount, IntBuffer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardDataAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, @Const Pointer y, @Cast(value="cudnnTensorStruct**") PointerPointer dyDesc, @Const Pointer dy, cudnnTensorStruct dhyDesc, @Const Pointer dhy, cudnnTensorStruct dcyDesc, @Const Pointer dcy, cudnnFilterStruct wDesc, @Const Pointer w, cudnnTensorStruct hxDesc, @Const Pointer hx, cudnnTensorStruct cxDesc, @Const Pointer cx, @Cast(value="cudnnTensorStruct**") PointerPointer dxDesc, Pointer dx, cudnnTensorStruct dhxDesc, Pointer dhx, cudnnTensorStruct dcxDesc, Pointer dcx, float findIntensity, int requestedAlgoCount, int[] returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNBackwardWeightsAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, IntPointer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNBackwardWeightsAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, IntBuffer count)
@Cast(value="cudnnStatus_t") public static int cudnnGetRNNBackwardWeightsAlgorithmMaxCount(cudnnContext handle, cudnnRNNStruct rnnDesc, int[] count)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, @Const Pointer y, float findIntensity, int requestedAlgoCount, IntPointer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, @Const Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, cudnnFilterStruct dwDesc, Pointer dw, @Const Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, @Const Pointer y, float findIntensity, int requestedAlgoCount, IntBuffer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, @Const Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, cudnnFilterStruct dwDesc, Pointer dw, @Const Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, @Const Pointer y, float findIntensity, int requestedAlgoCount, int[] returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, @Const Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, cudnnFilterStruct dwDesc, Pointer dw, @Const Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, @Const Pointer y, float findIntensity, int requestedAlgoCount, IntPointer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, @Const Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, cudnnFilterStruct dwDesc, Pointer dw, @Const Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, @Cast(value="cudnnTensorStruct**") @ByPtrPtr cudnnTensorStruct yDesc, @Const Pointer y, float findIntensity, int requestedAlgoCount, IntBuffer returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, @Const Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, cudnnFilterStruct dwDesc, Pointer dw, @Const Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnFindRNNBackwardWeightsAlgorithmEx(cudnnContext handle, cudnnRNNStruct rnnDesc, int seqLength, @Cast(value="cudnnTensorStruct**") PointerPointer xDesc, @Const Pointer x, cudnnTensorStruct hxDesc, @Const Pointer hx, @Cast(value="cudnnTensorStruct**") PointerPointer yDesc, @Const Pointer y, float findIntensity, int requestedAlgoCount, int[] returnedAlgoCount, @ByPtrPtr cudnnAlgorithmPerformanceStruct perfResults, @Const Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes, cudnnFilterStruct dwDesc, Pointer dw, @Const Pointer reserveSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnCreateSeqDataDescriptor(@ByPtrPtr cudnnSeqDataStruct seqDataDesc)
@Cast(value="cudnnStatus_t") public static int cudnnDestroySeqDataDescriptor(cudnnSeqDataStruct seqDataDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const IntPointer dimA, @Cast(value="const cudnnSeqDataAxis_t*") IntPointer axes, @Cast(value="size_t") long seqLengthArraySize, @Const IntPointer seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnSetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const IntBuffer dimA, @Cast(value="const cudnnSeqDataAxis_t*") IntBuffer axes, @Cast(value="size_t") long seqLengthArraySize, @Const IntBuffer seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnSetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc, @Cast(value="cudnnDataType_t") int dataType, int nbDims, @Const int[] dimA, @Cast(value="const cudnnSeqDataAxis_t*") int[] axes, @Cast(value="size_t") long seqLengthArraySize, @Const int[] seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnGetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc, @Cast(value="cudnnDataType_t*") IntPointer dataType, IntPointer nbDims, int nbDimsRequested, IntPointer dimA, @Cast(value="cudnnSeqDataAxis_t*") IntPointer axes, @Cast(value="size_t*") SizeTPointer seqLengthArraySize, @Cast(value="size_t") long seqLengthSizeRequsted, IntPointer seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnGetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc, @Cast(value="cudnnDataType_t*") IntBuffer dataType, IntBuffer nbDims, int nbDimsRequested, IntBuffer dimA, @Cast(value="cudnnSeqDataAxis_t*") IntBuffer axes, @Cast(value="size_t*") SizeTPointer seqLengthArraySize, @Cast(value="size_t") long seqLengthSizeRequsted, IntBuffer seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnGetSeqDataDescriptor(cudnnSeqDataStruct seqDataDesc, @Cast(value="cudnnDataType_t*") int[] dataType, int[] nbDims, int nbDimsRequested, int[] dimA, @Cast(value="cudnnSeqDataAxis_t*") int[] axes, @Cast(value="size_t*") SizeTPointer seqLengthArraySize, @Cast(value="size_t") long seqLengthSizeRequsted, int[] seqLengthArray, Pointer paddingFill)
@Cast(value="cudnnStatus_t") public static int cudnnCreateAttnDescriptor(@ByPtrPtr cudnnAttnStruct attnDesc)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyAttnDescriptor(cudnnAttnStruct attnDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetAttnDescriptor(cudnnAttnStruct attnDesc, @Cast(value="cudnnAttnQueryMap_t") int queryMap, int nHeads, double smScaler, @Cast(value="cudnnDataType_t") int dataType, @Cast(value="cudnnDataType_t") int computePrec, @Cast(value="cudnnMathType_t") int mathType, cudnnDropoutStruct attnDropoutDesc, cudnnDropoutStruct postDropoutDesc, int qSize, int kSize, int vSize, int qProjSize, int kProjSize, int vProjSize, int oProjSize, int qoMaxSeqLength, int kvMaxSeqLength, int maxBatchSize, int maxBeamSize)
@Cast(value="cudnnStatus_t") public static int cudnnGetAttnDescriptor(cudnnAttnStruct attnDesc, @Cast(value="cudnnAttnQueryMap_t*") IntPointer queryMap, IntPointer nHeads, DoublePointer smScaler, @Cast(value="cudnnDataType_t*") IntPointer dataType, @Cast(value="cudnnDataType_t*") IntPointer computePrec, @Cast(value="cudnnMathType_t*") IntPointer mathType, @ByPtrPtr cudnnDropoutStruct attnDropoutDesc, @ByPtrPtr cudnnDropoutStruct postDropoutDesc, IntPointer qSize, IntPointer kSize, IntPointer vSize, IntPointer qProjSize, IntPointer kProjSize, IntPointer vProjSize, IntPointer oProjSize, IntPointer qoMaxSeqLength, IntPointer kvMaxSeqLength, IntPointer maxBatchSize, IntPointer maxBeamSize)
@Cast(value="cudnnStatus_t") public static int cudnnGetAttnDescriptor(cudnnAttnStruct attnDesc, @Cast(value="cudnnAttnQueryMap_t*") IntBuffer queryMap, IntBuffer nHeads, DoubleBuffer smScaler, @Cast(value="cudnnDataType_t*") IntBuffer dataType, @Cast(value="cudnnDataType_t*") IntBuffer computePrec, @Cast(value="cudnnMathType_t*") IntBuffer mathType, @ByPtrPtr cudnnDropoutStruct attnDropoutDesc, @ByPtrPtr cudnnDropoutStruct postDropoutDesc, IntBuffer qSize, IntBuffer kSize, IntBuffer vSize, IntBuffer qProjSize, IntBuffer kProjSize, IntBuffer vProjSize, IntBuffer oProjSize, IntBuffer qoMaxSeqLength, IntBuffer kvMaxSeqLength, IntBuffer maxBatchSize, IntBuffer maxBeamSize)
@Cast(value="cudnnStatus_t") public static int cudnnGetAttnDescriptor(cudnnAttnStruct attnDesc, @Cast(value="cudnnAttnQueryMap_t*") int[] queryMap, int[] nHeads, double[] smScaler, @Cast(value="cudnnDataType_t*") int[] dataType, @Cast(value="cudnnDataType_t*") int[] computePrec, @Cast(value="cudnnMathType_t*") int[] mathType, @ByPtrPtr cudnnDropoutStruct attnDropoutDesc, @ByPtrPtr cudnnDropoutStruct postDropoutDesc, int[] qSize, int[] kSize, int[] vSize, int[] qProjSize, int[] kProjSize, int[] vProjSize, int[] oProjSize, int[] qoMaxSeqLength, int[] kvMaxSeqLength, int[] maxBatchSize, int[] maxBeamSize)
@Cast(value="cudnnStatus_t") public static int cudnnGetMultiHeadAttnBuffers(cudnnContext handle, cudnnAttnStruct attnDesc, @Cast(value="size_t*") SizeTPointer weightSizeInBytes, @Cast(value="size_t*") SizeTPointer workSpaceSizeInBytes, @Cast(value="size_t*") SizeTPointer reserveSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetMultiHeadAttnWeights(cudnnContext handle, cudnnAttnStruct attnDesc, @Cast(value="cudnnMultiHeadAttnWeightKind_t") int wKind, @Cast(value="size_t") long weightSizeInBytes, @Const Pointer w, cudnnTensorStruct wDesc, @Cast(value="void**") PointerPointer wAddr)
@Cast(value="cudnnStatus_t") public static int cudnnGetMultiHeadAttnWeights(cudnnContext handle, cudnnAttnStruct attnDesc, @Cast(value="cudnnMultiHeadAttnWeightKind_t") int wKind, @Cast(value="size_t") long weightSizeInBytes, @Const Pointer w, cudnnTensorStruct wDesc, @Cast(value="void**") @ByPtrPtr Pointer wAddr)
@Cast(value="cudnnStatus_t") public static int cudnnMultiHeadAttnForward(cudnnContext handle, cudnnAttnStruct attnDesc, int currIdx, @Const IntPointer loWinIdx, @Const IntPointer hiWinIdx, @Const IntPointer seqLengthArrayQRO, @Const IntPointer seqLengthArrayKV, cudnnSeqDataStruct qDesc, @Const Pointer queries, @Const Pointer residuals, cudnnSeqDataStruct kDesc, @Const Pointer keys, cudnnSeqDataStruct vDesc, @Const Pointer values, cudnnSeqDataStruct oDesc, Pointer out, @Cast(value="size_t") long weightSizeInBytes, @Const Pointer w, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer workSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes, Pointer reserveSpace)
@Cast(value="cudnnStatus_t") public static int cudnnMultiHeadAttnForward(cudnnContext handle, cudnnAttnStruct attnDesc, int currIdx, @Const IntBuffer loWinIdx, @Const IntBuffer hiWinIdx, @Const IntBuffer seqLengthArrayQRO, @Const IntBuffer seqLengthArrayKV, cudnnSeqDataStruct qDesc, @Const Pointer queries, @Const Pointer residuals, cudnnSeqDataStruct kDesc, @Const Pointer keys, cudnnSeqDataStruct vDesc, @Const Pointer values, cudnnSeqDataStruct oDesc, Pointer out, @Cast(value="size_t") long weightSizeInBytes, @Const Pointer w, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer workSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes, Pointer reserveSpace)
@Cast(value="cudnnStatus_t") public static int cudnnMultiHeadAttnForward(cudnnContext handle, cudnnAttnStruct attnDesc, int currIdx, @Const int[] loWinIdx, @Const int[] hiWinIdx, @Const int[] seqLengthArrayQRO, @Const int[] seqLengthArrayKV, cudnnSeqDataStruct qDesc, @Const Pointer queries, @Const Pointer residuals, cudnnSeqDataStruct kDesc, @Const Pointer keys, cudnnSeqDataStruct vDesc, @Const Pointer values, cudnnSeqDataStruct oDesc, Pointer out, @Cast(value="size_t") long weightSizeInBytes, @Const Pointer w, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer workSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes, Pointer reserveSpace)
@Cast(value="cudnnStatus_t") public static int cudnnMultiHeadAttnBackwardData(cudnnContext handle, cudnnAttnStruct attnDesc, @Const IntPointer loWinIdx, @Const IntPointer hiWinIdx, @Const IntPointer seqLengthArrayDQDO, @Const IntPointer seqLengthArrayDKDV, cudnnSeqDataStruct doDesc, @Const Pointer dout, cudnnSeqDataStruct dqDesc, Pointer dqueries, @Const Pointer queries, cudnnSeqDataStruct dkDesc, Pointer dkeys, @Const Pointer keys, cudnnSeqDataStruct dvDesc, Pointer dvalues, @Const Pointer values, @Cast(value="size_t") long weightSizeInBytes, @Const Pointer w, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer workSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes, Pointer reserveSpace)
@Cast(value="cudnnStatus_t") public static int cudnnMultiHeadAttnBackwardData(cudnnContext handle, cudnnAttnStruct attnDesc, @Const IntBuffer loWinIdx, @Const IntBuffer hiWinIdx, @Const IntBuffer seqLengthArrayDQDO, @Const IntBuffer seqLengthArrayDKDV, cudnnSeqDataStruct doDesc, @Const Pointer dout, cudnnSeqDataStruct dqDesc, Pointer dqueries, @Const Pointer queries, cudnnSeqDataStruct dkDesc, Pointer dkeys, @Const Pointer keys, cudnnSeqDataStruct dvDesc, Pointer dvalues, @Const Pointer values, @Cast(value="size_t") long weightSizeInBytes, @Const Pointer w, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer workSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes, Pointer reserveSpace)
@Cast(value="cudnnStatus_t") public static int cudnnMultiHeadAttnBackwardData(cudnnContext handle, cudnnAttnStruct attnDesc, @Const int[] loWinIdx, @Const int[] hiWinIdx, @Const int[] seqLengthArrayDQDO, @Const int[] seqLengthArrayDKDV, cudnnSeqDataStruct doDesc, @Const Pointer dout, cudnnSeqDataStruct dqDesc, Pointer dqueries, @Const Pointer queries, cudnnSeqDataStruct dkDesc, Pointer dkeys, @Const Pointer keys, cudnnSeqDataStruct dvDesc, Pointer dvalues, @Const Pointer values, @Cast(value="size_t") long weightSizeInBytes, @Const Pointer w, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer workSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes, Pointer reserveSpace)
@Cast(value="cudnnStatus_t") public static int cudnnMultiHeadAttnBackwardWeights(cudnnContext handle, cudnnAttnStruct attnDesc, @Cast(value="cudnnWgradMode_t") int addGrad, cudnnSeqDataStruct qDesc, @Const Pointer queries, cudnnSeqDataStruct kDesc, @Const Pointer keys, cudnnSeqDataStruct vDesc, @Const Pointer values, cudnnSeqDataStruct doDesc, @Const Pointer dout, @Cast(value="size_t") long weightSizeInBytes, @Const Pointer w, Pointer dw, @Cast(value="size_t") long workSpaceSizeInBytes, Pointer workSpace, @Cast(value="size_t") long reserveSpaceSizeInBytes, Pointer reserveSpace)
@Cast(value="cudnnStatus_t") public static int cudnnCreateCTCLossDescriptor(@ByPtrPtr cudnnCTCLossStruct ctcLossDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc, @Cast(value="cudnnDataType_t") int compType)
@Cast(value="cudnnStatus_t") public static int cudnnGetCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc, @Cast(value="cudnnDataType_t*") IntPointer compType)
@Cast(value="cudnnStatus_t") public static int cudnnGetCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc, @Cast(value="cudnnDataType_t*") IntBuffer compType)
@Cast(value="cudnnStatus_t") public static int cudnnGetCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc, @Cast(value="cudnnDataType_t*") int[] compType)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyCTCLossDescriptor(cudnnCTCLossStruct ctcLossDesc)
@Cast(value="cudnnStatus_t") public static int cudnnCTCLoss(cudnnContext handle, cudnnTensorStruct probsDesc, @Const Pointer probs, @Const IntPointer labels, @Const IntPointer labelLengths, @Const IntPointer inputLengths, Pointer costs, cudnnTensorStruct gradientsDesc, @Const Pointer gradients, @Cast(value="cudnnCTCLossAlgo_t") int algo, cudnnCTCLossStruct ctcLossDesc, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnCTCLoss(cudnnContext handle, cudnnTensorStruct probsDesc, @Const Pointer probs, @Const IntBuffer labels, @Const IntBuffer labelLengths, @Const IntBuffer inputLengths, Pointer costs, cudnnTensorStruct gradientsDesc, @Const Pointer gradients, @Cast(value="cudnnCTCLossAlgo_t") int algo, cudnnCTCLossStruct ctcLossDesc, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnCTCLoss(cudnnContext handle, cudnnTensorStruct probsDesc, @Const Pointer probs, @Const int[] labels, @Const int[] labelLengths, @Const int[] inputLengths, Pointer costs, cudnnTensorStruct gradientsDesc, @Const Pointer gradients, @Cast(value="cudnnCTCLossAlgo_t") int algo, cudnnCTCLossStruct ctcLossDesc, Pointer workspace, @Cast(value="size_t") long workSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetCTCLossWorkspaceSize(cudnnContext handle, cudnnTensorStruct probsDesc, cudnnTensorStruct gradientsDesc, @Const IntPointer labels, @Const IntPointer labelLengths, @Const IntPointer inputLengths, @Cast(value="cudnnCTCLossAlgo_t") int algo, cudnnCTCLossStruct ctcLossDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetCTCLossWorkspaceSize(cudnnContext handle, cudnnTensorStruct probsDesc, cudnnTensorStruct gradientsDesc, @Const IntBuffer labels, @Const IntBuffer labelLengths, @Const IntBuffer inputLengths, @Cast(value="cudnnCTCLossAlgo_t") int algo, cudnnCTCLossStruct ctcLossDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnGetCTCLossWorkspaceSize(cudnnContext handle, cudnnTensorStruct probsDesc, cudnnTensorStruct gradientsDesc, @Const int[] labels, @Const int[] labelLengths, @Const int[] inputLengths, @Cast(value="cudnnCTCLossAlgo_t") int algo, cudnnCTCLossStruct ctcLossDesc, @Cast(value="size_t*") SizeTPointer sizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnCreateAlgorithmDescriptor(@ByPtrPtr cudnnAlgorithmStruct algoDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetAlgorithmDescriptor(cudnnAlgorithmStruct algoDesc, @ByVal cudnnAlgorithm_t algorithm)
@Cast(value="cudnnStatus_t") public static int cudnnGetAlgorithmDescriptor(cudnnAlgorithmStruct algoDesc, cudnnAlgorithm_t algorithm)
@Cast(value="cudnnStatus_t") public static int cudnnCopyAlgorithmDescriptor(cudnnAlgorithmStruct src, cudnnAlgorithmStruct dest)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyAlgorithmDescriptor(cudnnAlgorithmStruct algoDesc)
@Cast(value="cudnnStatus_t") public static int cudnnCreateAlgorithmPerformance(@ByPtrPtr cudnnAlgorithmPerformanceStruct algoPerf, int numberToCreate)
@Cast(value="cudnnStatus_t") public static int cudnnSetAlgorithmPerformance(cudnnAlgorithmPerformanceStruct algoPerf, cudnnAlgorithmStruct algoDesc, @Cast(value="cudnnStatus_t") int status, float time, @Cast(value="size_t") long memory)
@Cast(value="cudnnStatus_t") public static int cudnnGetAlgorithmPerformance(cudnnAlgorithmPerformanceStruct algoPerf, @ByPtrPtr cudnnAlgorithmStruct algoDesc, @Cast(value="cudnnStatus_t*") IntPointer status, FloatPointer time, @Cast(value="size_t*") SizeTPointer memory)
@Cast(value="cudnnStatus_t") public static int cudnnGetAlgorithmPerformance(cudnnAlgorithmPerformanceStruct algoPerf, @ByPtrPtr cudnnAlgorithmStruct algoDesc, @Cast(value="cudnnStatus_t*") IntBuffer status, FloatBuffer time, @Cast(value="size_t*") SizeTPointer memory)
@Cast(value="cudnnStatus_t") public static int cudnnGetAlgorithmPerformance(cudnnAlgorithmPerformanceStruct algoPerf, @ByPtrPtr cudnnAlgorithmStruct algoDesc, @Cast(value="cudnnStatus_t*") int[] status, float[] time, @Cast(value="size_t*") SizeTPointer memory)
@Cast(value="cudnnStatus_t") public static int cudnnDestroyAlgorithmPerformance(@ByPtrPtr cudnnAlgorithmPerformanceStruct algoPerf, int numberToDestroy)
@Cast(value="cudnnStatus_t") public static int cudnnGetAlgorithmSpaceSize(cudnnContext handle, cudnnAlgorithmStruct algoDesc, @Cast(value="size_t*") SizeTPointer algoSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnSaveAlgorithm(cudnnContext handle, cudnnAlgorithmStruct algoDesc, Pointer algoSpace, @Cast(value="size_t") long algoSpaceSizeInBytes)
@Cast(value="cudnnStatus_t") public static int cudnnRestoreAlgorithm(cudnnContext handle, Pointer algoSpace, @Cast(value="size_t") long algoSpaceSizeInBytes, cudnnAlgorithmStruct algoDesc)
@Cast(value="cudnnStatus_t") public static int cudnnSetCallback(@Cast(value="unsigned") int mask, Pointer udata, cudnnCallback_t fptr)
@Cast(value="cudnnStatus_t") public static int cudnnGetCallback(@Cast(value="unsigned*") IntPointer mask, @Cast(value="void**") PointerPointer udata, @ByPtrPtr cudnnCallback_t fptr)
@Cast(value="cudnnStatus_t") public static int cudnnGetCallback(@Cast(value="unsigned*") IntPointer mask, @Cast(value="void**") @ByPtrPtr Pointer udata, @ByPtrPtr cudnnCallback_t fptr)
@Cast(value="cudnnStatus_t") public static int cudnnGetCallback(@Cast(value="unsigned*") IntBuffer mask, @Cast(value="void**") @ByPtrPtr Pointer udata, @ByPtrPtr cudnnCallback_t fptr)
@Cast(value="cudnnStatus_t") public static int cudnnGetCallback(@Cast(value="unsigned*") int[] mask, @Cast(value="void**") @ByPtrPtr Pointer udata, @ByPtrPtr cudnnCallback_t fptr)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNDescriptor_v6(cudnnContext handle, cudnnRNNStruct rnnDesc, int hiddenSize, int numLayers, cudnnDropoutStruct dropoutDesc, @Cast(value="cudnnRNNInputMode_t") int inputMode, @Cast(value="cudnnDirectionMode_t") int direction, @Cast(value="cudnnRNNMode_t") int mode, @Cast(value="cudnnRNNAlgo_t") int algo, @Cast(value="cudnnDataType_t") int mathPrec)
@Cast(value="cudnnStatus_t") public static int cudnnSetRNNDescriptor_v5(cudnnRNNStruct rnnDesc, int hiddenSize, int numLayers, cudnnDropoutStruct dropoutDesc, @Cast(value="cudnnRNNInputMode_t") int inputMode, @Cast(value="cudnnDirectionMode_t") int direction, @Cast(value="cudnnRNNMode_t") int mode, @Cast(value="cudnnDataType_t") int mathPrec)
Copyright © 2019. All rights reserved.