Searched refs:lstm_cell (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | rnn_cells_test.py | 48 lstm_cell = rnn_cells.MaskedBasicLSTMCell(self.dim)
49 lstm_cell(inputs, state)
72 lstm_cell = rnn_cells.MaskedLSTMCell(self.dim)
73 lstm_cell(inputs, state)
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | strip_pruning_vars_test.py | 112 def lstm_cell(): function
117 [lstm_cell() for _ in range(number_of_layers)], state_is_tuple=True)
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | rnn_cell_test.py | 782 lstm_cell = rnn_cell.BasicLSTMCell(
787 lstm_cell, 0, state_is_tuple=state_is_tuple)
791 lstm_cell, -1, state_is_tuple=state_is_tuple)
793 lstm_cell = rnn_cell.BasicLSTMCell(num_units, state_is_tuple=True)
798 lstm_cell, 4, state_is_tuple=False)
810 lstm_cell = rnn_cell.BasicLSTMCell(
813 lstm_cell, attn_length, state_is_tuple=state_is_tuple)
861 lstm_cell = rnn_cell.BasicLSTMCell(
864 lstm_cell, attn_length, state_is_tuple=state_is_tuple)
941 lstm_cell = rnn_cell.BasicLSTMCell(
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_test.py | 641 lstm_cell = rnn_cell_impl.LSTMCell(1, name="basic_lstm_cell")
642 lstm_cell(array_ops.ones([1, 1]),
643 state=lstm_cell.get_initial_state(inputs=None,
646 self.evaluate([v.initializer for v in lstm_cell.variables])
649 self.assertAllEqual([10.] * 4, self.evaluate(lstm_cell._bias))
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 3028 lstm_cell = tf.contrib.rnn.BasicLSTMCell(
3031 lstm_cell, inputs_after_split, dtype=tf.float32)
4385 lstm_cell = tf.lite.experimental.nn.TFLiteLSTMCell(
4389 lstm_cell, input_value, dtype=tf.float32, time_major=True)
4397 lstm_cell = tf.lite.experimental.nn.TFLiteLSTMCell(
4399 outs, _ = tf.nn.static_rnn(lstm_cell, input_values, dtype=tf.float32)
|
| /external/tensorflow/tensorflow/examples/udacity/ |
| D | 6_lstm.ipynb | 545 " def lstm_cell(i, o, state):\n",
569 " output, state = lstm_cell(i, output, state)\n",
601 " sample_output, sample_state = lstm_cell(\n",
|