org.deeplearning4j.nn.layers.mkldnn

Class MKLDNNLSTMHelper

java.lang.Object

org.deeplearning4j.nn.layers.mkldnn.MKLDNNLSTMHelper

All Implemented Interfaces:

LayerHelper, LSTMHelper

public class MKLDNNLSTMHelper
extends Object
implements LSTMHelper

Constructor Summary

Constructors

Constructor and Description
MKLDNNLSTMHelper()

Method Summary

Modifier and Type	Method and Description
FwdPassReturn	activate(Layer layer, NeuralNetConfiguration conf, IActivation gateActivationFn, INDArray input, INDArray recurrentWeights, INDArray inputWeights, INDArray biases, boolean training, INDArray prevOutputActivations, INDArray prevMemCellState, boolean forBackprop, boolean forwards, String inputWeightKey, INDArray maskArray, boolean hasPeepholeConnections, LayerWorkspaceMgr workspaceMgr)
Pair<Gradient,INDArray>	backpropGradient(NeuralNetConfiguration conf, IActivation gateActivationFn, INDArray input, INDArray recurrentWeights, INDArray inputWeights, INDArray epsilon, boolean truncatedBPTT, int tbpttBackwardLength, FwdPassReturn fwdPass, boolean forwards, String inputWeightKey, String recurrentWeightKey, String biasWeightKey, Map<String,INDArray> gradientViews, INDArray maskArray, boolean hasPeepholeConnections, LayerWorkspaceMgr workspaceMgr)
boolean	checkSupported(IActivation gateActivationFn, IActivation activationFn, boolean hasPeepholeConnections)
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

Constructor Detail

MKLDNNLSTMHelper

public MKLDNNLSTMHelper()

Method Detail

checkSupported

public boolean checkSupported(IActivation gateActivationFn,
                              IActivation activationFn,
                              boolean hasPeepholeConnections)

Specified by:

checkSupported in interface LSTMHelper

backpropGradient

public Pair<Gradient,INDArray> backpropGradient(NeuralNetConfiguration conf,
                                                IActivation gateActivationFn,
                                                INDArray input,
                                                INDArray recurrentWeights,
                                                INDArray inputWeights,
                                                INDArray epsilon,
                                                boolean truncatedBPTT,
                                                int tbpttBackwardLength,
                                                FwdPassReturn fwdPass,
                                                boolean forwards,
                                                String inputWeightKey,
                                                String recurrentWeightKey,
                                                String biasWeightKey,
                                                Map<String,INDArray> gradientViews,
                                                INDArray maskArray,
                                                boolean hasPeepholeConnections,
                                                LayerWorkspaceMgr workspaceMgr)

Specified by:

backpropGradient in interface LSTMHelper

activate

public FwdPassReturn activate(Layer layer,
                              NeuralNetConfiguration conf,
                              IActivation gateActivationFn,
                              INDArray input,
                              INDArray recurrentWeights,
                              INDArray inputWeights,
                              INDArray biases,
                              boolean training,
                              INDArray prevOutputActivations,
                              INDArray prevMemCellState,
                              boolean forBackprop,
                              boolean forwards,
                              String inputWeightKey,
                              INDArray maskArray,
                              boolean hasPeepholeConnections,
                              LayerWorkspaceMgr workspaceMgr)

Specified by:

activate in interface LSTMHelper

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.