org.deeplearning4j.nn.layers.mkldnn

Class MKLDNNSubsamplingHelper

java.lang.Object

org.deeplearning4j.nn.layers.mkldnn.MKLDNNSubsamplingHelper

All Implemented Interfaces:

SubsamplingHelper, LayerHelper

public class MKLDNNSubsamplingHelper
extends Object
implements SubsamplingHelper

MKL-DNN Subsampling (2d) helper

Author:

Alex Black

Field Summary

Fields

Modifier and Type	Field and Description
protected OpContext	context

Constructor Summary

Constructors

Constructor and Description
MKLDNNSubsamplingHelper(org.nd4j.linalg.api.buffer.DataType dataType)

Method Summary

Modifier and Type	Method and Description
INDArray	activate(INDArray input, boolean training, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode, int[] dilation, LayerWorkspaceMgr workspaceMgr)
Pair<Gradient,INDArray>	backpropGradient(INDArray input, INDArray epsilon, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode, int[] dilation, LayerWorkspaceMgr workspaceMgr)
boolean	checkSupported()
Map<String,Long>	helperMemoryUse() Return the currently allocated memory for the helper. (a) Excludes: any shared memory used by multiple helpers/layers (b) Excludes any temporary memory (c) Includes all memory that persists for longer than the helper method This is mainly used for debugging and reporting purposes.

Methods inherited from class java.lang.Object

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

Field Detail

context

protected OpContext context

Constructor Detail

MKLDNNSubsamplingHelper

public MKLDNNSubsamplingHelper(org.nd4j.linalg.api.buffer.DataType dataType)

Method Detail

checkSupported

public boolean checkSupported()

Specified by:

checkSupported in interface SubsamplingHelper

backpropGradient

public Pair<Gradient,INDArray> backpropGradient(INDArray input,
                                                INDArray epsilon,
                                                int[] kernel,
                                                int[] strides,
                                                int[] pad,
                                                PoolingType poolingType,
                                                ConvolutionMode convolutionMode,
                                                int[] dilation,
                                                LayerWorkspaceMgr workspaceMgr)

Specified by:

backpropGradient in interface SubsamplingHelper

activate

public INDArray activate(INDArray input,
                         boolean training,
                         int[] kernel,
                         int[] strides,
                         int[] pad,
                         PoolingType poolingType,
                         ConvolutionMode convolutionMode,
                         int[] dilation,
                         LayerWorkspaceMgr workspaceMgr)

Specified by:

activate in interface SubsamplingHelper

helperMemoryUse

public Map<String,Long> helperMemoryUse()

Description copied from interface: LayerHelper

Return the currently allocated memory for the helper.
(a) Excludes: any shared memory used by multiple helpers/layers
(b) Excludes any temporary memory (c) Includes all memory that persists for longer than the helper method
This is mainly used for debugging and reporting purposes. Returns a map:
Key: The name of the type of memory
Value: The amount of memory

Specified by:

helperMemoryUse in interface LayerHelper

Returns:

Map of memory, may be null if none is used.