Welcome to mirror list, hosted at ThFree Co, Russian Federation.

RNNReLU.lua - github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 377e3f288c0a88c8fa0bef89b4beeb7d15834034 (plain)
1
2
3
4
5
6
7
local RNNReLU, parent = torch.class('cudnn.RNNReLU', 'cudnn.RNN')

function RNNReLU:__init(inputSize, hiddenSize, numLayers, batchFirst, dropout, rememberStates)
    parent.__init(self,inputSize, hiddenSize, numLayers, batchFirst, dropout, rememberStates)
    self.mode = 'CUDNN_RNN_RELU'
    self:reset()
end