Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.cudnnCreateReduceTensorDescriptor(cudnnReduceTensorDescriptor reduceTensorDesc) |
static int |
JCudnn.cudnnDestroyReduceTensorDescriptor(cudnnReduceTensorDescriptor reduceTensorDesc) |
static int |
JCudnn.cudnnGetReduceTensorDescriptor(cudnnReduceTensorDescriptor reduceTensorDesc,
int[] reduceTensorOp,
int[] reduceTensorCompType,
int[] reduceTensorNanOpt,
int[] reduceTensorIndices,
int[] reduceTensorIndicesType) |
static int |
JCudnn.cudnnGetReductionIndicesSize(cudnnHandle handle,
cudnnReduceTensorDescriptor reduceTensorDesc,
cudnnTensorDescriptor aDesc,
cudnnTensorDescriptor cDesc,
long[] sizeInBytes)
Helper function to return the minimum size of the index space to be passed to the reduction given the input and
output tensors
|
static int |
JCudnn.cudnnGetReductionWorkspaceSize(cudnnHandle handle,
cudnnReduceTensorDescriptor reduceTensorDesc,
cudnnTensorDescriptor aDesc,
cudnnTensorDescriptor cDesc,
long[] sizeInBytes)
Helper function to return the minimum size of the workspace to be passed to the reduction given the input and output
tensors
|
static int |
JCudnn.cudnnReduceTensor(cudnnHandle handle,
cudnnReduceTensorDescriptor reduceTensorDesc,
Pointer indices,
long indicesSizeInBytes,
Pointer workspace,
long workspaceSizeInBytes,
Pointer alpha,
cudnnTensorDescriptor aDesc,
Pointer A,
Pointer beta,
cudnnTensorDescriptor cDesc,
Pointer C)
The indices space is ignored for reduce ops other than min or max.
|
static int |
JCudnn.cudnnSetReduceTensorDescriptor(cudnnReduceTensorDescriptor reduceTensorDesc,
int reduceTensorOp,
int reduceTensorCompType,
int reduceTensorNanOpt,
int reduceTensorIndices,
int reduceTensorIndicesType) |
Copyright © 2020. All rights reserved.