org.deeplearning4j.nn.layers.convolution

Interface ConvolutionHelper

All Superinterfaces:

LayerHelper

All Known Implementing Classes:

MKLDNNConvHelper

public interface ConvolutionHelper
extends LayerHelper

Method Summary

Modifier and Type	Method and Description
INDArray	activate(INDArray z, IActivation afn, boolean training)
Pair<Gradient,INDArray>	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)
boolean	checkSupported()
INDArray	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)

Methods inherited from interface org.deeplearning4j.nn.layers.LayerHelper

helperMemoryUse

Method Detail

checkSupported

boolean checkSupported()

Specified by:

checkSupported in interface LayerHelper

backpropGradient

Pair<Gradient,INDArray> 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)

preOutput

INDArray 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)

activate

INDArray activate(INDArray z,
                  IActivation afn,
                  boolean training)