cudnn.torch/RNNReLU.lua at master · soumith/cudnn.torch · GitHub

Sitelet https://github.com/soumith/cudnn.torch/blob/master/RNNReLU.lua

7 lines (6 loc) · 310 Bytes

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()