Searched refs:MaskedBasicLSTMCell (Results 1 – 4 of 4) sorted by relevance
37 class MaskedBasicLSTMCell(tf_rnn.BasicLSTMCell): class83 super(MaskedBasicLSTMCell, self).__init__(93 super(MaskedBasicLSTMCell, self).build(inputs_shape)
48 lstm_cell = rnn_cells.MaskedBasicLSTMCell(self.dim)
25 from tensorflow.contrib.model_pruning.python.layers.rnn_cells import MaskedBasicLSTMCell
113 return rnn_cells.MaskedBasicLSTMCell(