| Modifier and Type | Method and Description |
|---|---|
static int |
JCudnn.cudnnCreatePoolingDescriptor(cudnnPoolingDescriptor poolingDesc) |
static int |
JCudnn.cudnnDestroyPoolingDescriptor(cudnnPoolingDescriptor poolingDesc) |
static int |
JCudnn.cudnnGetPooling2dDescriptor(cudnnPoolingDescriptor poolingDesc,
int[] mode,
int[] maxpoolingNanOpt,
int[] windowHeight,
int[] windowWidth,
int[] verticalPadding,
int[] horizontalPadding,
int[] verticalStride,
int[] horizontalStride) |
static int |
JCudnn.cudnnGetPooling2dForwardOutputDim(cudnnPoolingDescriptor poolingDesc,
cudnnTensorDescriptor inputTensorDesc,
int[] n,
int[] c,
int[] h,
int[] w) |
static int |
JCudnn.cudnnGetPoolingNdDescriptor(cudnnPoolingDescriptor poolingDesc,
int nbDimsRequested,
int[] mode,
int[] maxpoolingNanOpt,
int[] nbDims,
int[] windowDimA,
int[] paddingA,
int[] strideA) |
static int |
JCudnn.cudnnGetPoolingNdForwardOutputDim(cudnnPoolingDescriptor poolingDesc,
cudnnTensorDescriptor inputTensorDesc,
int nbDims,
int[] outputTensorDimA) |
static int |
JCudnn.cudnnPoolingBackward(cudnnHandle handle,
cudnnPoolingDescriptor poolingDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y,
cudnnTensorDescriptor dyDesc,
jcuda.Pointer dy,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor dxDesc,
jcuda.Pointer dx) |
static int |
JCudnn.cudnnPoolingForward(cudnnHandle handle,
cudnnPoolingDescriptor poolingDesc,
jcuda.Pointer alpha,
cudnnTensorDescriptor xDesc,
jcuda.Pointer x,
jcuda.Pointer beta,
cudnnTensorDescriptor yDesc,
jcuda.Pointer y) |
static int |
JCudnn.cudnnSetPooling2dDescriptor(cudnnPoolingDescriptor poolingDesc,
int mode,
int maxpoolingNanOpt,
int windowHeight,
int windowWidth,
int verticalPadding,
int horizontalPadding,
int verticalStride,
int horizontalStride) |
static int |
JCudnn.cudnnSetPoolingNdDescriptor(cudnnPoolingDescriptor poolingDesc,
int mode,
int maxpoolingNanOpt,
int nbDims,
int[] windowDimA,
int[] paddingA,
int[] strideA) |
Copyright © 2017. All rights reserved.