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,
int[] direction) |
static int |
cudnn.cudnnGetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDimsRequested,
IntBuffer destFormat,
IntBuffer padBeforeA,
IntBuffer padAfterA,
IntBuffer foldA,
IntBuffer direction) |
static int |
cudnn.cudnnGetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDimsRequested,
IntPointer destFormat,
IntPointer padBeforeA,
IntPointer padAfterA,
IntPointer foldA,
IntPointer 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,
int direction) |
static int |
cudnn.cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDims,
int destFormat,
IntBuffer padBeforeA,
IntBuffer padAfterA,
IntBuffer foldA,
int direction) |
static int |
cudnn.cudnnSetTensorTransformDescriptor(cudnnTensorTransformStruct transformDesc,
int nbDims,
int destFormat,
IntPointer padBeforeA,
IntPointer padAfterA,
IntPointer foldA,
int 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 © 2024. All rights reserved.