org.apache.sysml.runtime.matrix.data

Class LibMatrixDNNConv2d

java.lang.Object

org.apache.sysml.runtime.matrix.data.LibMatrixDNNConv2d

public class LibMatrixDNNConv2d
extends Object

This class contains the set of operators used for performing conv2d

Constructor Summary

Constructors

Constructor and Description
LibMatrixDNNConv2d()

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

Methods inherited from class java.lang.Object

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

Constructor Detail

LibMatrixDNNConv2d

public LibMatrixDNNConv2d()

Method Detail

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