Modifier and Type | Method and Description |
---|---|
static int |
JCudnn.cudnnCreatePoolingDescriptor(cudnnPoolingDescriptor poolingDesc)
Create an instance of pooling descriptor
|
static int |
JCudnn.cudnnDestroyPoolingDescriptor(cudnnPoolingDescriptor poolingDesc)
Destroy an instance of pooling descriptor
|
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,
Pointer alpha,
cudnnTensorDescriptor yDesc,
Pointer y,
cudnnTensorDescriptor dyDesc,
Pointer dy,
cudnnTensorDescriptor xDesc,
Pointer x,
Pointer beta,
cudnnTensorDescriptor dxDesc,
Pointer dx)
Function to perform backward pooling
|
static int |
JCudnn.cudnnPoolingForward(cudnnHandle handle,
cudnnPoolingDescriptor poolingDesc,
Pointer alpha,
cudnnTensorDescriptor xDesc,
Pointer x,
Pointer beta,
cudnnTensorDescriptor yDesc,
Pointer y)
Function to perform forward pooling
|
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 © 2020. All rights reserved.