static LibMatrixCuDNNPoolingDescriptors |
LibMatrixCuDNNPoolingDescriptors.cudnnPoolingBackwardDescriptors(GPUContext gCtx,
String instName,
int N,
int C,
int H,
int W,
int K,
int R,
int S,
int pad_h,
int pad_w,
int stride_h,
int stride_w,
int P,
int Q,
LibMatrixDNN.PoolingType poolingType) |
Get descriptors for maxpooling backward operation
|
static LibMatrixCuDNNPoolingDescriptors |
LibMatrixCuDNNPoolingDescriptors.cudnnPoolingDescriptors(GPUContext gCtx,
String instName,
int N,
int C,
int H,
int W,
int K,
int R,
int S,
int pad_h,
int pad_w,
int stride_h,
int stride_w,
int P,
int Q,
LibMatrixDNN.PoolingType poolingType) |
Get descriptors for maxpooling operation
|
static ArrayList<Callable<Long>> |
LibMatrixDNNPooling.getPoolingBackwardWorkers(DnnParameters params,
boolean performReluBackward,
LibMatrixDNN.PoolingType poolType) |
Factory method that returns list of callable tasks for performing maxpooling backward operation
|
static ArrayList<Callable<Long>> |
LibMatrixDNNPooling.getPoolingWorkers(DnnParameters params,
LibMatrixDNN.PoolingType poolType) |
Factory method that returns list of callable tasks for performing pooling operation
|
static void |
LibMatrixCuDNN.pooling(GPUContext gCtx,
String instName,
MatrixObject image,
MatrixObject outputBlock,
int N,
int C,
int H,
int W,
int K,
int R,
int S,
int pad_h,
int pad_w,
int stride_h,
int stride_w,
int P,
int Q,
LibMatrixDNN.PoolingType poolingType,
double intermediateMemoryBudget) |
performs maxpooling on GPU by exploiting cudnnPoolingForward(...)
|
static void |
LibMatrixDNN.pooling(MatrixBlock input,
MatrixBlock output,
DnnParameters params,
LibMatrixDNN.PoolingType poolType) |
|
static void |
LibMatrixCuDNN.poolingBackward(GPUContext gCtx,
String instName,
MatrixObject image,
MatrixObject dout,
MatrixObject maxpoolOutput,
MatrixObject outputBlock,
int N,
int C,
int H,
int W,
int K,
int R,
int S,
int pad_h,
int pad_w,
int stride_h,
int stride_w,
int P,
int Q,
LibMatrixDNN.PoolingType poolingType,
double intermediateMemoryBudget) |
Performs maxpoolingBackward on GPU by exploiting cudnnPoolingBackward(...)
This method computes the backpropogation errors for previous layer of maxpooling operation
|
static void |
LibMatrixDNN.poolingBackward(MatrixBlock input,
MatrixBlock dout,
MatrixBlock outputBlock,
DnnParameters params,
boolean performReluBackward,
LibMatrixDNN.PoolingType poolType) |
This method computes the backpropogation errors for previous layer of pooling operation
|
static void |
LibMatrixDNNPooling.poolingDenseStride1Pad0(LibMatrixDNN.PoolingType pType,
double minVal,
double pFact,
double[] in,
double[] out,
int rl,
int ru,
int ii,
int oi,
int C,
int P,
int Q,
int R,
int S,
int H,
int W) |
|