Class LibMatrixDNNConv2d
- java.lang.Object
-
- org.apache.sysds.runtime.matrix.data.LibMatrixDNNConv2d
-
public class LibMatrixDNNConv2d extends Object
This class contains the set of operators used for performing conv2d
-
-
Constructor Summary
Constructors Constructor Description LibMatrixDNNConv2d()
-
Method Summary
All Methods Static Methods Concrete Methods Modifier and Type Method Description static ArrayList<Callable<Long>>
getConv2dBackwardDataWorkers(DnnParameters params)
Factory method that returns list of callable tasks for performing conv2d backward datastatic ArrayList<Callable<Long>>
getConv2dBackwardFilterWorkers(DnnParameters params)
Factory method that returns list of callable tasks for performing conv2d backward filterstatic ArrayList<Callable<Long>>
getConv2dWorkers(DnnParameters params)
Factory method that returns list of callable tasks for performing conv2d
-
-
-
Method Detail
-
getConv2dWorkers
public static ArrayList<Callable<Long>> getConv2dWorkers(DnnParameters params)
Factory method that returns list of callable tasks for performing conv2d- Parameters:
params
- convolution parameters- Returns:
- list of callable tasks for performing conv2d
-
getConv2dBackwardFilterWorkers
public static ArrayList<Callable<Long>> getConv2dBackwardFilterWorkers(DnnParameters params)
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
-
getConv2dBackwardDataWorkers
public static ArrayList<Callable<Long>> getConv2dBackwardDataWorkers(DnnParameters params)
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
-
-