| Package | Description |
|---|---|
| org.bytedeco.cuda.global |
| Modifier and Type | Method and Description |
|---|---|
static int |
cudnn.cudnnCreateTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc)
Create an empty tensor transform descriptor
|
static int |
cudnn.cudnnDestroyTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc)
Destroys a previously created tensor transform descriptor.
|
static int |
cudnn.cudnnGetFoldedConvBackwardDataDescriptors(cudnnContext handle,
cudnnFilterStruct filterDesc,
cudnnTensorStruct diffDesc,
cudnnConvolutionStruct convDesc,
cudnnTensorStruct gradDesc,
int transformFormat,
cudnnFilterStruct foldedFilterDesc,
cudnnTensorStruct paddedDiffDesc,
cudnnConvolutionStruct foldedConvDesc,
cudnnTensorStruct foldedGradDesc,
cudnnTensorTransformStruct filterFoldTransDesc,
cudnnTensorTransformStruct diffPadTransDesc,
cudnnTensorTransformStruct gradFoldTransDesc,
cudnnTensorTransformStruct gradUnfoldTransDesc) |
static int |
cudnn.cudnnGetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDimsRequested,
int[] destFormat,
int[] padBeforeA,
int[] padAfterA,
int[] foldA,
long[] direction) |
static int |
cudnn.cudnnGetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDimsRequested,
IntBuffer destFormat,
IntBuffer padBeforeA,
IntBuffer padAfterA,
IntBuffer foldA,
LongBuffer direction) |
static int |
cudnn.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 int |
cudnn.cudnnInitTransformDest(cudnnTensorTransformStruct transformDesc,
cudnnTensorStruct srcDesc,
cudnnTensorStruct destDesc,
SizeTPointer destSizeInBytes)
Create a destination descriptor for cudnnTransformTensor
|
static int |
cudnn.cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDims,
int destFormat,
int[] padBeforeA,
int[] padAfterA,
int[] foldA,
long direction) |
static int |
cudnn.cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDims,
int destFormat,
IntBuffer padBeforeA,
IntBuffer padAfterA,
IntBuffer foldA,
long direction) |
static int |
cudnn.cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDims,
int destFormat,
IntPointer padBeforeA,
IntPointer padAfterA,
IntPointer foldA,
long direction)
Initialize a previously created tensor transform descriptor.
|
static int |
cudnn.cudnnTransformFilter(cudnnContext handle,
cudnnTensorTransformStruct transDesc,
Pointer alpha,
cudnnFilterStruct srcDesc,
Pointer srcData,
Pointer beta,
cudnnFilterStruct destDesc,
Pointer destData) |
static int |
cudnn.cudnnTransformTensorEx(cudnnContext handle,
cudnnTensorTransformStruct transDesc,
Pointer alpha,
cudnnTensorStruct srcDesc,
Pointer srcData,
Pointer beta,
cudnnTensorStruct destDesc,
Pointer destData) |
Copyright © 2019. All rights reserved.