diff options
Diffstat (limited to 'init.lua')
-rw-r--r-- | init.lua | 5 |
1 files changed, 5 insertions, 0 deletions
@@ -123,6 +123,11 @@ require('cudnn.VolumetricBatchNormalization') require('cudnn.SpatialCrossEntropyCriterion') require('cudnn.TemporalConvolution') require('cudnn.RNN') +require('cudnn.RNNTanh') +require('cudnn.RNNReLU') +require('cudnn.BLSTM') +require('cudnn.LSTM') +require('cudnn.GRU') require('cudnn.functional') require('cudnn.convert') |