public class LibMatrixDNNConv2d extends Object
| Constructor and Description |
|---|
LibMatrixDNNConv2d() |
| Modifier and Type | Method and Description |
|---|---|
static ArrayList<Callable<Long>> |
getConv2dBackwardDataWorkers(DnnParameters params)
Factory method that returns list of callable tasks for performing conv2d backward data
|
static ArrayList<Callable<Long>> |
getConv2dBackwardFilterWorkers(DnnParameters params)
Factory method that returns list of callable tasks for performing conv2d backward filter
|
static ArrayList<Callable<Long>> |
getConv2dWorkers(DnnParameters params)
Factory method that returns list of callable tasks for performing conv2d
|
public static ArrayList<Callable<Long>> getConv2dWorkers(DnnParameters params)
params - convolution parameterspublic static ArrayList<Callable<Long>> getConv2dBackwardFilterWorkers(DnnParameters params)
params - convolution parameterspublic static ArrayList<Callable<Long>> getConv2dBackwardDataWorkers(DnnParameters params)
params - convolution parametersCopyright © 2021 The Apache Software Foundation. All rights reserved.