Uses of Class
org.deeplearning4j.nn.conf.layers.ConvolutionLayer.AlgoMode
-
-
Uses of ConvolutionLayer.AlgoMode in org.deeplearning4j.nn.conf
Fields in org.deeplearning4j.nn.conf declared as ConvolutionLayer.AlgoMode Modifier and Type Field Description protected ConvolutionLayer.AlgoMode
NeuralNetConfiguration.Builder. cudnnAlgoMode
Methods in org.deeplearning4j.nn.conf with parameters of type ConvolutionLayer.AlgoMode Modifier and Type Method 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. -
Uses of ConvolutionLayer.AlgoMode in org.deeplearning4j.nn.conf.layers
Fields in org.deeplearning4j.nn.conf.layers declared as ConvolutionLayer.AlgoMode Modifier and Type Field Description protected ConvolutionLayer.AlgoMode
ConvolutionLayer.BaseConvBuilder. cudnnAlgoMode
Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.protected ConvolutionLayer.AlgoMode
ConvolutionLayer. cudnnAlgoMode
Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.Methods in org.deeplearning4j.nn.conf.layers that return ConvolutionLayer.AlgoMode Modifier and Type Method 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.Methods in org.deeplearning4j.nn.conf.layers with parameters of type ConvolutionLayer.AlgoMode Modifier and Type Method Description T
ConvolutionLayer.BaseConvBuilder. cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)
Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory. -
Uses of ConvolutionLayer.AlgoMode in org.deeplearning4j.nn.layers.convolution
Methods in org.deeplearning4j.nn.layers.convolution with parameters of type ConvolutionLayer.AlgoMode Modifier and Type Method 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)
-
Uses of ConvolutionLayer.AlgoMode in org.deeplearning4j.nn.layers.mkldnn
Methods in org.deeplearning4j.nn.layers.mkldnn with parameters of type ConvolutionLayer.AlgoMode Modifier and Type Method 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)
-
Uses of ConvolutionLayer.AlgoMode in org.deeplearning4j.nn.transferlearning
Fields in org.deeplearning4j.nn.transferlearning declared as ConvolutionLayer.AlgoMode Modifier and Type Field Description protected ConvolutionLayer.AlgoMode
FineTuneConfiguration. cudnnAlgoMode
Methods in org.deeplearning4j.nn.transferlearning with parameters of type ConvolutionLayer.AlgoMode Modifier and Type Method 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.
-