Modifier and Type | Field and Description |
---|---|
protected ConvolutionLayer.AlgoMode |
NeuralNetConfiguration.Builder.cudnnAlgoMode |
Modifier and Type | Method and Description |
---|---|
NeuralNetConfiguration.Builder |
NeuralNetConfiguration.Builder.cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)
Sets the cuDNN algo mode for convolutional layers, which impacts performance and memory usage of cuDNN.
|
Modifier and Type | Field and Description |
---|---|
protected ConvolutionLayer.AlgoMode |
ConvolutionLayer.cudnnAlgoMode
Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.
|
protected ConvolutionLayer.AlgoMode |
ConvolutionLayer.BaseConvBuilder.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 |
---|---|
T |
ConvolutionLayer.BaseConvBuilder.cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)
Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.
|
Modifier and Type | Method and Description |
---|---|
Pair<Gradient,INDArray> |
ConvolutionHelper.backpropGradient(INDArray input,
INDArray weights,
INDArray bias,
INDArray delta,
int[] kernel,
int[] strides,
int[] pad,
INDArray biasGradView,
INDArray weightGradView,
IActivation afn,
ConvolutionLayer.AlgoMode mode,
ConvolutionLayer.BwdFilterAlgo bwdFilterAlgo,
ConvolutionLayer.BwdDataAlgo bwdDataAlgo,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
INDArray |
ConvolutionHelper.preOutput(INDArray input,
INDArray weights,
INDArray bias,
int[] kernel,
int[] strides,
int[] pad,
ConvolutionLayer.AlgoMode mode,
ConvolutionLayer.FwdAlgo fwdAlgo,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
Modifier and Type | Method and Description |
---|---|
Pair<Gradient,INDArray> |
MKLDNNConvHelper.backpropGradient(INDArray input,
INDArray weights,
INDArray bias,
INDArray delta,
int[] kernel,
int[] strides,
int[] pad,
INDArray biasGradView,
INDArray weightGradView,
IActivation afn,
ConvolutionLayer.AlgoMode mode,
ConvolutionLayer.BwdFilterAlgo bwdFilterAlgo,
ConvolutionLayer.BwdDataAlgo bwdDataAlgo,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
INDArray |
MKLDNNConvHelper.preOutput(INDArray input,
INDArray weights,
INDArray bias,
int[] kernel,
int[] strides,
int[] pad,
ConvolutionLayer.AlgoMode mode,
ConvolutionLayer.FwdAlgo fwdAlgo,
ConvolutionMode convolutionMode,
int[] dilation,
CNN2DFormat format,
LayerWorkspaceMgr workspaceMgr) |
Modifier and Type | Field and Description |
---|---|
protected ConvolutionLayer.AlgoMode |
FineTuneConfiguration.cudnnAlgoMode |
Modifier and Type | Method and Description |
---|---|
FineTuneConfiguration.Builder |
FineTuneConfiguration.Builder.cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)
Sets the cuDNN algo mode for convolutional layers, which impacts performance and memory usage of cuDNN.
|
Copyright © 2021. All rights reserved.