org.deeplearning4j.nn.layers.convolution.subsampling

Class CudnnSubsamplingHelper

java.lang.Object

org.deeplearning4j.nn.layers.BaseCudnnHelper

org.deeplearning4j.nn.layers.convolution.subsampling.CudnnSubsamplingHelper

All Implemented Interfaces:

org.deeplearning4j.nn.layers.convolution.subsampling.SubsamplingHelper

public class CudnnSubsamplingHelper
extends BaseCudnnHelper
implements org.deeplearning4j.nn.layers.convolution.subsampling.SubsamplingHelper

cuDNN-based helper for the subsampling 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
CudnnSubsamplingHelper()

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	activate(org.nd4j.linalg.api.ndarray.INDArray input, boolean training, int[] kernel, int[] strides, int[] pad, org.deeplearning4j.nn.conf.layers.PoolingType poolingType, org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)
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 epsilon, int[] kernel, int[] strides, int[] pad, org.deeplearning4j.nn.conf.layers.PoolingType poolingType, 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.subsampling.SubsamplingHelper

checkSupported

Constructor Detail

CudnnSubsamplingHelper

public CudnnSubsamplingHelper()

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 epsilon,
                                                                                                                                       int[] kernel,
                                                                                                                                       int[] strides,
                                                                                                                                       int[] pad,
                                                                                                                                       org.deeplearning4j.nn.conf.layers.PoolingType poolingType,
                                                                                                                                       org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)

Specified by:

backpropGradient in interface org.deeplearning4j.nn.layers.convolution.subsampling.SubsamplingHelper

activate

public org.nd4j.linalg.api.ndarray.INDArray activate(org.nd4j.linalg.api.ndarray.INDArray input,
                                                     boolean training,
                                                     int[] kernel,
                                                     int[] strides,
                                                     int[] pad,
                                                     org.deeplearning4j.nn.conf.layers.PoolingType poolingType,
                                                     org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode)

Specified by:

activate in interface org.deeplearning4j.nn.layers.convolution.subsampling.SubsamplingHelper