Home
last modified time | relevance | path

Searched refs:LSTMCell (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/
Drnn_cell_impl.py46 LSTMCell = rnn_cell_impl.LSTMCell variable
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py373 lstm = rnn_cell.LSTMCell(10)
380 lstm = rnn_cell.LSTMCell(10, dtype=dtype)
394 cell = rnn_cell.LSTMCell(
417 cell = rnn_cell.LSTMCell(
449 cell = rnn_cell.LSTMCell(
483 cell = rnn_cell.LSTMCell(
530 return rnn_cell.LSTMCell(
592 cell = rnn_cell.LSTMCell(
618 cell_notuple = rnn_cell.LSTMCell(
624 cell_tuple = rnn_cell.LSTMCell(
[all …]
Drnn_test.py156 rnn_cell_impl.LSTMCell,
331 self._assert_cell_builds(rnn_cell_impl.LSTMCell, f32, 5, 7, 3)
332 self._assert_cell_builds(rnn_cell_impl.LSTMCell, f64, 5, 7, 3)
349 lstm_cell = rnn_cell_impl.LSTMCell(1, name="basic_lstm_cell")
365 cell = rnn_cell_impl.LSTMCell(
387 cell = rnn_cell_impl.LSTMCell(
498 cell = rnn_cell_impl.LSTMCell(
564 cell = rnn_cell_impl.LSTMCell(
631 cell = rnn_cell_impl.LSTMCell(
/external/tensorflow/tensorflow/python/keras/integration_test/
Dlegacy_rnn_test.py30 [tf.nn.rnn_cell.LSTMCell(1) for _ in range(2)])
41 [tf.keras.layers.LSTMCell(1) for _ in range(2)])
132 cell = tf.keras.layers.LSTMCell(output_shape)
170 [tf.keras.layers.LSTMCell(2 * output_shape),
171 tf.keras.layers.LSTMCell(output_shape)])
313 tf.nn.rnn_cell.LSTMCell(32, use_peepholes=True, cell_clip=True),
338 "LSTMCell": tf.nn.rnn_cell.LSTMCell,
/external/tensorflow/tensorflow/python/keras/layers/
Drnn_cell_wrapper_v2_test.py157 cell = rnn_cell_impl.LSTMCell(10)
164 cell = layers.LSTMCell(10)
174 cell = layers.LSTMCell(10)
227 cell = layers.LSTMCell(10)
D__init__.py227 from tensorflow.python.keras.layers.recurrent_v2 import LSTMCell
231 from tensorflow.python.keras.layers.recurrent import LSTMCell as LSTMCellV1
235 LSTMCellV2 = LSTMCell
240 from tensorflow.python.keras.layers.recurrent import LSTMCell
244 from tensorflow.python.keras.layers.recurrent_v2 import LSTMCell as LSTMCellV2
248 LSTMCellV1 = LSTMCell
Drecurrent_test.py583 cells = [keras.layers.LSTMCell(1),
584 keras.layers.LSTMCell(1)]
605 cells = [keras.layers.LSTMCell(1),
606 keras.layers.LSTMCell(1)]
705 keras.layers.LSTMCell,
778 keras.layers.LSTMCell],
855 cells = [keras.layers.LSTMCell(3),
856 keras.layers.LSTMCell(6)]
914 cell = keras.layers.LSTMCell(32)
1283 keras.layers.LSTMCell,
[all …]
Drnn_cell_wrapper_v2.py104 if isinstance(self.cell, recurrent.LSTMCell):
Dlstm_test.py170 lstm_cells = [keras.layers.LSTMCell(10), keras.layers.LSTMCell(5)]
187 cell = keras.layers.LSTMCell(5)
Drecurrent_v2.py843 class LSTMCell(recurrent.LSTMCell): class
937 super(LSTMCell, self).__init__(
/external/tensorflow/tensorflow/compiler/tests/
Dlstm.py50 def LSTMCell(weights, m_prev, c_prev, x, pad): function
116 m, c = LSTMCell(weights, m, c, x_seq[seq], pad_seq[seq])
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.rnn_cell.pbtxt24 name: "LSTMCell"
Dtensorflow.nn.rnn_cell.-l-s-t-m-cell.pbtxt1 path: "tensorflow.nn.rnn_cell.LSTMCell"
3 is_instance: "<class \'tensorflow.python.keras.layers.legacy_rnn.rnn_cell_impl.LSTMCell\'>"
Dtensorflow.keras.layers.-l-s-t-m-cell.pbtxt1 path: "tensorflow.keras.layers.LSTMCell"
3 is_instance: "<class \'tensorflow.python.keras.layers.recurrent.LSTMCell\'>"
Dtensorflow.keras.experimental.-peephole-l-s-t-m-cell.pbtxt4 is_instance: "<class \'tensorflow.python.keras.layers.recurrent.LSTMCell\'>"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.layers.-l-s-t-m-cell.pbtxt1 path: "tensorflow.keras.layers.LSTMCell"
3 is_instance: "<class \'tensorflow.python.keras.layers.recurrent_v2.LSTMCell\'>"
4 is_instance: "<class \'tensorflow.python.keras.layers.recurrent.LSTMCell\'>"
Dtensorflow.keras.experimental.-peephole-l-s-t-m-cell.pbtxt4 is_instance: "<class \'tensorflow.python.keras.layers.recurrent.LSTMCell\'>"
Dtensorflow.keras.layers.pbtxt244 name: "LSTMCell"
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_variable_test.py528 cell_fw = rnn_cell_impl.LSTMCell(300)
529 cell_bw = rnn_cell_impl.LSTMCell(300)
/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/
Drnn_cell_impl.py824 class LSTMCell(LayerRNNCell): class
913 super(LSTMCell, self).__init__(
1105 base_config = super(LSTMCell, self).get_config()
/external/tensorflow/tensorflow/python/keras/benchmarks/layer_benchmarks/
Dlayer_benchmarks_test.py134 {"cell": tf.keras.layers.LSTMCell(1)}, {"input_shape": (1, 1, 1)}, 100),
/external/tensorflow/tensorflow/lite/schema/
Dschema_v0.fbs192 // An implementation of TensorFlow LSTMCell and CoupledInputForgetGateLSTMCell
Dschema_v2.fbs213 // An implementation of TensorFlow LSTMCell and CoupledInputForgetGateLSTMCell
Dschema_v1.fbs205 // An implementation of TensorFlow LSTMCell and CoupledInputForgetGateLSTMCell
/external/tensorflow/tensorflow/python/keras/tests/
Dmodel_architectures.py50 layer = keras.layers.RNN([keras.layers.LSTMCell(2) for _ in range(3)])

12