Interface ConvolutionHelper
-
- All Superinterfaces:
LayerHelper
- All Known Implementing Classes:
MKLDNNConvHelper
public interface ConvolutionHelper extends LayerHelper
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method 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 interfaceLayerHelper
-
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)
-
-