org.deeplearning4j.nn.layers.mkldnn

Class MKLDNNLocalResponseNormalizationHelper

java.lang.Object

org.deeplearning4j.nn.layers.mkldnn.BaseMKLDNNHelper

org.deeplearning4j.nn.layers.mkldnn.MKLDNNLocalResponseNormalizationHelper

All Implemented Interfaces:

LayerHelper, LocalResponseNormalizationHelper

public class MKLDNNLocalResponseNormalizationHelper
extends BaseMKLDNNHelper
implements LocalResponseNormalizationHelper

MKL-DNN Local response normalization helper

Author:

Alex Black

Field Summary

Fields

Modifier and Type	Field and Description
protected OpContext	context

Constructor Summary

Constructors

Constructor and Description
MKLDNNLocalResponseNormalizationHelper(DataType dataType)

Method Summary

Modifier and Type	Method and Description
INDArray	activate(INDArray x, boolean training, double k, double n, double alpha, double beta, LayerWorkspaceMgr workspaceMgr)
Pair<Gradient,INDArray>	backpropGradient(INDArray input, INDArray epsilon, double k, double n, double alpha, double beta, LayerWorkspaceMgr workspaceMgr)
boolean	checkSupported(double k, double n, double alpha, double beta)
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 org.deeplearning4j.nn.layers.mkldnn.BaseMKLDNNHelper

mklDnnEnabled

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

MKLDNNLocalResponseNormalizationHelper

public MKLDNNLocalResponseNormalizationHelper(DataType dataType)

Method Detail

checkSupported

public boolean checkSupported(double k,
                              double n,
                              double alpha,
                              double beta)

Specified by:

checkSupported in interface LocalResponseNormalizationHelper

backpropGradient

public Pair<Gradient,INDArray> backpropGradient(INDArray input,
                                                INDArray epsilon,
                                                double k,
                                                double n,
                                                double alpha,
                                                double beta,
                                                LayerWorkspaceMgr workspaceMgr)

Specified by:

backpropGradient in interface LocalResponseNormalizationHelper

activate

public INDArray activate(INDArray x,
                         boolean training,
                         double k,
                         double n,
                         double alpha,
                         double beta,
                         LayerWorkspaceMgr workspaceMgr)

Specified by:

activate in interface LocalResponseNormalizationHelper

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.