| Modifier and Type | Class and Description |
|---|---|
class |
CudnnConvolutionHelper
cuDNN-based helper for the convolution layer.
|
| Modifier and Type | Class and Description |
|---|---|
class |
CudnnSubsamplingHelper
cuDNN-based helper for the subsampling layer.
|
| Modifier and Type | Class and Description |
|---|---|
class |
CudnnDropoutHelper
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
|
| Modifier and Type | Class and Description |
|---|---|
class |
CudnnBatchNormalizationHelper
cuDNN-based helper for the batch normalization layer.
|
class |
CudnnLocalResponseNormalizationHelper
cuDNN-based helper for the local response normalization layer.
|
| Modifier and Type | Class and Description |
|---|---|
class |
CudnnLSTMHelper
cuDNN-based helper for the recurrent LSTM layer (no peephole connections).
|
Copyright © 2018. All rights reserved.