- checkCuda(int) - Static method in class org.deeplearning4j.nn.layers.BaseCudnnHelper
-
- checkCudnn(int) - Static method in class org.deeplearning4j.nn.layers.BaseCudnnHelper
-
- checkSupported() - Method in class org.deeplearning4j.nn.layers.BaseCudnnHelper
-
- checkSupported(double) - Method in class org.deeplearning4j.nn.layers.normalization.CudnnBatchNormalizationHelper
-
- checkSupported(double, double, double, double) - Method in class org.deeplearning4j.nn.layers.normalization.CudnnLocalResponseNormalizationHelper
-
- checkSupported(IActivation, IActivation, boolean) - Method in class org.deeplearning4j.nn.layers.recurrent.CudnnLSTMHelper
-
- createHandles() - Method in class org.deeplearning4j.nn.layers.BaseCudnnHelper.CudnnContext
-
- CudnnBatchNormalizationHelper - Class in org.deeplearning4j.nn.layers.normalization
-
cuDNN-based helper for the batch normalization layer.
- CudnnBatchNormalizationHelper() - Constructor for class org.deeplearning4j.nn.layers.normalization.CudnnBatchNormalizationHelper
-
- CudnnContext() - Constructor for class org.deeplearning4j.nn.layers.BaseCudnnHelper.CudnnContext
-
- CudnnContext(BaseCudnnHelper.CudnnContext) - Constructor for class org.deeplearning4j.nn.layers.BaseCudnnHelper.CudnnContext
-
- CudnnConvolutionHelper - Class in org.deeplearning4j.nn.layers.convolution
-
cuDNN-based helper for the convolution layer.
- CudnnConvolutionHelper() - Constructor for class org.deeplearning4j.nn.layers.convolution.CudnnConvolutionHelper
-
- CudnnConvolutionHelper.CudnnForwardArgs - Class in org.deeplearning4j.nn.layers.convolution
-
- CudnnDropoutHelper - Class in org.deeplearning4j.nn.layers.dropout
-
CuDNN dropout helper
Note that for repeatability between calls (for example, for gradient checks), we need to do two things:
(a) set the ND4J RNG seed
(b) clear the rngStates field
- CudnnDropoutHelper() - Constructor for class org.deeplearning4j.nn.layers.dropout.CudnnDropoutHelper
-
- CudnnForwardArgs() - Constructor for class org.deeplearning4j.nn.layers.convolution.CudnnConvolutionHelper.CudnnForwardArgs
-
- CudnnLocalResponseNormalizationHelper - Class in org.deeplearning4j.nn.layers.normalization
-
cuDNN-based helper for the local response normalization layer.
- CudnnLocalResponseNormalizationHelper() - Constructor for class org.deeplearning4j.nn.layers.normalization.CudnnLocalResponseNormalizationHelper
-
- CudnnLSTMHelper - Class in org.deeplearning4j.nn.layers.recurrent
-
cuDNN-based helper for the recurrent LSTM layer (no peephole connections).
- CudnnLSTMHelper() - Constructor for class org.deeplearning4j.nn.layers.recurrent.CudnnLSTMHelper
-
- CudnnSubsamplingHelper - Class in org.deeplearning4j.nn.layers.convolution.subsampling
-
cuDNN-based helper for the subsampling layer.
- CudnnSubsamplingHelper() - Constructor for class org.deeplearning4j.nn.layers.convolution.subsampling.CudnnSubsamplingHelper
-