public class LibMatrixDNNRelu extends Object
Modifier and Type | Class and Description |
---|---|
static class |
LibMatrixDNNRelu.ReluBackward
Performs the operation: (X gt 0) * dout
|
Constructor and Description |
---|
LibMatrixDNNRelu() |
Modifier and Type | Method and Description |
---|---|
static ArrayList<Callable<Long>> |
getReluBackwardWorkers(DnnParameters params)
Factory method that returns list of callable tasks for performing relu backward operation
|
public static ArrayList<Callable<Long>> getReluBackwardWorkers(DnnParameters params)
params
- convolution parametersCopyright © 2018 The Apache Software Foundation. All rights reserved.