| 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.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.cudnnTransformTensorEx(cudnnContext handle,
cudnnTensorTransformStruct transDesc,
Pointer alpha,
cudnnTensorStruct srcDesc,
Pointer srcData,
Pointer beta,
cudnnTensorStruct destDesc,
Pointer destData) |
Copyright © 2019. All rights reserved.