public class LibMatrixDNNPooling extends Object
| Constructor and Description |
|---|
LibMatrixDNNPooling() |
| Modifier and Type | Method and Description |
|---|---|
static ArrayList<Callable<Long>> |
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>> |
getPoolingWorkers(DnnParameters params,
LibMatrixDNN.PoolingType poolType)
Factory method that returns list of callable tasks for performing pooling operation
|
static void |
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) |
public static ArrayList<Callable<Long>> getPoolingWorkers(DnnParameters params, LibMatrixDNN.PoolingType poolType)
params - convolution parameterspoolType - type of poolingpublic static ArrayList<Callable<Long>> getPoolingBackwardWorkers(DnnParameters params, boolean performReluBackward, LibMatrixDNN.PoolingType poolType)
params - convolution parametersperformReluBackward - whether to perform ReLU backwardpoolType - type of pooling operation to performpublic static void 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 © 2021 The Apache Software Foundation. All rights reserved.