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, org.deeplearning4j.nn.layers.LayerHelper

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, TENSOR_FORMAT

Constructor Summary

Constructors

Constructor and Description
CudnnSubsamplingHelper()

Method Summary

Modifier and Type	Method and Description
INDArray	activate(INDArray input, boolean training, int[] kernel, int[] strides, int[] pad, org.deeplearning4j.nn.conf.layers.PoolingType poolingType, org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode, int[] dilation, org.deeplearning4j.nn.workspace.LayerWorkspaceMgr workspaceMgr)
org.nd4j.linalg.primitives.Pair<org.deeplearning4j.nn.gradient.Gradient,INDArray>	backpropGradient(INDArray input, INDArray epsilon, int[] kernel, int[] strides, int[] pad, org.deeplearning4j.nn.conf.layers.PoolingType poolingType, org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode, int[] dilation, org.deeplearning4j.nn.workspace.LayerWorkspaceMgr workspaceMgr)
Map<String,Long>	helperMemoryUse()

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

adaptForTensorDescr, checkCuda, checkCudnn, checkSupported, toCudnnDataType

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.nd4j.linalg.primitives.Pair<org.deeplearning4j.nn.gradient.Gradient,INDArray> backpropGradient(INDArray input,
                                                                                                          INDArray epsilon,
                                                                                                          int[] kernel,
                                                                                                          int[] strides,
                                                                                                          int[] pad,
                                                                                                          org.deeplearning4j.nn.conf.layers.PoolingType poolingType,
                                                                                                          org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode,
                                                                                                          int[] dilation,
                                                                                                          org.deeplearning4j.nn.workspace.LayerWorkspaceMgr workspaceMgr)

Specified by:

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

activate

public INDArray activate(INDArray input,
                         boolean training,
                         int[] kernel,
                         int[] strides,
                         int[] pad,
                         org.deeplearning4j.nn.conf.layers.PoolingType poolingType,
                         org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode,
                         int[] dilation,
                         org.deeplearning4j.nn.workspace.LayerWorkspaceMgr workspaceMgr)

Specified by:

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

helperMemoryUse

public Map<String,Long> helperMemoryUse()

Specified by:

helperMemoryUse in interface org.deeplearning4j.nn.layers.LayerHelper