Searched refs:MaskedLSTMCell (Results 1 – 4 of 4) sorted by relevance
172 class MaskedLSTMCell(tf_rnn.LSTMCell): class226 super(MaskedLSTMCell, self).__init__(242 super(MaskedLSTMCell, self).build(inputs_shape)
72 lstm_cell = rnn_cells.MaskedLSTMCell(self.dim)
26 from tensorflow.contrib.model_pruning.python.layers.rnn_cells import MaskedLSTMCell
42 * [rnn_cells.MaskedLSTMCell](https://github.com/tensorflow/tensorflow/tree/master/tensorflow/cont…