org.deeplearning4j.nn.layers

Class BaseCudnnHelper

java.lang.Object

org.deeplearning4j.nn.layers.BaseCudnnHelper

Direct Known Subclasses:

CudnnBatchNormalizationHelper, CudnnConvolutionHelper, CudnnLocalResponseNormalizationHelper, CudnnLSTMHelper, CudnnSubsamplingHelper

public abstract class BaseCudnnHelper
extends Object

Functionality shared by all cuDNN-based helpers.

Author:

saudet

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
protected static class	BaseCudnnHelper.CudnnContext
protected static class	BaseCudnnHelper.DataCache
protected static class	BaseCudnnHelper.TensorArray

Field Summary

Fields

Modifier and Type	Field and Description
protected org.bytedeco.javacpp.Pointer	alpha
protected org.bytedeco.javacpp.Pointer	beta
protected int	dataType
protected int	dataTypeSize
protected org.bytedeco.javacpp.SizeTPointer	sizeInBytes
protected static int	tensorFormat

Constructor Summary

Constructors

Constructor and Description
BaseCudnnHelper()

Method Summary

Modifier and Type	Method and Description
protected static void	checkCuda(int error)
protected static void	checkCudnn(int status)
boolean	checkSupported()

Methods inherited from class java.lang.Object

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

Field Detail

tensorFormat

protected static final int tensorFormat

See Also:

Constant Field Values

dataType

protected int dataType

dataTypeSize

protected int dataTypeSize

alpha

protected org.bytedeco.javacpp.Pointer alpha

beta

protected org.bytedeco.javacpp.Pointer beta

sizeInBytes

protected org.bytedeco.javacpp.SizeTPointer sizeInBytes

Constructor Detail

BaseCudnnHelper

public BaseCudnnHelper()

Method Detail

checkCuda

protected static void checkCuda(int error)

checkCudnn

protected static void checkCudnn(int status)

checkSupported

public boolean checkSupported()