org.apache.sysml.runtime.matrix.data

Class LibMatrixDNNHelper

java.lang.Object

org.apache.sysml.runtime.matrix.data.LibMatrixDNNHelper

public class LibMatrixDNNHelper
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	LibMatrixDNNHelper.ReluBackward Performs the operation: (X gt 0) * dout

Constructor Summary

Constructors

Constructor and Description
LibMatrixDNNHelper()

Method Summary

Modifier and Type	Method and Description
static ArrayList<Callable<Long>>	getConv2dBackwardDataWorkers(ConvolutionParameters params) Factory method that returns list of callable tasks for performing conv2d backward data
static ArrayList<Callable<Long>>	getConv2dBackwardFilterWorkers(ConvolutionParameters params) Factory method that returns list of callable tasks for performing conv2d backward filter
static ArrayList<Callable<Long>>	getConv2dWorkers(ConvolutionParameters params) Factory method that returns list of callable tasks for performing conv2d
static ArrayList<Callable<Long>>	getMaxPoolingBackwardWorkers(ConvolutionParameters params, boolean performReluBackward) Factory method that returns list of callable tasks for performing maxpooling backward operation
static ArrayList<Callable<Long>>	getMaxPoolingWorkers(ConvolutionParameters params) Factory method that returns list of callable tasks for performing maxpooling operation
static ArrayList<Callable<Long>>	getReluBackwardWorkers(ConvolutionParameters params) Factory method that returns list of callable tasks for performing relu backward operation

Methods inherited from class java.lang.Object

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

Constructor Detail

LibMatrixDNNHelper

public LibMatrixDNNHelper()

Method Detail

getMaxPoolingWorkers

public static ArrayList<Callable<Long>> getMaxPoolingWorkers(ConvolutionParameters params)
                                                      throws DMLRuntimeException

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

Parameters:

params - convolution parameters

Returns:

list of callable tasks for performing maxpooling operation

Throws:

DMLRuntimeException - if error occurs

getMaxPoolingBackwardWorkers

public static ArrayList<Callable<Long>> getMaxPoolingBackwardWorkers(ConvolutionParameters params,
                                                                     boolean performReluBackward)
                                                              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

Returns:

list of callable tasks for performing maxpooling backward operation

Throws:

DMLRuntimeException - if error occurs

getReluBackwardWorkers

public static ArrayList<Callable<Long>> getReluBackwardWorkers(ConvolutionParameters params)
                                                        throws DMLRuntimeException

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

Parameters:

params - convolution parameters

Returns:

list of callable tasks for performing relu backward operation

Throws:

DMLRuntimeException - if error occurs

getConv2dWorkers

public static ArrayList<Callable<Long>> getConv2dWorkers(ConvolutionParameters params)
                                                  throws DMLRuntimeException

Factory method that returns list of callable tasks for performing conv2d

Parameters:

params - convolution parameters

Returns:

list of callable tasks for performing conv2d

Throws:

DMLRuntimeException - if error occurs

getConv2dBackwardFilterWorkers

public static ArrayList<Callable<Long>> getConv2dBackwardFilterWorkers(ConvolutionParameters params)
                                                                throws DMLRuntimeException

Factory method that returns list of callable tasks for performing conv2d backward filter

Parameters:

params - convolution parameters

Returns:

list of callable tasks for performing conv2d backward filter

Throws:

DMLRuntimeException - if error occurs

getConv2dBackwardDataWorkers

public static ArrayList<Callable<Long>> getConv2dBackwardDataWorkers(ConvolutionParameters params)
                                                              throws DMLRuntimeException

Factory method that returns list of callable tasks for performing conv2d backward data

Parameters:

params - convolution parameters

Returns:

list of callable tasks for performing conv2d backward data

Throws:

DMLRuntimeException - if error occurs