org.apache.sysml.runtime.matrix.data

Class LibMatrixDNNPooling

java.lang.Object

org.apache.sysml.runtime.matrix.data.LibMatrixDNNPooling

public class LibMatrixDNNPooling
extends Object

This class contains the set of operators used for performing pooling

Constructor Summary

Constructors

Constructor and Description
LibMatrixDNNPooling()

Method Summary

Modifier and Type	Method and Description
static ArrayList<Callable<Long>>	getPoolingBackwardWorkers(ConvolutionParameters params, boolean performReluBackward, LibMatrixDNN.PoolingType poolType) Factory method that returns list of callable tasks for performing maxpooling backward operation
static ArrayList<Callable<Long>>	getPoolingWorkers(ConvolutionParameters params, LibMatrixDNN.PoolingType poolType) Factory method that returns list of callable tasks for performing pooling operation

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LibMatrixDNNPooling

public LibMatrixDNNPooling()

Method Detail

getPoolingWorkers

public static ArrayList<Callable<Long>> getPoolingWorkers(ConvolutionParameters params,
                                                          LibMatrixDNN.PoolingType poolType)
                                                   throws DMLRuntimeException

Factory method that returns list of callable tasks for performing pooling operation

Parameters:

params - convolution parameters

poolType - type of pooling

Returns:

list of callable tasks for performing pooling operation

Throws:

DMLRuntimeException - if error occurs

getPoolingBackwardWorkers

public static ArrayList<Callable<Long>> getPoolingBackwardWorkers(ConvolutionParameters params,
                                                                  boolean performReluBackward,
                                                                  LibMatrixDNN.PoolingType poolType)
                                                           throws DMLRuntimeException

Factory method that returns list of callable tasks for performing maxpooling backward operation

Parameters:

params - convolution parameters

performReluBackward - whether to perform ReLU backward

poolType - type of pooling operation to perform

Returns:

list of callable tasks for performing maxpooling backward operation

Throws:

DMLRuntimeException - if error occurs