org.deeplearning4j.nn.layers.convolution

Class CudnnConvolutionHelper

java.lang.Object

org.deeplearning4j.nn.layers.BaseCudnnHelper

org.deeplearning4j.nn.layers.convolution.CudnnConvolutionHelper

All Implemented Interfaces:

org.deeplearning4j.nn.layers.convolution.ConvolutionHelper

public class CudnnConvolutionHelper
extends BaseCudnnHelper
implements org.deeplearning4j.nn.layers.convolution.ConvolutionHelper

cuDNN-based helper for the convolution layer.

Author:

saudet

Nested Class Summary

Nested classes/interfaces inherited from class org.deeplearning4j.nn.layers.BaseCudnnHelper

BaseCudnnHelper.CudnnContext, BaseCudnnHelper.DataCache, BaseCudnnHelper.TensorArray

Field Summary

Fields inherited from class org.deeplearning4j.nn.layers.BaseCudnnHelper

alpha, beta, dataType, dataTypeSize, sizeInBytes, tensorFormat

Constructor Summary

Constructors

Constructor and Description
CudnnConvolutionHelper()

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	activate(org.nd4j.linalg.api.ndarray.INDArray z, org.nd4j.linalg.activations.IActivation afn)
org.deeplearning4j.berkeley.Pair<org.deeplearning4j.nn.gradient.Gradient,org.nd4j.linalg.api.ndarray.INDArray>	backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray weights, org.nd4j.linalg.api.ndarray.INDArray delta, int[] kernel, int[] strides, int[] pad, org.nd4j.linalg.api.ndarray.INDArray biasGradView, org.nd4j.linalg.api.ndarray.INDArray weightGradView, org.nd4j.linalg.activations.IActivation afn, org.deeplearning4j.nn.conf.layers.ConvolutionLayer.AlgoMode mode, org.deeplearning4j.nn.conf.layers.ConvolutionLayer.BwdFilterAlgo bwdFilterAlgo, org.deeplearning4j.nn.conf.layers.ConvolutionLayer.BwdDataAlgo bwdDataAlgo, org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)
org.nd4j.linalg.api.ndarray.INDArray	preOutput(org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray weights, org.nd4j.linalg.api.ndarray.INDArray bias, int[] kernel, int[] strides, int[] pad, org.deeplearning4j.nn.conf.layers.ConvolutionLayer.AlgoMode mode, org.deeplearning4j.nn.conf.layers.ConvolutionLayer.FwdAlgo fwdAlgo, org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)

Methods inherited from class org.deeplearning4j.nn.layers.BaseCudnnHelper

checkCuda, checkCudnn, checkSupported

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.deeplearning4j.nn.layers.convolution.ConvolutionHelper

checkSupported

Constructor Detail

CudnnConvolutionHelper

public CudnnConvolutionHelper()

Method Detail

backpropGradient

public org.deeplearning4j.berkeley.Pair<org.deeplearning4j.nn.gradient.Gradient,org.nd4j.linalg.api.ndarray.INDArray> backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input,
                                                                                                                                       org.nd4j.linalg.api.ndarray.INDArray weights,
                                                                                                                                       org.nd4j.linalg.api.ndarray.INDArray delta,
                                                                                                                                       int[] kernel,
                                                                                                                                       int[] strides,
                                                                                                                                       int[] pad,
                                                                                                                                       org.nd4j.linalg.api.ndarray.INDArray biasGradView,
                                                                                                                                       org.nd4j.linalg.api.ndarray.INDArray weightGradView,
                                                                                                                                       org.nd4j.linalg.activations.IActivation afn,
                                                                                                                                       org.deeplearning4j.nn.conf.layers.ConvolutionLayer.AlgoMode mode,
                                                                                                                                       org.deeplearning4j.nn.conf.layers.ConvolutionLayer.BwdFilterAlgo bwdFilterAlgo,
                                                                                                                                       org.deeplearning4j.nn.conf.layers.ConvolutionLayer.BwdDataAlgo bwdDataAlgo,
                                                                                                                                       org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)

Specified by:

backpropGradient in interface org.deeplearning4j.nn.layers.convolution.ConvolutionHelper

preOutput

public org.nd4j.linalg.api.ndarray.INDArray preOutput(org.nd4j.linalg.api.ndarray.INDArray input,
                                                      org.nd4j.linalg.api.ndarray.INDArray weights,
                                                      org.nd4j.linalg.api.ndarray.INDArray bias,
                                                      int[] kernel,
                                                      int[] strides,
                                                      int[] pad,
                                                      org.deeplearning4j.nn.conf.layers.ConvolutionLayer.AlgoMode mode,
                                                      org.deeplearning4j.nn.conf.layers.ConvolutionLayer.FwdAlgo fwdAlgo,
                                                      org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)

Specified by:

preOutput in interface org.deeplearning4j.nn.layers.convolution.ConvolutionHelper

activate

public org.nd4j.linalg.api.ndarray.INDArray activate(org.nd4j.linalg.api.ndarray.INDArray z,
                                                     org.nd4j.linalg.activations.IActivation afn)

Specified by:

activate in interface org.deeplearning4j.nn.layers.convolution.ConvolutionHelper