Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_test.py137 [rnn_cell_impl.LSTMCell(1) for _ in range(2)])
148 [keras.layers.LSTMCell(1) for _ in range(2)])
185 rnn_cell_impl.LSTMCell,
360 self._assert_cell_builds(rnn_cell_impl.LSTMCell, f32, 5, 7, 3)
361 self._assert_cell_builds(rnn_cell_impl.LSTMCell, f64, 5, 7, 3)
444 cell = keras.layers.LSTMCell(output_shape)
483 [keras.layers.LSTMCell(2 * output_shape),
484 keras.layers.LSTMCell(output_shape)])
641 lstm_cell = rnn_cell_impl.LSTMCell(1, name="basic_lstm_cell")
655 rnn_cell_impl.LSTMCell(32, use_peepholes=True, cell_clip=True),
[all …]
Drnn_cell_test.py370 lstm = rnn_cell.LSTMCell(10)
377 lstm = rnn_cell.LSTMCell(10, dtype=dtype)
391 cell = rnn_cell.LSTMCell(
414 cell = rnn_cell.LSTMCell(
446 cell = rnn_cell.LSTMCell(
480 cell = rnn_cell.LSTMCell(
527 return rnn_cell.LSTMCell(
589 cell = rnn_cell.LSTMCell(
615 cell_notuple = rnn_cell.LSTMCell(
621 cell_tuple = rnn_cell.LSTMCell(
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dlstm.py47 def LSTMCell(weights, m_prev, c_prev, x, pad): function
113 m, c = LSTMCell(weights, m, c, x_seq[seq], pad_seq[seq])
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Dbasic_decoder_test.py53 cell = rnn_cell.LSTMCell(cell_depth)
139 cell = rnn_cell.LSTMCell(vocabulary_size)
215 cell = rnn_cell.LSTMCell(vocabulary_size)
284 cell = rnn_cell.LSTMCell(vocabulary_size)
374 cell = rnn_cell.LSTMCell(cell_depth)
527 cell = rnn_cell.LSTMCell(vocabulary_size)
608 cell = rnn_cell.LSTMCell(vocabulary_size)
Dbasic_decoder_v2_test.py57 cell = rnn_cell.LSTMCell(cell_depth)
142 cell = rnn_cell.LSTMCell(vocabulary_size)
219 cell = rnn_cell.LSTMCell(vocabulary_size)
291 cell = rnn_cell.LSTMCell(vocabulary_size)
381 cell = rnn_cell.LSTMCell(cell_depth)
533 cell = rnn_cell.LSTMCell(vocabulary_size)
612 cell = rnn_cell.LSTMCell(vocabulary_size)
Ddecoder_v2_test.py55 cell = rnn_cell.LSTMCell(cell_depth)
129 cell = rnn_cell.LSTMCell(cell_depth)
Ddecoder_test.py53 cell = rnn_cell.LSTMCell(cell_depth)
132 cell = rnn_cell.LSTMCell(cell_depth)
Dattention_wrapper_test.py93 cell = wrapper.AttentionWrapper(rnn_cell.LSTMCell(num_units), mechanism)
196 cell = rnn_cell.LSTMCell(cell_depth)
300 cell = rnn_cell.LSTMCell(num_units)
429 cell = rnn_cell.LSTMCell(num_units)
1024 rnn_cell.LSTMCell(2),
Dattention_wrapper_v2_test.py307 cell = keras.layers.LSTMCell(cell_depth,
408 cell = keras.layers.LSTMCell(self.units, recurrent_activation="sigmoid")
435 cell = keras.layers.LSTMCell(self.units, recurrent_activation="sigmoid")
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/g3doc/
DREADME.md10 TensorFlow, LSTM ops are expressed as a "cell" (e.g., `tf.nn.rnn_cell.LSTMCell`,
37 First step is replacing `tf.nn.rnn_cell.LSTMCell` with
43 `tf.lite.experimental.nn.dynamic_rnn` are just normal `tf.nn.rnn_cell.LSTMCell`
74 - tf.nn.rnn_cell.LSTMCell(
186 # (OpHinted LSTMCell).
377 * `num_unit_shards` & `num_proj_shards` in LSTMCell are not supported as
/external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/
Dgrid_rnn_cell.py130 rnn.LSTMCell, num_units=num_units, state_is_tuple=state_is_tuple)
443 return rnn.LSTMCell(
476 return rnn.LSTMCell(
513 return rnn.LSTMCell(
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Drnn_test.py61 rnn_cell.LSTMCell(
68 rnn_cell.LSTMCell(
240 rnn_cell.LSTMCell(
247 rnn_cell.LSTMCell(
Dlstm_ops_test.py143 basic_cell = rnn_cell.LSTMCell(
265 cell = rnn_cell.LSTMCell(10)
266 pcell = rnn_cell.LSTMCell(10, use_peepholes=True)
369 rnn_cell.LSTMCell(2, use_peepholes=True, state_is_tuple=True)
/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.ops.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\'>"
/external/tensorflow/tensorflow/examples/saved_model/integration_tests/
Dexport_rnn_cell.py37 root.rnn_cell = tf.keras.layers.LSTMCell(units=10, recurrent_initializer=None)
Dexport_text_rnn_model.py41 self._lstm_cell = tf.keras.layers.LSTMCell(units=state_size)
/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.LSTMCell\'>"
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent_test.py603 cells = [keras.layers.LSTMCell(1),
604 keras.layers.LSTMCell(1)]
625 cells = [keras.layers.LSTMCell(1),
626 keras.layers.LSTMCell(1)]
685 keras.layers.LSTMCell]:
754 keras.layers.LSTMCell],
805 cells = [keras.layers.LSTMCell(3),
806 keras.layers.LSTMCell(6)]
1195 keras.layers.LSTMCell,
1210 rnn_cell.LSTMCell,
Dlstm_test.py140 lstm_cells = [keras.layers.LSTMCell(10), keras.layers.LSTMCell(5)]
/external/tensorflow/tensorflow/contrib/recurrent/python/kernel_tests/
Dfunctional_rnn_test.py40 subcells = [rnn_cell_impl.LSTMCell(cell_size) for cell_size in cell_sizes]
56 'lstm': (rnn_cell_impl.LSTMCell, [_NUM_UNITS]),
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drnn_common.py57 'lstm': contrib_rnn.LSTMCell,
/external/tensorflow/tensorflow/contrib/timeseries/examples/
Dlstm.py94 self._lstm_cell = tf.nn.rnn_cell.LSTMCell(num_units=self._num_units)
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Drnn_cells.py172 class MaskedLSTMCell(tf_rnn.LSTMCell):

12