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.AlgoModeNeuralNetConfiguration.Builder. cudnnAlgoModeMethods in org.deeplearning4j.nn.conf with parameters of type ConvolutionLayer.AlgoMode Modifier and Type Method Description NeuralNetConfiguration.BuilderNeuralNetConfiguration.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.AlgoModeConvolutionLayer.BaseConvBuilder. cudnnAlgoModeDefaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.protected ConvolutionLayer.AlgoModeConvolutionLayer. cudnnAlgoModeDefaults 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.AlgoModeConvolutionLayer.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 TConvolutionLayer.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)INDArrayConvolutionHelper. 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)INDArrayMKLDNNConvHelper. 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.AlgoModeFineTuneConfiguration. cudnnAlgoModeMethods in org.deeplearning4j.nn.transferlearning with parameters of type ConvolutionLayer.AlgoMode Modifier and Type Method Description FineTuneConfiguration.BuilderFineTuneConfiguration.Builder. cudnnAlgoMode(ConvolutionLayer.AlgoMode cudnnAlgoMode)Sets the cuDNN algo mode for convolutional layers, which impacts performance and memory usage of cuDNN.
-