Package | Description |
---|---|
org.deeplearning4j.nn.conf.layers | |
org.deeplearning4j.nn.layers.convolution |
Modifier and Type | Field and Description |
---|---|
protected ConvolutionLayer.AlgoMode |
ConvolutionLayer.cudnnAlgoMode
Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.
|
Modifier and Type | Method and Description |
---|---|
static ConvolutionLayer.AlgoMode |
ConvolutionLayer.AlgoMode.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static ConvolutionLayer.AlgoMode[] |
ConvolutionLayer.AlgoMode.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
ConvolutionLayer.Builder |
ConvolutionLayer.Builder.cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)
Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.
|
Modifier and Type | Method and Description |
---|---|
Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray> |
ConvolutionHelper.backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray weights,
org.nd4j.linalg.api.ndarray.INDArray delta,
int[] kernel,
int[] strides,
int[] pad,
org.nd4j.linalg.api.ndarray.INDArray biasGradView,
org.nd4j.linalg.api.ndarray.INDArray weightGradView,
String afn,
ConvolutionLayer.AlgoMode mode) |
org.nd4j.linalg.api.ndarray.INDArray |
ConvolutionHelper.preOutput(org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray weights,
org.nd4j.linalg.api.ndarray.INDArray bias,
int[] kernel,
int[] strides,
int[] pad,
ConvolutionLayer.AlgoMode mode) |
Copyright © 2016. All Rights Reserved.