| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| gru_ops.h | 30 struct GRUCell {
31 GRUCell(const int batch_size, const int input_size, const int cell_size)
69 struct GRUBlockCellFprop : public GRUCell {
72 : GRUCell(batch_size, input_size, cell_size) {}
125 struct GRUBlockCellBprop : public GRUCell {
128 : GRUCell(batch_size, input_size, cell_size) {}
|
| /external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/kernel_tests/ |
| seq2seq_test.py | 55 rnn_cell.GRUCell(2), inp, dtype=dtypes.float32)
57 cell = core_rnn_cell.OutputProjectionWrapper(rnn_cell.GRUCell(2), 4)
73 cell = core_rnn_cell.OutputProjectionWrapper(rnn_cell.GRUCell(2), 4)
89 cell = core_rnn_cell.OutputProjectionWrapper(rnn_cell.GRUCell(2), 4)
322 cell_fn = lambda: rnn_cell.GRUCell(2)
347 cell_fn = lambda: rnn_cell.GRUCell(2)
373 cell_fn = lambda: rnn_cell.GRUCell(2)
397 cell_fn = lambda: rnn_cell.GRUCell(2)
486 cell_fn = lambda: rnn_cell.GRUCell(2)
788 # [rnn_cell.GRUCell(24) for _ in range(2)]
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| gru_ops_test.py | 79 output = rnn_cell.GRUCell(cell_size)(x, h)
129 cell = rnn_cell.GRUCell(cell_size)
193 output = rnn_cell.GRUCell(cell_size)(x, h)
259 cell = rnn_cell.GRUCell(cell_size)
332 #### Benchmarking GRUBlockCell vs GRUCell.
341 """Benchmark training speed between GRUBlockCell vs GRUCell."""
360 cell = rnn_cell.GRUCell(cell_size)
416 """Benchmark inference speed between GRUBlockCell vs GRUCell."""
433 cell = rnn_cell.GRUCell(cell_size)
473 """Benchmark single bprop step speed between GRUBlockCell vs GRUCell.""
[all...] |
| rnn_cell_test.py | 241 rnn_cell_impl.GRUCell(3), 2)
259 rnn_cell_impl.GRUCell(3), num_proj=3)
277 rnn_cell_impl.GRUCell(2), embedding_classes=3, embedding_size=2)
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ops/ |
| seq2seq_ops_test.py | 85 cell = rnn_cell.GRUCell(2)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| rnn_common.py | 58 'gru': contrib_rnn.GRUCell,}
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| rnn_test.py | 183 rnn_cell_impl.GRUCell,
358 self._assert_cell_builds(rnn_cell_impl.GRUCell, f32, 5, 7, 3)
359 self._assert_cell_builds(rnn_cell_impl.GRUCell, f64, 5, 7, 3)
410 cell = keras.layers.GRUCell(output_shape)
658 rnn_cell_impl.GRUCell(32, dtype=dtypes.float32)
674 # keras.LSTMCell and GRUCell.
679 "GRUCell": rnn_cell_impl.GRUCell,
[all...] |
| rnn_cell_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| models.py | 337 cell_fn = contrib_rnn.GRUCell
|
| /external/tensorflow/tensorflow/python/keras/ |
| integration_test.py | 152 model.add(keras.layers.RNN(rnn_cell.GRUCell(y_train.shape[-1],
|
| /external/tensorflow/tensorflow/contrib/recurrent/python/kernel_tests/ |
| functional_rnn_test.py | 55 'gru': (rnn_cell_impl.GRUCell, [_NUM_UNITS]),
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| __init__.py | 152 from tensorflow.python.keras.layers.recurrent import GRUCell
|
| recurrent.py | [all...] |
| recurrent_test.py | 464 cells = [keras.layers.recurrent.GRUCell(8),
479 # Test GRUCell reset_after property.
482 cells = [keras.layers.recurrent.GRUCell(32, reset_after=True)]
684 keras.layers.GRUCell,
753 cell=[keras.layers.SimpleRNNCell, keras.layers.GRUCell,
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| rnn_cell_impl.py | 480 @tf_export(v1=["nn.rnn_cell.GRUCell"])
481 class GRUCell(LayerRNNCell):
506 @deprecated(None, "This class is equivalent as tf.keras.layers.GRUCell,"
517 super(GRUCell, self).__init__(
605 base_config = super(GRUCell, self).get_config()
[all...] |
| /external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/ |
| grid_rnn_cell.py | 553 cell_fn=lambda n: rnn.GRUCell(num_units=n),
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| cudnn_rnn_ops.py | 76 class CudnnCompatibleGRUCell(rnn_cell_impl.GRUCell):
77 r"""Cudnn Compatible GRUCell.
79 A GRU impl akin to `tf.nn.rnn_cell.GRUCell` to use along with
100 Other GRU (see `tf.nn.rnn_cell.GRUCell` and `tf.contrib.rnn.GRUBlockCell`):
[all...] |
