| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | gru_ops.h | 30 struct GRUCell { struct
31 GRUCell(const int batch_size, const int input_size, const int cell_size) in GRUCell() argument
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/ |
| D | 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)
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),
|
| D | 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)
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/ |
| D | 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)
854 [rnn_cell.GRUCell(24) for _ in range(2)], state_is_tuple=True)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.rnn_cell.pbtxt | 20 name: "GRUCell"
|
| D | tensorflow.keras.layers.-g-r-u-cell.pbtxt | 1 path: "tensorflow.keras.layers.GRUCell"
3 is_instance: "<class \'tensorflow.python.keras.layers.recurrent.GRUCell\'>"
|
| D | tensorflow.nn.rnn_cell.-g-r-u-cell.pbtxt | 1 path: "tensorflow.nn.rnn_cell.GRUCell"
3 is_instance: "<class \'tensorflow.python.ops.rnn_cell_impl.GRUCell\'>"
|
| D | tensorflow.keras.layers.pbtxt | 164 name: "GRUCell"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-g-r-u-cell.pbtxt | 1 path: "tensorflow.keras.layers.GRUCell"
3 is_instance: "<class \'tensorflow.python.keras.layers.recurrent.GRUCell\'>"
|
| D | tensorflow.keras.layers.pbtxt | 156 name: "GRUCell"
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ops/ |
| D | seq2seq_ops_test.py | 85 cell = rnn_cell.GRUCell(2)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | 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)
679 "GRUCell": rnn_cell_impl.GRUCell,
|
| D | rnn_cell_test.py | 945 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/ |
| D | rnn_common.py | 58 'gru': contrib_rnn.GRUCell,}
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | models.py | 337 cell_fn = contrib_rnn.GRUCell
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | integration_test.py | 152 model.add(keras.layers.RNN(rnn_cell.GRUCell(y_train.shape[-1],
|
| /external/tensorflow/tensorflow/contrib/recurrent/python/kernel_tests/ |
| D | functional_rnn_test.py | 55 'gru': (rnn_cell_impl.GRUCell, [_NUM_UNITS]),
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | __init__.py | 152 from tensorflow.python.keras.layers.recurrent import GRUCell
|
| D | recurrent_test.py | 464 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,
|
| D | recurrent.py | 1515 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/ |
| D | rnn_cell_impl.py | 481 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/ |
| D | grid_rnn_cell.py | 553 cell_fn=lambda n: rnn.GRUCell(num_units=n),
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 76 class CudnnCompatibleGRUCell(rnn_cell_impl.GRUCell):
|
| /external/tensorflow/ |
| D | RELEASE.md | 1167 `RNNCell`, `LSTMCell`, `GRUCell`, and a number of other cells. These
|