Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.cudnnCreateTensorTransformDescriptor(cudnnTensorTransformDescriptor transformDesc)
Create an empty tensor transform descriptor
|
static int |
JCudnn.cudnnDestroyTensorTransformDescriptor(cudnnTensorTransformDescriptor transformDesc)
Destroys a previously created tensor transform descriptor.
|
static int |
JCudnn.cudnnGetFoldedConvBackwardDataDescriptors(cudnnHandle handle,
cudnnFilterDescriptor filterDesc,
cudnnTensorDescriptor diffDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor gradDesc,
int transformFormat,
cudnnFilterDescriptor foldedFilterDesc,
cudnnTensorDescriptor paddedDiffDesc,
cudnnConvolutionDescriptor foldedConvDesc,
cudnnTensorDescriptor foldedGradDesc,
cudnnTensorTransformDescriptor filterFoldTransDesc,
cudnnTensorTransformDescriptor diffPadTransDesc,
cudnnTensorTransformDescriptor gradFoldTransDesc,
cudnnTensorTransformDescriptor gradUnfoldTransDesc)
Helper function to calculate folding descriptors for dgrad
|
static int |
JCudnn.cudnnGetTensorTransformDescriptor(cudnnTensorTransformDescriptor transformDesc,
int nbDimsRequested,
int[] destFormat,
int[] padBeforeA,
int[] padAfterA,
int[] foldA,
int[] direction)
Retrieves the values stored in a previously initialized tensor transform
descriptor.
|
static int |
JCudnn.cudnnInitTransformDest(cudnnTensorTransformDescriptor transformDesc,
cudnnTensorDescriptor srcDesc,
cudnnTensorDescriptor destDesc,
long[] destSizeInBytes)
Create a destination descriptor for cudnnTransformTensor
|
static int |
JCudnn.cudnnSetTensorTransformDescriptor(cudnnTensorTransformDescriptor transformDesc,
int nbDims,
int destFormat,
int[] padBeforeA,
int[] padAfterA,
int[] foldA,
int direction)
Initialize a previously created tensor transform descriptor.
|
static int |
JCudnn.cudnnTransformFilter(cudnnHandle handle,
cudnnTensorTransformDescriptor transDesc,
Pointer alpha,
cudnnFilterDescriptor srcDesc,
Pointer srcData,
Pointer beta,
cudnnFilterDescriptor destDesc,
Pointer destData) |
static int |
JCudnn.cudnnTransformTensorEx(cudnnHandle handle,
cudnnTensorTransformDescriptor transDesc,
Pointer alpha,
cudnnTensorDescriptor srcDesc,
Pointer srcData,
Pointer beta,
cudnnTensorDescriptor destDesc,
Pointer destData) |
Copyright © 2020. All rights reserved.