Package | Description |
---|---|
org.apache.sysds.runtime.matrix.data |
Modifier and Type | Method and Description |
---|---|
static LibMatrixDNN.PoolingType |
LibMatrixDNN.PoolingType.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static LibMatrixDNN.PoolingType[] |
LibMatrixDNN.PoolingType.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
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) |
Copyright © 2020 The Apache Software Foundation. All rights reserved.