Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.cudnnCreateOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc) |
static int |
JCudnn.cudnnDestroyOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc) |
static int |
JCudnn.cudnnGetOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc,
int[] opTensorOp,
int[] opTensorCompType,
int[] opTensorNanOpt) |
static int |
JCudnn.cudnnOpTensor(cudnnHandle handle,
cudnnOpTensorDescriptor opTensorDesc,
Pointer alpha1,
cudnnTensorDescriptor aDesc,
Pointer A,
Pointer alpha2,
cudnnTensorDescriptor bDesc,
Pointer B,
Pointer beta,
cudnnTensorDescriptor cDesc,
Pointer C)
B tensor is ignored for CUDNN_OP_TENSOR_SQRT, CUDNN_OP_TENSOR_NOT.
|
static int |
JCudnn.cudnnSetOpTensorDescriptor(cudnnOpTensorDescriptor opTensorDesc,
int opTensorOp,
int opTensorCompType,
int opTensorNanOpt) |
Copyright © 2020. All rights reserved.