Home
last modified time | relevance | path

Searched refs:GRUCell (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dgru_ops.h30 struct GRUCell { struct
31 GRUCell(const int batch_size, const int input_size, const int cell_size) in GRUCell() function
69 struct GRUBlockCellFprop : public GRUCell { argument
72 : GRUCell(batch_size, input_size, cell_size) {} in GRUBlockCellFprop()
125 struct GRUBlockCellBprop : public GRUCell {
128 : GRUCell(batch_size, input_size, cell_size) {} in GRUBlockCellBprop()
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Dgru_ops_test.py79 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)
360 cell = rnn_cell.GRUCell(cell_size)
433 cell = rnn_cell.GRUCell(cell_size)
484 output = rnn_cell.GRUCell(cell_size)(array_ops.identity(x),
Drnn_cell_test.py241 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)
1335 base_cell = rnn_cell.GRUCell(3)
1340 rnn_cell.GRUCell(3), carry_bias_init=-100.0)
2007 [rnn_cell.GRUCell(2) for _ in range(2)],
2011 [rnn_cell.GRUCell(2) for _ in range(2)],
/external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/kernel_tests/
Dseq2seq_test.py55 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)
854 [rnn_cell.GRUCell(24) for _ in range(2)], state_is_tuple=True)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.rnn_cell.pbtxt20 name: "GRUCell"
Dtensorflow.keras.layers.-g-r-u-cell.pbtxt1 path: "tensorflow.keras.layers.GRUCell"
3 is_instance: "<class \'tensorflow.python.keras.layers.recurrent.GRUCell\'>"
Dtensorflow.nn.rnn_cell.-g-r-u-cell.pbtxt1 path: "tensorflow.nn.rnn_cell.GRUCell"
3 is_instance: "<class \'tensorflow.python.ops.rnn_cell_impl.GRUCell\'>"
Dtensorflow.keras.layers.pbtxt164 name: "GRUCell"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.layers.-g-r-u-cell.pbtxt1 path: "tensorflow.keras.layers.GRUCell"
3 is_instance: "<class \'tensorflow.python.keras.layers.recurrent.GRUCell\'>"
Dtensorflow.keras.layers.pbtxt156 name: "GRUCell"
/external/tensorflow/tensorflow/contrib/learn/python/learn/ops/
Dseq2seq_ops_test.py85 cell = rnn_cell.GRUCell(2)
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_test.py183 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)
679 "GRUCell": rnn_cell_impl.GRUCell,
Drnn_cell_test.py945 cell = rnn_cell.GRUCell(30)
1981 cell = rnn_cell.GRUCell(num_units=num_units)
2029 cell = rnn_cell.GRUCell(num_units=num_units)
2540 g, _ = rnn_cell_impl.GRUCell(2)(x, m)
2553 g, _ = rnn_cell_impl.GRUCell(2)(x, m)
2821 base_cell = rnn_cell_impl.GRUCell(
2845 base_cell = rnn_cell_impl.GRUCell(
2869 cell = rnn_cell_impl.GRUCell(3)
2900 cell = rnn_cell_impl.DeviceWrapper(rnn_cell_impl.GRUCell(3), gpu_dev)
2923 [rnn_cell_impl.GRUCell(2) for _ in range(2)],
[all …]
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drnn_common.py58 'gru': contrib_rnn.GRUCell,}
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dmodels.py337 cell_fn = contrib_rnn.GRUCell
/external/tensorflow/tensorflow/python/keras/
Dintegration_test.py152 model.add(keras.layers.RNN(rnn_cell.GRUCell(y_train.shape[-1],
/external/tensorflow/tensorflow/contrib/recurrent/python/kernel_tests/
Dfunctional_rnn_test.py55 'gru': (rnn_cell_impl.GRUCell, [_NUM_UNITS]),
/external/tensorflow/tensorflow/python/keras/layers/
D__init__.py152 from tensorflow.python.keras.layers.recurrent import GRUCell
Drecurrent_test.py464 cells = [keras.layers.recurrent.GRUCell(8),
482 cells = [keras.layers.recurrent.GRUCell(32, reset_after=True)]
684 keras.layers.GRUCell,
753 cell=[keras.layers.SimpleRNNCell, keras.layers.GRUCell,
Drecurrent.py1515 class GRUCell(DropoutRNNCellMixin, Layer): class
1588 super(GRUCell, self).__init__(**kwargs)
1777 base_config = super(GRUCell, self).get_config()
1899 cell = GRUCell(
/external/tensorflow/tensorflow/python/ops/
Drnn_cell_impl.py481 class GRUCell(LayerRNNCell): class
517 super(GRUCell, self).__init__(
605 base_config = super(GRUCell, self).get_config()
/external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/
Dgrid_rnn_cell.py553 cell_fn=lambda n: rnn.GRUCell(num_units=n),
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py76 class CudnnCompatibleGRUCell(rnn_cell_impl.GRUCell):
/external/tensorflow/
DRELEASE.md1167 `RNNCell`, `LSTMCell`, `GRUCell`, and a number of other cells. These