Class MKLDNNSubsamplingHelper
- java.lang.Object
-
- org.deeplearning4j.nn.layers.mkldnn.MKLDNNSubsamplingHelper
-
- All Implemented Interfaces:
SubsamplingHelper
,LayerHelper
public class MKLDNNSubsamplingHelper extends Object implements SubsamplingHelper
-
-
Constructor Summary
Constructors Constructor Description MKLDNNSubsamplingHelper(DataType dataType)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description INDArray
activate(INDArray input, boolean training, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode, int[] dilation, CNN2DFormat format, LayerWorkspaceMgr workspaceMgr)
Pair<Gradient,INDArray>
backpropGradient(INDArray input, INDArray epsilon, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode, int[] dilation, CNN2DFormat format, 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.
-
-
-
Field Detail
-
context
protected OpContext context
-
-
Constructor Detail
-
MKLDNNSubsamplingHelper
public MKLDNNSubsamplingHelper(DataType dataType)
-
-
Method Detail
-
checkSupported
public boolean checkSupported()
- Specified by:
checkSupported
in interfaceLayerHelper
-
backpropGradient
public Pair<Gradient,INDArray> backpropGradient(INDArray input, INDArray epsilon, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode, int[] dilation, CNN2DFormat format, LayerWorkspaceMgr workspaceMgr)
- Specified by:
backpropGradient
in interfaceSubsamplingHelper
-
activate
public INDArray activate(INDArray input, boolean training, int[] kernel, int[] strides, int[] pad, PoolingType poolingType, ConvolutionMode convolutionMode, int[] dilation, CNN2DFormat format, LayerWorkspaceMgr workspaceMgr)
- Specified by:
activate
in interfaceSubsamplingHelper
-
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 interfaceLayerHelper
- Returns:
- Map of memory, may be null if none is used.
-
-