Home
last modified time | relevance | path

Searched refs:rnn_cell (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
DBUILD31 "//tensorflow/python:rnn_cell",
37 name = "rnn_cell",
38 srcs = ["rnn_cell.py"],
44 "//tensorflow/python:rnn_cell",
62 ":rnn_cell",
85 ":rnn_cell",
108 ":rnn_cell",
131 ":rnn_cell",
147 ":rnn_cell",
/external/tensorflow/tensorflow/python/keras/layers/
Drnn_cell_wrapper_v2_test.py120 rnn_cell = wrapper(base_cell)
121 rnn_layer = layers.RNN(rnn_cell)
129 self.assertLen(rnn_cell.weights, 3)
130 self.assertCountEqual([v.name for v in rnn_cell.weights], expected_weights)
131 self.assertCountEqual([v.name for v in rnn_cell.trainable_variables],
133 self.assertCountEqual([v.name for v in rnn_cell.non_trainable_variables],
135 self.assertCountEqual([v.name for v in rnn_cell.cell.weights],
146 rnn_cell = wrapper(base_cell)
149 _ = rnn_cell(inputs, [state, state])
Drecurrent.py3023 def _caching_device(rnn_cell): argument
3043 if not getattr(rnn_cell, '_enable_caching_device', False):
3057 if (rnn_cell._dtype_policy.compute_dtype !=
3058 rnn_cell._dtype_policy.variable_dtype):
3071 def _config_for_enable_caching_device(rnn_cell): argument
3086 if rnn_cell._enable_caching_device != default_enable_caching_device:
3087 return {'enable_caching_device': rnn_cell._enable_caching_device}
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dunidirectional_sequence_rnn.py50 rnn_cell = tf.lite.experimental.nn.TfLiteRNNCell(parameters["units"])
52 rnn_cell, input_value, dtype=tf.float32, time_major=True)
60 rnn_cell = tf.lite.experimental.nn.TfLiteRNNCell(parameters["units"])
61 outs, _ = tf.nn.static_rnn(rnn_cell, input_values, dtype=tf.float32)
Dunfused_gru.py43 cell_fw = tf.compat.v1.nn.rnn_cell.GRUCell(parameters["units"])
44 cell_bw = tf.compat.v1.nn.rnn_cell.GRUCell(parameters["units"])
Dlstm.py61 lstm_cell = tf.compat.v1.nn.rnn_cell.BasicLSTMCell(
/external/tensorflow/tensorflow/python/keras/integration_test/
Dlegacy_rnn_test.py29 self._cell = tf.nn.rnn_cell.MultiRNNCell(
30 [tf.nn.rnn_cell.LSTMCell(1) for _ in range(2)])
302 cell = tf.nn.rnn_cell.BasicRNNCell(output_shape)
313 tf.nn.rnn_cell.LSTMCell(32, use_peepholes=True, cell_clip=True),
314 tf.nn.rnn_cell.BasicLSTMCell(32, dtype=tf.float32),
315 tf.nn.rnn_cell.BasicRNNCell(32, activation="relu", dtype=tf.float32),
316 tf.nn.rnn_cell.GRUCell(32, dtype=tf.float32)
336 "BasicRNNCell": tf.nn.rnn_cell.BasicRNNCell,
337 "GRUCell": tf.nn.rnn_cell.GRUCell,
338 "LSTMCell": tf.nn.rnn_cell.LSTMCell,
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py44 from tensorflow.python.ops import rnn_cell
59 class Plus1RNNCell(rnn_cell.RNNCell):
74 class DummyMultiDimensionalLSTM(rnn_cell.RNNCell):
109 class NestedRNNCell(rnn_cell.RNNCell):
242 full_dropout_cell = rnn_cell.DropoutWrapper(
373 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(
[all …]
/external/tensorflow/tensorflow/python/
D__init__.py129 from tensorflow.python.ops import rnn_cell
154 nn.rnn_cell = rnn_cell
/external/tensorflow/tensorflow/python/profiler/internal/
Dmodel_analyzer_testlib.py29 from tensorflow.python.ops import rnn_cell
65 cell = rnn_cell.BasicRNNCell(16)
DBUILD35 "//tensorflow/python:rnn_cell",
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.rnn_cell.-l-s-t-m-state-tuple.pbtxt1 path: "tensorflow.nn.rnn_cell.LSTMStateTuple"
Dtensorflow.nn.rnn_cell.pbtxt1 path: "tensorflow.nn.rnn_cell"
Dtensorflow.nn.rnn_cell.-r-n-n-cell.pbtxt1 path: "tensorflow.nn.rnn_cell.RNNCell"
Dtensorflow.nn.rnn_cell.-residual-wrapper.pbtxt1 path: "tensorflow.nn.rnn_cell.ResidualWrapper"
Dtensorflow.nn.rnn_cell.-multi-r-n-n-cell.pbtxt1 path: "tensorflow.nn.rnn_cell.MultiRNNCell"
Dtensorflow.nn.rnn_cell.-l-s-t-m-cell.pbtxt1 path: "tensorflow.nn.rnn_cell.LSTMCell"
Dtensorflow.nn.rnn_cell.-dropout-wrapper.pbtxt1 path: "tensorflow.nn.rnn_cell.DropoutWrapper"
Dtensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt3 is_instance: "<class \'tensorflow.lite.experimental.examples.lstm.rnn_cell.TFLiteLSTMCell\'>"
Dtensorflow.nn.rnn_cell.-basic-r-n-n-cell.pbtxt1 path: "tensorflow.nn.rnn_cell.BasicRNNCell"
Dtensorflow.lite.experimental.nn.-tf-lite-r-n-n-cell.pbtxt3 is_instance: "<class \'tensorflow.lite.experimental.examples.lstm.rnn_cell.TfLiteRNNCell\'>"
/external/tensorflow/tensorflow/python/keras/tests/
Dintegration_test.py31 from tensorflow.python.keras.layers.legacy_rnn import rnn_cell_impl as rnn_cell unknown
224 model.add(keras.layers.RNN(rnn_cell.LSTMCell(5), return_sequences=True,
226 model.add(keras.layers.RNN(rnn_cell.GRUCell(y_train.shape[-1],
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dgradients_test.py42 from tensorflow.python.ops import rnn_cell
77 cell = rnn_cell.BasicLSTMCell(state_size)
94 cell = rnn_cell.BasicLSTMCell(state_size)
190 cell = rnn_cell.BasicLSTMCell(state_size)
204 loop_init_state = rnn_cell.LSTMStateTuple(
/external/tensorflow/tensorflow/lite/g3doc/convert/
Drnn.md114 …b.com/tensorflow/lingvo/blob/91a4609dbc2579748a95110eda59c66d17c594c5/lingvo/core/rnn_cell.py#L228)
118 ….com/tensorflow/lingvo/blob/91a4609dbc2579748a95110eda59c66d17c594c5/lingvo/core/rnn_cell.py#L1173)
153 ….com/tensorflow/lingvo/blob/91a4609dbc2579748a95110eda59c66d17c594c5/lingvo/core/rnn_cell.py#L228).
/external/tensorflow/tensorflow/python/tools/api/generator/
Dapi_init_files_v1.bzl62 "nn/rnn_cell/__init__.py",

12